refactor: zk log debug.

pull/2202/head
Gordon 1 year ago
parent ac94ede57f
commit fd9cd3c7ad

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

Loading…
Cancel
Save