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

Pass client to more of the runtime

Browse Source
This commit is contained in:
Ben Toogood
2020-05-11 17:57:39 +01:00
parent f892b41299
commit efb64b7dbb
10 changed files with 56 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
config/source/service/service.go
View File

@@ -24,7 +24,7 @@ type service struct {
}
func (m *service) Read() (set *source.ChangeSet, err error) {
client := proto.NewConfigService(m.serviceName, client.DefaultClient)
client := proto.NewConfigService(m.serviceName, m.opts.Client)
req, err := client.Read(context.Background(), &proto.ReadRequest{
Namespace: m.namespace,
Path: m.path,
@@ -37,7 +37,7 @@ func (m *service) Read() (set *source.ChangeSet, err error) {
}
func (m *service) Watch() (w source.Watcher, err error) {
client := proto.NewConfigService(m.serviceName, client.DefaultClient)
client := proto.NewConfigService(m.serviceName, m.opts.Client)
stream, err := client.Watch(context.Background(), &proto.WatchRequest{
Namespace: m.namespace,
Path: m.path,
@@ -87,6 +87,10 @@ func NewSource(opts ...source.Option) source.Source {
}
}
if options.Client == nil {
options.Client = client.DefaultClient
}
s := &service{
serviceName: addr,
opts: options,