fix: Leak in AutoReconnect

philippseith · Nov 18, 2021 · e50db63 · e50db63
1 parent 63fb61e
commit e50db63
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/client.go b/client.go
@@ -183,16 +183,15 @@ func (c *client) run() error {
 	// Run the loop
 	err = loop.Run(isLoopConnected)
 
+	if err == nil {
+		err = loop.hubConn.Close("", false) // allowReconnect value is ignored as servers never initiate a connection
+	}
+
 	// Reset conn to allow reconnecting
 	c.mx.Lock()
 	c.conn = nil
 	c.mx.Unlock()
 
-	if err != nil {
-		return err
-	}
-	err = loop.hubConn.Close("", false) // allowReconnect value is ignored as servers never initiate a connection
-
 	return err
 }