(*PeerRecord.MarshalRecord): Use ToProtobuf() to get protobuf struct

This commit is contained in:
Vibhav Pant 2020-03-25 17:42:49 +05:30
parent 4013be10ed
commit 075ac7484c
No known key found for this signature in database
GPG Key ID: BA939833A10FCBAA

View File

@ -153,16 +153,11 @@ func (r *PeerRecord) UnmarshalRecord(bytes []byte) error {
// This method is called automatically when constructing a routing.Envelope // This method is called automatically when constructing a routing.Envelope
// using Seal or PeerRecord.Sign. // using Seal or PeerRecord.Sign.
func (r *PeerRecord) MarshalRecord() ([]byte, error) { func (r *PeerRecord) MarshalRecord() ([]byte, error) {
idBytes, err := r.PeerID.MarshalBinary() msg, err := r.ToProtobuf()
if err != nil { if err != nil {
return nil, err return nil, err
} }
msg := pb.PeerRecord{ return proto.Marshal(msg)
PeerId: idBytes,
Addresses: addrsToProtobuf(r.Addrs),
Seq: r.Seq,
}
return proto.Marshal(&msg)
} }
// Equal returns true if the other PeerRecord is identical to this one. // Equal returns true if the other PeerRecord is identical to this one.