2020-12-26 17:32:45 +02:00
|
|
|
package grpc
|
|
|
|
|
|
|
|
import (
|
2021-10-12 13:55:53 +02:00
|
|
|
"go-micro.dev/v4/config/source"
|
2021-10-13 14:31:23 +02:00
|
|
|
proto "github.com/asim/go-micro/plugins/config/source/grpc/v4/proto"
|
2020-12-26 17:32:45 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type watcher struct {
|
|
|
|
stream proto.Source_WatchClient
|
|
|
|
}
|
|
|
|
|
|
|
|
func newWatcher(stream proto.Source_WatchClient) (*watcher, error) {
|
|
|
|
return &watcher{
|
|
|
|
stream: stream,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *watcher) Next() (*source.ChangeSet, error) {
|
|
|
|
rsp, err := w.stream.Recv()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return toChangeSet(rsp.ChangeSet), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *watcher) Stop() error {
|
|
|
|
return w.stream.CloseSend()
|
|
|
|
}
|