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/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)

Loading…
Cancel
Save