unistack-org/micro-router-register
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 7 Wiki Activity

cleanup router watcher logic

Browse Source
This commit is contained in:
Asim Aslam 2020-08-08 01:04:38 +01:00 committed by Vasiliy Tolstov
parent 1519dcb3e1
commit 3761461d15
1 changed files with 7 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
registry.go
View File

@ -570,39 +570,27 @@ func (r *rtr) start() error {
}() }()
go func() { go func() {
var err error
var w registry.Watcher
for { for {
select { select {
case <-r.exit: case <-r.exit:
if w != nil {
w.Stop()
}
return return
default: default:
if w == nil { w, err := r.options.Registry.Watch(registry.WatchDomain(registry.WildcardDomain))
w, err = r.options.Registry.Watch(registry.WatchDomain(registry.WildcardDomain)) if err != nil {
if err != nil { if logger.V(logger.WarnLevel, logger.DefaultLogger) {
if logger.V(logger.WarnLevel, logger.DefaultLogger) { logger.Warnf("failed creating registry watcher: %v", err)
logger.Warnf("failed creating registry watcher: %v", err)
}
time.Sleep(time.Second)
continue
} }
time.Sleep(time.Second)
continue
} }
// watchRegistry calls stop when it's done
if err := r.watchRegistry(w); err != nil { if err := r.watchRegistry(w); err != nil {
if logger.V(logger.WarnLevel, logger.DefaultLogger) { if logger.V(logger.WarnLevel, logger.DefaultLogger) {
logger.Warnf("Error watching the registry: %v", err) logger.Warnf("Error watching the registry: %v", err)
} }
time.Sleep(time.Second) time.Sleep(time.Second)
} }
if w != nil {
w.Stop()
w = nil
}
} }
} }
}() }()