Set registry TTL properly via protobuf Options

This commit is contained in:
Milos Gajdos 2019-09-27 14:29:00 +01:00
parent e973bfaa25
commit d85ca7abd2

View File

@ -58,13 +58,12 @@ func (s *serviceRegistry) Register(srv *registry.Service, opts ...registry.Regis
o(&options) o(&options)
} }
ctx := context.Background() // encode srv into protobuf and pack Register TTL into it
if options.TTL.Nanoseconds() != 0.0 { pbSrv := ToProto(srv)
ctx = context.WithValue(ctx, "register_ttl", options.TTL) pbSrv.Options.Ttl = int64(options.TTL.Seconds())
}
// register the service // register the service
_, err := s.client.Register(ctx, ToProto(srv), s.callOpts()...) _, err := s.client.Register(context.TODO(), pbSrv, s.callOpts()...)
if err != nil { if err != nil {
return err return err
} }