Close the connection on auth failure

cdb-dev · Aug 13, 2013 · 8d3e731 · 8d3e731
1 parent c4c0b85
commit 8d3e731
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/zk/conn.go b/zk/conn.go
@@ -175,9 +175,12 @@ func (c *Conn) loop() {
 	for {
 		c.connect()
 		err := c.authenticate()
-		if err == ErrSessionExpired {
+		switch {
+		case err == ErrSessionExpired:
 			c.invalidateWatches(err)
-		} else if err == nil {
+		case err != nil && c.conn != nil:
+			c.conn.Close()
+		case err == nil:
 			closeChan := make(chan bool) // channel to tell send loop stop
 			var wg sync.WaitGroup