micro/registry/handler/handler.go

83 lines
2.2 KiB
Go
Raw Normal View History

2019-09-09 19:20:17 +03:00
package handler
import (
"context"
"time"
2019-09-09 19:20:17 +03:00
"github.com/micro/go-micro/errors"
"github.com/micro/go-micro/registry"
pb "github.com/micro/go-micro/registry/proto"
"github.com/micro/go-micro/registry/service"
)
type Registry struct {
// internal registry
Registry registry.Registry
}
2019-09-09 19:25:47 +03:00
func (r *Registry) GetService(ctx context.Context, req *pb.GetRequest, rsp *pb.GetResponse) error {
2019-09-09 19:20:17 +03:00
services, err := r.Registry.GetService(req.Service)
2019-09-09 19:25:47 +03:00
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
2019-09-09 19:20:17 +03:00
for _, srv := range services {
rsp.Services = append(rsp.Services, service.ToProto(srv))
}
return nil
}
2019-09-09 19:25:47 +03:00
func (r *Registry) Register(ctx context.Context, req *pb.Service, rsp *pb.EmptyResponse) error {
var regOpts []registry.RegisterOption
2019-09-27 13:25:06 +03:00
ttl := time.Duration(req.Options.Ttl)
regOpts = append(regOpts, registry.RegisterTTL(ttl))
err := r.Registry.Register(service.ToService(req), regOpts...)
2019-09-09 19:20:17 +03:00
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
2019-09-09 19:20:17 +03:00
return nil
}
2019-09-09 19:25:47 +03:00
func (r *Registry) Deregister(ctx context.Context, req *pb.Service, rsp *pb.EmptyResponse) error {
err := r.Registry.Deregister(service.ToService(req))
2019-09-09 19:20:17 +03:00
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
return nil
}
2019-09-09 19:25:47 +03:00
func (r *Registry) ListServices(ctx context.Context, req *pb.ListRequest, rsp *pb.ListResponse) error {
2019-09-09 19:20:17 +03:00
services, err := r.Registry.ListServices()
2019-09-09 19:25:47 +03:00
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
2019-09-09 19:20:17 +03:00
for _, srv := range services {
rsp.Services = append(rsp.Services, service.ToProto(srv))
}
return nil
}
2019-09-09 19:25:47 +03:00
func (r *Registry) Watch(ctx context.Context, req *pb.WatchRequest, rsp pb.Registry_WatchStream) error {
watcher, err := r.Registry.Watch(registry.WatchService(req.Service))
2019-09-09 19:20:17 +03:00
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
for {
next, err := watcher.Next()
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
err = rsp.Send(&pb.Result{
Action: next.Action,
Service: service.ToProto(next.Service),
})
if err != nil {
return errors.InternalServerError("go.micro.registry", err.Error())
}
}
return nil
}