include ttl in registration response

This commit is contained in:
vyzo 2019-01-18 15:31:50 +02:00
parent 1ee2b55a9f
commit 8846a4b090
2 changed files with 4 additions and 2 deletions

View File

@ -101,9 +101,11 @@ func pbToPeerInfo(p *pb.Message_PeerInfo) (pstore.PeerInfo, error) {
return pstore.PeerInfo{ID: id, Addrs: addrs}, nil 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 := new(pb.Message_RegisterResponse)
r.Status = pb.Message_OK.Enum() r.Status = pb.Message_OK.Enum()
r.Ttl = &ttl64
return r return r
} }

2
svc.go
View File

@ -172,7 +172,7 @@ func (rz *RendezvousService) handleRegister(p peer.ID, m *pb.Message_Register) *
rzs.Register(p, ns, maddrs, ttl) rzs.Register(p, ns, maddrs, ttl)
} }
return newRegisterResponse() return newRegisterResponse(ttl)
} }
func (rz *RendezvousService) handleUnregister(p peer.ID, m *pb.Message_Unregister) error { func (rz *RendezvousService) handleUnregister(p peer.ID, m *pb.Message_Unregister) error {