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

Use existing consul client for watcher

Browse Source
This commit is contained in:
Blair McMillan 2019-05-16 12:37:48 +10:00
parent 3dfe7d8703
commit 389d141c5a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
registry/consul/watcher.go
View File

@ -2,6 +2,8 @@ package consul
import ( import (
"errors" "errors"
"log"
"os"
"sync" "sync"
"github.com/hashicorp/consul/api" "github.com/hashicorp/consul/api"
@ -43,7 +45,7 @@ func newConsulWatcher(cr *consulRegistry, opts ...registry.WatchOption) (registr
} }
wp.Handler = cw.handle wp.Handler = cw.handle
go wp.Run(cr.Address) go wp.RunWithClientAndLogger(cr.Client, log.New(os.Stderr, "", log.LstdFlags))
cw.wp = wp cw.wp = wp
return cw, nil return cw, nil
@ -208,7 +210,7 @@ func (cw *consulWatcher) handle(idx uint64, data interface{}) {
}) })
if err == nil { if err == nil {
wp.Handler = cw.serviceHandler wp.Handler = cw.serviceHandler
go wp.Run(cw.r.Address) go wp.RunWithClientAndLogger(cw.r.Client, log.New(os.Stderr, "", log.LstdFlags))
cw.watchers[service] = wp cw.watchers[service] = wp
cw.next <- &registry.Result{Action: "create", Service: &registry.Service{Name: service}} cw.next <- &registry.Result{Action: "create", Service: &registry.Service{Name: service}}
} }