diff --git a/pkg/discoveryregistry/zookeeper/zk.go b/pkg/discoveryregistry/zookeeper/zk.go index 09fe6a6f0..36b75f0af 100644 --- a/pkg/discoveryregistry/zookeeper/zk.go +++ b/pkg/discoveryregistry/zookeeper/zk.go @@ -18,6 +18,10 @@ const ( timeout = 5 ) +type Logger interface { + Printf(string, ...interface{}) +} + type ZkClient struct { zkServers []string zkRoot string @@ -40,7 +44,7 @@ type ZkClient struct { balancerName string RoundRobin - logger *log.ZkLogger + logger Logger } type ZkOption func(*ZkClient) @@ -76,7 +80,7 @@ func WithTimeout(timeout int) ZkOption { } } -func WithLogger(logger *log.ZkLogger) ZkOption { +func WithLogger(logger Logger) ZkOption { return func(client *ZkClient) { client.logger = logger }