Add timeout hack to mitigate timeouts

capitancambio · matt-fleming · commit d5dab1fc890b · 2024-02-21T18:30:12.000Z
diff --git a/src/databricks/sql/thrift_backend.py b/src/databricks/sql/thrift_backend.py
@@ -206,8 +206,12 @@ def __init__(
             **additional_transport_args,  # type: ignore
         )
 
-        timeout = THRIFT_SOCKET_TIMEOUT or kwargs.get("_socket_timeout", DEFAULT_SOCKET_TIMEOUT)
         # setTimeout defaults to 15 minutes and is expected in ms
+        # HACK!
+        timeout = THRIFT_SOCKET_TIMEOUT or kwargs.get("_socket_timeout", DEFAULT_SOCKET_TIMEOUT)
+        logger.info(f"Setting timeout HACK! to {timeout}")
+
+        # setTimeout defaults to None (i.e. no timeout), and is expected in ms
         self._transport.setTimeout(timeout and (float(timeout) * 1000.0))
 
         self._transport.setCustomHeaders(dict(http_headers))
