diff --git a/pkg/common/discoveryregister/discoveryregister.go b/pkg/common/discoveryregister/discoveryregister.go index b664ee34e..287b9dbf0 100644 --- a/pkg/common/discoveryregister/discoveryregister.go +++ b/pkg/common/discoveryregister/discoveryregister.go @@ -20,6 +20,8 @@ import ( "github.com/openimsdk/tools/discovery" "github.com/openimsdk/tools/discovery/zookeeper" "github.com/openimsdk/tools/errs" + "github.com/openimsdk/tools/log" + "os" "time" ) @@ -33,6 +35,10 @@ const ( func NewDiscoveryRegister(zookeeperConfig *config.ZooKeeper, share *config.Share) (discovery.SvcDiscoveryRegistry, error) { switch share.Env { case zookeeperConst: + zklogger, err := log.NewConsoleZapLogger("", 6, false, config.Version, os.Stdout) + if err != nil { + return nil, errs.Wrap(err) + } return zookeeper.NewZkClient( zookeeperConfig.Address, zookeeperConfig.Schema, @@ -40,6 +46,7 @@ func NewDiscoveryRegister(zookeeperConfig *config.ZooKeeper, share *config.Share zookeeper.WithUserNameAndPassword(zookeeperConfig.Username, zookeeperConfig.Password), zookeeper.WithRoundRobin(), zookeeper.WithTimeout(10), + zookeeper.WithLogger(zklogger), ) case kubenetesConst: return kubernetes.NewK8sDiscoveryRegister(share.RpcRegisterName.MessageGateway)