You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
790 B
44 lines
790 B
package etcd
|
|
|
|
import (
|
|
"github.com/go-kratos/kratos/v2/config"
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
)
|
|
|
|
type watcher struct {
|
|
source *source
|
|
ch clientv3.WatchChan
|
|
closeChan chan struct{}
|
|
}
|
|
|
|
func newWatcher(s *source) *watcher {
|
|
w := &watcher{
|
|
source: s,
|
|
closeChan: make(chan struct{}),
|
|
}
|
|
|
|
var opts []clientv3.OpOption
|
|
if s.options.prefix {
|
|
opts = append(opts, clientv3.WithPrefix())
|
|
}
|
|
w.ch = s.client.Watch(s.options.ctx, s.options.path, opts...)
|
|
|
|
return w
|
|
}
|
|
|
|
func (s *watcher) Next() ([]*config.KeyValue, error) {
|
|
select {
|
|
case resp := <-s.ch:
|
|
if resp.Err() != nil {
|
|
return nil, resp.Err()
|
|
}
|
|
return s.source.Load()
|
|
case <-s.closeChan:
|
|
return nil, config.ErrWatcherStopped
|
|
}
|
|
}
|
|
|
|
func (s *watcher) Stop() error {
|
|
close(s.closeChan)
|
|
return nil
|
|
}
|
|
|