update cluster sub

xobserve · Apr 3, 2019 · ca3e175 · ca3e175
1 parent 50b4e9c
commit ca3e175
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/broker/client.go b/broker/client.go
@@ -3,6 +3,7 @@ package broker
 import (
 	"bufio"
 	"errors"
+	"fmt"
 	"net"
 	"sync/atomic"
 	"time"
@@ -58,6 +59,7 @@ func (c *client) process() {
 		c.bk.Unlock()
 
 		c.closed = true
+		c.closech <- struct{}{}
 		c.conn.Close()
 		g.L.Debug("client closed", zap.Uint64("conn_id", c.id))
 	}()
@@ -142,6 +144,7 @@ func (c *client) onSubscribe(topic []byte) error {
 func (c *client) sendLoop() {
 	defer func() {
 		// when disconnect, automaticly unsubscribe the topic
+		fmt.Println("here1111:", c.subs)
 		for tid := range c.subs {
 			c.bk.cluster.Unsubscribe(tid, c.id)
 		}