blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-11-05 03:42:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/3.6.1-code-conventions' into 3.6.1-code-conventions

Browse Source
This commit is contained in:
skiffer-git 2024-04-22 11:46:20 +08:00
commit 530f179526
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/common/discoveryregister/discoveryregister.go
View File

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