fix: SSH Connection refused traceback (#261)

aerospike · Jan 26, 2024 · 49bdbb3 · 49bdbb3
1 parent 368a0ad
commit 49bdbb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/live_cluster/ssh.py b/lib/live_cluster/ssh.py
@@ -125,7 +125,7 @@ async def create_connection(self, ip) -> SSHConnection:
             async with SSHConnectionFactory.semaphore_host_dict[ip].semaphore:
                 logger.debug(f"{ip}: Creating connection")
                 return SSHConnection(await asyncssh.connect(ip, options=self.opts))
-        except asyncssh.DisconnectError as e:
+        except (asyncssh.DisconnectError, ConnectionRefusedError) as e:
             raise SSHConnectionError(e)
         finally:
             SSHConnectionFactory.semaphore_host_dict[ip].count -= 1