blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

test discov

Browse Source
This commit is contained in:
wangchuxiao 2023-06-01 18:44:43 +08:00
parent 7087bbe76d
commit 2000b26049
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/discoveryregistry/zookeeper/zk.go
View File

@ -1,11 +1,13 @@
package openKeeper
import (
"context"
"net"
"strconv"
"sync"
"time"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
"github.com/go-zookeeper/zk"
"google.golang.org/grpc"
"google.golang.org/grpc/resolver"
@ -102,8 +104,8 @@ func NewClient(zkServers []string, zkRoot string, options ...ZkOption) (*ZkClien
}
resolver.Register(client)
var wg sync.WaitGroup
wg.Add(1)
// go client.refresh(&wg)
wg.Add(2)
go client.refresh(&wg)
go client.watch(&wg)
wg.Wait()
return client, nil
@ -130,6 +132,7 @@ func (s *ZkClient) ensureAndCreate(node string) error {
func (s *ZkClient) refresh(wg *sync.WaitGroup) {
wg.Done()
for range s.ticker.C {
log.ZDebug(context.Background(), "refresh local conns")
s.lock.Lock()
for rpcName := range s.resolvers {
s.flushResolver(rpcName)