diff --git a/proto.go b/proto.go index b145f00..9ae60f2 100644 --- a/proto.go +++ b/proto.go @@ -101,9 +101,11 @@ func pbToPeerInfo(p *pb.Message_PeerInfo) (pstore.PeerInfo, error) { return pstore.PeerInfo{ID: id, Addrs: addrs}, nil } -func newRegisterResponse() *pb.Message_RegisterResponse { +func newRegisterResponse(ttl int) *pb.Message_RegisterResponse { + ttl64 := int64(ttl) r := new(pb.Message_RegisterResponse) r.Status = pb.Message_OK.Enum() + r.Ttl = &ttl64 return r } diff --git a/svc.go b/svc.go index 57227f5..0be1f07 100644 --- a/svc.go +++ b/svc.go @@ -172,7 +172,7 @@ func (rz *RendezvousService) handleRegister(p peer.ID, m *pb.Message_Register) * rzs.Register(p, ns, maddrs, ttl) } - return newRegisterResponse() + return newRegisterResponse(ttl) } func (rz *RendezvousService) handleUnregister(p peer.ID, m *pb.Message_Unregister) error {