Add getMetadata to connection interface

DE-labtory · Sep 22, 2018 · 5faee3f · 5faee3f
1 parent 1b2a40b
commit 5faee3f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/server/server.go b/server/server.go
@@ -38,7 +38,7 @@ func (s Server) BifrostStream(streamServer pb.StreamService_BifrostStreamServer)
 	_, cf := context.WithCancel(context.Background())
 	streamWrapper := bifrost.NewServerStreamWrapper(streamServer, cf)
 
-	pub, metaData, err := handShake(streamWrapper, s.pubKey)
+	pub, metaData, err := s.handShake(streamWrapper, s.pubKey)
 
 	if err != nil {
 		return err
@@ -53,7 +53,7 @@ func (s Server) BifrostStream(streamServer pb.StreamService_BifrostStreamServer)
 	return nil
 }
 
-func handShake(streamWrapper bifrost.StreamWrapper, pubKey key.PubKey) (key.PubKey, map[string]string, error) {
+func (s Server) handShake(streamWrapper bifrost.StreamWrapper, pubKey key.PubKey) (key.PubKey, map[string]string, error) {
 
 	err := requestInfo(streamWrapper)
 
@@ -69,7 +69,7 @@ func handShake(streamWrapper bifrost.StreamWrapper, pubKey key.PubKey) (key.PubK
 		return nil, nil, err
 	}
 
-	err = sendInfo(streamWrapper, pubKey, metaData)
+	err = sendInfo(streamWrapper, pubKey, s.metaData)
 
 	if err != nil {
 		streamWrapper.Close()