fix deadloop bug (#102)

* fix deadloop bug * revert import
dubbogo · Jul 19, 2022 · 01d5bc0 · 01d5bc0
1 parent 927aad2
commit 01d5bc0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/database/kv/zk/client.go b/database/kv/zk/client.go
@@ -215,12 +215,17 @@ func NewMockZookeeperClient(name string, timeout time.Duration, opts ...Option)
 // HandleZkEvent handles zookeeper events
 func (d *DefaultHandler) HandleZkEvent(z *ZookeeperClient) {
 	var (
+		ok    bool
 		state int
 		event zk.Event
 	)
 	for {
 		select {
-		case event = <-z.Session:
+		case event, ok = <-z.Session:
+			if !ok {
+				// channel already closed
+				return
+			}
 			switch event.State {
 			case zk.StateDisconnected:
 				atomic.StoreUint32(&z.valid, 0)