blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-25 11:06:43 +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 package openKeeper
import ( import (
"context"
"net" "net"
"strconv" "strconv"
"sync" "sync"
"time" "time"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
"github.com/go-zookeeper/zk" "github.com/go-zookeeper/zk"
"google.golang.org/grpc" "google.golang.org/grpc"
"google.golang.org/grpc/resolver" "google.golang.org/grpc/resolver"
@ -102,8 +104,8 @@ func NewClient(zkServers []string, zkRoot string, options ...ZkOption) (*ZkClien
} }
resolver.Register(client) resolver.Register(client)
var wg sync.WaitGroup var wg sync.WaitGroup
wg.Add(1) wg.Add(2)
// go client.refresh(&wg) go client.refresh(&wg)
go client.watch(&wg) go client.watch(&wg)
wg.Wait() wg.Wait()
return client, nil return client, nil
@ -130,6 +132,7 @@ func (s *ZkClient) ensureAndCreate(node string) error {
func (s *ZkClient) refresh(wg *sync.WaitGroup) { func (s *ZkClient) refresh(wg *sync.WaitGroup) {
wg.Done() wg.Done()
for range s.ticker.C { for range s.ticker.C {
log.ZDebug(context.Background(), "refresh local conns")
s.lock.Lock() s.lock.Lock()
for rpcName := range s.resolvers { for rpcName := range s.resolvers {
s.flushResolver(rpcName) s.flushResolver(rpcName)