diff --git a/pkg/discoveryregistry/zookeeper/zk.go b/pkg/discoveryregistry/zookeeper/zk.go index 1247dcc07..19f06f91a 100644 --- a/pkg/discoveryregistry/zookeeper/zk.go +++ b/pkg/discoveryregistry/zookeeper/zk.go @@ -34,7 +34,7 @@ type ZkClient struct { node string ticker *time.Ticker - lock sync.RWMutex + lock Lock options []grpc.DialOption resolvers map[string]*Resolver @@ -91,6 +91,7 @@ func NewClient(zkServers []string, zkRoot string, options ...ZkOption) (*ZkClien timeout: timeout, localConns: make(map[string][]resolver.Address), resolvers: make(map[string]*Resolver), + lock: &FakeLock{}, } client.ticker = time.NewTicker(defaultFreq) for _, option := range options { @@ -196,3 +197,13 @@ func (s *ZkClient) AddOption(opts ...grpc.DialOption) { func (s *ZkClient) GetClientLocalConns() map[string][]resolver.Address { return s.localConns } + +type Lock interface { + Lock() + Unlock() +} + +type FakeLock struct{} + +func (s *FakeLock) Lock() {} +func (s *FakeLock) Unlock() {}