modify to fill IP in PeerInfo when build response peer info

DE-labtory · Nov 4, 2018 · 85927f2 · 85927f2
1 parent 12c6ccb
commit 85927f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/util.go b/util.go
@@ -41,14 +41,15 @@ type KeyOpts struct {
 	PubKey Key
 }
 
-func BuildResponsePeerInfo(pubKey Key, metaData map[string]string) (*pb.Envelope, error) {
+func BuildResponsePeerInfo(ip string, pubKey Key, metaData map[string]string) (*pb.Envelope, error) {
 	b, err := pubKey.ToByte()
 
 	if err != nil {
 		return nil, err
 	}
 
 	pi := &PeerInfo{
+		IP:          ip,
 		PubKeyBytes: b,
 		IsPrivate:   pubKey.IsPrivate(),
 		MetaData:    metaData,