fix: adjust connection states, isconnected()

nedpals · Feb 1, 2024 · 727f2f2 · 727f2f2
1 parent 815f944
commit 727f2f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/daemon/client/client.go b/server/daemon/client/client.go
@@ -18,9 +18,9 @@ type ConnectionState int
 
 const (
 	NotConnectedState ConnectionState = 0
+	ShutdownState     ConnectionState = iota
 	ConnectedState    ConnectionState = iota
 	InitializedState  ConnectionState = iota
-	ShutdownState     ConnectionState = iota
 )
 
 const defaultConnectDelay = 750 * time.Millisecond
@@ -58,7 +58,7 @@ func (c *Client) processIdField() jsonrpc2.CallOption {
 }
 
 func (c *Client) IsConnected() bool {
-	return c.rpcConn != nil && c.connState == ConnectedState
+	return c.rpcConn != nil && c.connState >= ConnectedState
 }
 
 func (c *Client) tryReconnect(reason error) error {