fix: mock zk registry (#1351)

apache · Jul 30, 2021 · 92df41c · 92df41c
1 parent b2ae992
commit 92df41c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,7 @@
 *.so
 *.dylib
 *.jar
+*.log
 
 # Test binary, build with `go test -c`
 *.test

diff --git a/registry/zookeeper/registry.go b/registry/zookeeper/registry.go
@@ -108,6 +108,7 @@ func newMockZkRegistry(url *common.URL, opts ...gxzookeeper.Option) (*zk.TestClu
 	if err != nil {
 		return nil, nil, err
 	}
+	r.WaitGroup().Add(1)
 	go zookeeper.HandleClientRestart(r)
 	r.InitListeners()
 	return c, r, nil