should also consider disconnected case with disconnecting case

eclipse · Mar 31, 2024 · f024744 · f024744
1 parent 65bb896
commit f024744
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/client.go b/client.go
@@ -516,7 +516,7 @@ func (c *client) internalConnLost(whyConnLost error) {
 	DEBUG.Println(CLI, "internalConnLost called")
 	disDone, err := c.status.ConnectionLost(c.options.AutoReconnect && c.status.ConnectionStatus() > connecting)
 	if err != nil {
-		if err == errConnLossWhileDisconnecting {
+		if errors.Is(err, errConnLossWhileDisconnecting) || errors.Is(err, errAlreadyDisconnected) {
 			return // Loss of connection is expected or already being handled
 		}
 		ERROR.Println(CLI, fmt.Sprintf("internalConnLost unexpected status: %s", err.Error()))

diff --git a/status.go b/status.go
@@ -169,7 +169,7 @@ func (c *connectionStatus) connected(success bool) error {
 	}()
 
 	// Status may have moved to disconnecting in the interim (i.e. at users request)
-	if c.status == disconnecting {
+	if c.status == disconnecting || c.status == disconnected {
 		return errAbortConnection
 	}
 	if success {