unistack-org/micro
4
0
Fork 3
Code Issues 19 Pull Requests 1 Actions Packages Projects Releases 259 Wiki Activity

Naming changes

Browse Source
This commit is contained in:
Ben Toogood 2020-05-13 13:38:13 +01:00
parent e5c1fbc591
commit 25c937fd0e
2 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
registry/service/service.go
View File

@ -22,8 +22,8 @@ type serviceRegistry struct {
name string
// address
address []string
// registry is the proto client
registry pb.RegistryService
// client to call registry
client pb.RegistryService
}
func (s *serviceRegistry) callOpts() []client.CallOption {
@ -55,7 +55,7 @@ func (s *serviceRegistry) Init(opts ...registry.Option) error {
cli = grpc.NewClient()
}
s.registry = pb.NewRegistryService(DefaultService, cli)
s.client = pb.NewRegistryService(DefaultService, cli)
return nil
}
@ -78,7 +78,7 @@ func (s *serviceRegistry) Register(srv *registry.Service, opts ...registry.Regis
pbSrv.Options.Ttl = int64(options.TTL.Seconds())
// register the service
_, err := s.registry.Register(options.Context, pbSrv, s.callOpts()...)
_, err := s.client.Register(options.Context, pbSrv, s.callOpts()...)
if err != nil {
return err
}
@ -96,7 +96,7 @@ func (s *serviceRegistry) Deregister(srv *registry.Service, opts ...registry.Der
}
// deregister the service
_, err := s.registry.Deregister(options.Context, ToProto(srv), s.callOpts()...)
_, err := s.client.Deregister(options.Context, ToProto(srv), s.callOpts()...)
if err != nil {
return err
}
@ -112,7 +112,7 @@ func (s *serviceRegistry) GetService(name string, opts ...registry.GetOption) ([
options.Context = context.TODO()
}
rsp, err := s.registry.GetService(options.Context, &pb.GetRequest{
rsp, err := s.client.GetService(options.Context, &pb.GetRequest{
Service: name,
}, s.callOpts()...)
@ -136,7 +136,7 @@ func (s *serviceRegistry) ListServices(opts ...registry.ListOption) ([]*registry
options.Context = context.TODO()
}
rsp, err := s.registry.ListServices(options.Context, &pb.ListRequest{}, s.callOpts()...)
rsp, err := s.client.ListServices(options.Context, &pb.ListRequest{}, s.callOpts()...)
if err != nil {
return nil, err
}
@ -158,7 +158,7 @@ func (s *serviceRegistry) Watch(opts ...registry.WatchOption) (registry.Watcher,
options.Context = context.TODO()
}
stream, err := s.registry.Watch(options.Context, &pb.WatchRequest{
stream, err := s.client.Watch(options.Context, &pb.WatchRequest{
Service: options.Service,
}, s.callOpts()...)
@ -202,9 +202,9 @@ func NewRegistry(opts ...registry.Option) registry.Registry {
name := DefaultService
return &serviceRegistry{
opts: options,
name: name,
address: addrs,
registry: pb.NewRegistryService(name, cli),
opts: options,
name: name,
address: addrs,
client: pb.NewRegistryService(name, cli),
}
}

4
service.go
View File

@ -16,7 +16,7 @@ import (
"github.com/micro/go-micro/v2/debug/trace"
"github.com/micro/go-micro/v2/logger"
"github.com/micro/go-micro/v2/plugin"
srvRegistry "github.com/micro/go-micro/v2/registry/service"
registrySrv "github.com/micro/go-micro/v2/registry/service"
"github.com/micro/go-micro/v2/runtime"
"github.com/micro/go-micro/v2/server"
"github.com/micro/go-micro/v2/store"
@ -118,7 +118,7 @@ func (s *service) Init(opts ...Option) {
// Set the client for the micro clients
s.opts.Auth.Init(auth.WithClient(s.Client()))
s.opts.Registry.Init(srvRegistry.WithClient(s.Client()))
s.opts.Registry.Init(registrySrv.WithClient(s.Client()))
s.opts.Runtime.Init(runtime.WithClient(s.Client()))
s.opts.Store.Init(store.WithClient(s.Client()))
})