From d359c34ca577bd20749806cef001a81525ec482a Mon Sep 17 00:00:00 2001
From: pantianying <601666418@qq.com>
Date: Tue, 26 Nov 2019 14:12:49 +0800
Subject: [PATCH] fix
---
registry/zookeeper/listener.go | 1 +
registry/zookeeper/registry.go | 2 +-
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/registry/zookeeper/listener.go b/registry/zookeeper/listener.go
index 777c460f5..c25028d58 100644
--- a/registry/zookeeper/listener.go
+++ b/registry/zookeeper/listener.go
@@ -77,6 +77,7 @@ type RegistryConfigurationListener struct {
}
func NewRegistryConfigurationListener(client *zk.ZookeeperClient, reg *zkRegistry) *RegistryConfigurationListener {
+ reg.wg.Add(1)
return &RegistryConfigurationListener{client: client, registry: reg, events: make(chan *config_center.ConfigChangeEvent, 32)}
}
func (l *RegistryConfigurationListener) Process(configType *config_center.ConfigChangeEvent) {
diff --git a/registry/zookeeper/registry.go b/registry/zookeeper/registry.go
index dc408a15f..13fb6c751 100644
--- a/registry/zookeeper/registry.go
+++ b/registry/zookeeper/registry.go
@@ -432,7 +432,7 @@ func (r *zkRegistry) Subscribe(url *common.URL, notifyListener registry.NotifyLi
time.Sleep(time.Duration(RegistryConnDelay) * time.Second)
continue
}
- r.wg.Add(1)
+
for {
if serviceEvent, err := listener.Next(); err != nil {
listener.Close()
--
GitLab