Add TTL options to memory registry.

2019-09-27 11:25:06 +01:00
parent 27bd9581bf
commit e973bfaa25
3 changed files with 108 additions and 53 deletions
--- a/registry/handler/handler.go
+++ b/registry/handler/handler.go
@@ -28,10 +28,8 @@ func (r *Registry) GetService(ctx context.Context, req *pb.GetRequest, rsp *pb.G

 func (r *Registry) Register(ctx context.Context, req *pb.Service, rsp *pb.EmptyResponse) error {
 	var regOpts []registry.RegisterOption
-	regTTL, ok := ctx.Value("register_ttl").(time.Duration)
-	if ok {
-		regOpts = append(regOpts, registry.RegisterTTL(regTTL))
-	}
+	ttl := time.Duration(req.Options.Ttl)
+	regOpts = append(regOpts, registry.RegisterTTL(ttl))

 	err := r.Registry.Register(service.ToService(req), regOpts...)
 	if err != nil {