8c7c27c573
* registry/memory: add support for the domain options * registry/memory: swap Fatal test cases with Error * registry/memory: fix wildcard not found bug * registry/memory: replace locks with rlocks * registry/memory: fix deregistration bug
41 lines
641 B
Go
41 lines
641 B
Go
package memory
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/micro/go-micro/v2/registry"
|
|
)
|
|
|
|
type Watcher struct {
|
|
id string
|
|
wo registry.WatchOptions
|
|
res chan *registry.Result
|
|
exit chan bool
|
|
}
|
|
|
|
func (m *Watcher) Next() (*registry.Result, error) {
|
|
for {
|
|
select {
|
|
case r := <-m.res:
|
|
if len(m.wo.Service) > 0 && m.wo.Service != r.Service.Name {
|
|
continue
|
|
}
|
|
if m.wo.Domain != registry.WildcardDomain && m.wo.Domain != m.wo.Domain {
|
|
continue
|
|
}
|
|
return r, nil
|
|
case <-m.exit:
|
|
return nil, errors.New("watcher stopped")
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *Watcher) Stop() {
|
|
select {
|
|
case <-m.exit:
|
|
return
|
|
default:
|
|
close(m.exit)
|
|
}
|
|
}
|