Added Registry TTL to memory registry. Tracking node lifetimes.

This commit is contained in:
Milos Gajdos
2019-09-25 16:20:42 +01:00
parent 219d759f1d
commit 16c7b3a390
3 changed files with 114 additions and 17 deletions

View File

@@ -2,6 +2,7 @@ package handler
import (
"context"
"time"
"github.com/micro/go-micro/errors"
"github.com/micro/go-micro/registry"
@@ -26,10 +27,17 @@ 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 {
err := r.Registry.Register(service.ToService(req))
var regOpts []registry.RegisterOption
regTTL, ok := ctx.Value("register_ttl").(time.Duration)
if ok {
regOpts = append(regOpts, registry.RegisterTTL(regTTL))
}
err := r.Registry.Register(service.ToService(req), regOpts...)
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
return nil
}