Improve status checks before closing underlying connection

exasol · Nov 3, 2023 · dc3a230 · dc3a230
1 parent 500fc32
commit dc3a230
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/exasol/driver/websocket/_connection.py b/exasol/driver/websocket/_connection.py
@@ -119,10 +119,12 @@ def connection(self):
 
     def close(self):
         """See also :py:meth: `Connection.close`"""
-        if not self._connection:
+        connection_to_close = self._connection
+        self._connection = None
+        if connection_to_close is None or connection_to_close.is_closed:
             return
         try:
-            self._connection.close()
+            connection_to_close.close()
         except Exception as ex:
             raise Error() from ex