add more http_client closures

Author: Varun0157

src/databricks/sql/backend/thrift_backend.py

@@ -232,7 +232,7 @@ def __init__(
         try:
             self._transport.open()
         except:
-            self._transport.release_connection()
+            self._transport.close()
             raise
 
         self._request_lock = threading.RLock()
@@ -607,7 +607,7 @@ def open_session(self, session_configuration, catalog, schema) -> SessionId:
             self._session_id_hex = session_id.hex_guid
             return session_id
         except:
-            self._transport.release_connection()
+            self._transport.close()
             raise
 
     def close_session(self, session_id: SessionId) -> None:

src/databricks/sql/client.py

@@ -284,6 +284,7 @@ def read(self) -> Optional[OAuthToken]:
                 if hasattr(self, "session")
                 else None,
             )
+            self.http_client.close()
             raise e
 
         self.use_inline_params = self._set_use_inline_params_with_warning(
@@ -362,8 +363,9 @@ def __exit__(self, exc_type, exc_value, traceback):
     def __del__(self):
         if self.open:
             logger.debug(
-                "Closing unclosed connection for session "
-                "{}".format(self.get_session_id_hex())
+                "Closing unclosed connection for session " "{}".format(
+                    self.get_session_id_hex()
+                )
             )
             try:
                 self._close(close_cursors=False)

src/databricks/sql/telemetry/telemetry_client.py

@@ -359,6 +359,7 @@ def close(self):
         """Flush remaining events before closing"""
         logger.debug("Closing TelemetryClient for connection %s", self._session_id_hex)
         self._flush()
+        self._http_client.close()
 
 
 class TelemetryClientFactory:
@@ -460,7 +461,6 @@ def initialize_telemetry_client(
     ):
         """Initialize a telemetry client for a specific connection if telemetry is enabled"""
         try:
-
             with TelemetryClientFactory._lock:
                 TelemetryClientFactory._initialize()