unistack-org/micro
4
0
Fork 2
Code Issues 18 Pull Requests Actions Packages Projects Releases 324 Wiki Activity

Add registry handler

Browse Source
This commit is contained in:
Asim Aslam
2019-09-09 09:20:17 -07:00
parent b076ef906a
commit 1f44d7a4a1
4 changed files with 77 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
registry/service/service.go
View File

@@ -59,7 +59,7 @@ func (s *serviceRegistry) Register(srv *registry.Service, opts ...registry.Regis
}
// register the service
_, err := s.client.Register(context.TODO(), toProto(srv), s.callOpts()...)
_, err := s.client.Register(context.TODO(), ToProto(srv), s.callOpts()...)
if err != nil {
return err
}
@@ -69,7 +69,7 @@ func (s *serviceRegistry) Register(srv *registry.Service, opts ...registry.Regis
func (s *serviceRegistry) Deregister(srv *registry.Service) error {
// deregister the service
_, err := s.client.Deregister(context.TODO(), toProto(srv), s.callOpts()...)
_, err := s.client.Deregister(context.TODO(), ToProto(srv), s.callOpts()...)
if err != nil {
return err
}
@@ -87,7 +87,7 @@ func (s *serviceRegistry) GetService(name string) ([]*registry.Service, error) {
var services []*registry.Service
for _, service := range rsp.Services {
services = append(services, toService(service))
services = append(services, ToService(service))
}
return services, nil
}
@@ -100,7 +100,7 @@ func (s *serviceRegistry) ListServices() ([]*registry.Service, error) {
var services []*registry.Service
for _, service := range rsp.Services {
services = append(services, toService(service))
services = append(services, ToService(service))
}
return services, nil

4
registry/service/util.go
View File

@@ -37,7 +37,7 @@ func toValues(v []*pb.Value) []*registry.Value {
return vs
}
func toProto(s *registry.Service) *pb.Service {
func ToProto(s *registry.Service) *pb.Service {
var endpoints []*pb.Endpoint
for _, ep := range s.Endpoints {
var request, response *pb.Value
@@ -85,7 +85,7 @@ func toProto(s *registry.Service) *pb.Service {
}
}
func toService(s *pb.Service) *registry.Service {
func ToService(s *pb.Service) *registry.Service {
var endpoints []*registry.Endpoint
for _, ep := range s.Endpoints {
var request, response *registry.Value

2
registry/service/watcher.go
View File

@@ -26,7 +26,7 @@ func (s *serviceWatcher) Next() (*registry.Result, error) {
return &registry.Result{
Action: r.Action,
Service: toService(r.Service),
Service: ToService(r.Service),
}, nil
}
}