close socket on connection failure

Mapepire-IBMi · Aug 30, 2024 · c791a84 · c791a84
1 parent 4c4adb8
commit c791a84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/mapepire_python/client/sql_job.py b/mapepire_python/client/sql_job.py
@@ -111,6 +111,7 @@ def connect(self, db2_server: Union[DaemonServer, Dict[str, Any]]) -> Dict[Any,
             self._status = JobStatus.Ready
         else:
             self._status = JobStatus.NotStarted
+            self.close()
             raise Exception(result.get("error", "Failed to connect to server"))
 
         self.id = result["job"]

diff --git a/mapepire_python/pool/pool_job.py b/mapepire_python/pool/pool_job.py
@@ -197,7 +197,7 @@ async def connect(self, db2_server: Union[DaemonServer, Dict[str, Any]]) -> Dict
             self.status = JobStatus.Ready
         else:
             self.status = JobStatus.NotStarted
-            print(result)
+            await self.close()
             raise Exception(result.get("error", "Failed to connect to server"))
 
         self.id = result["job"]