1
0
mirror of https://github.com/go-kratos/kratos.git synced 2025-02-19 19:10:08 +02:00
包子 eaf0ceab0c
chore(registry): unified management of decentralized warehouses ()
* chore(registry): unified management of decentralized warehouses
2021-08-24 11:35:26 +08:00

42 lines
673 B
Go

package registry
import (
"context"
"github.com/go-kratos/kratos/v2/registry"
)
var (
_ registry.Watcher = &watcher{}
)
type watcher struct {
ctx context.Context
cancel context.CancelFunc
event chan struct{}
set *serviceSet
}
func (w watcher) Next() (services []*registry.ServiceInstance, err error) {
select {
case <-w.ctx.Done():
err = w.ctx.Err()
case <-w.event:
}
ss, ok := w.set.services.Load().([]*registry.ServiceInstance)
if ok {
for _, s := range ss {
services = append(services, s)
}
}
return
}
func (w *watcher) Stop() error {
w.cancel()
w.set.lock.Lock()
defer w.set.lock.Unlock()
delete(w.set.watcher, w)
return nil
}