diff --git a/src/databricks/sql/auth/retry.py b/src/databricks/sql/auth/retry.py
index eedcc773f..57cfeed58 100755
--- a/src/databricks/sql/auth/retry.py
+++ b/src/databricks/sql/auth/retry.py
@@ -345,23 +345,24 @@ def should_retry(self, method: str, status_code: int) -> Tuple[bool, str]:
if a retry would violate the configured policy.
"""
+ logger.info(f"Received status code {status_code} for {method} request")
+
# Request succeeded. Don't retry.
if status_code == 200:
return False, "200 codes are not retried"
if status_code == 401:
- raise NonRecoverableNetworkError(
- "Received 401 - UNAUTHORIZED. Confirm your authentication credentials."
+ return (
+ False,
+ "Received 401 - UNAUTHORIZED. Confirm your authentication credentials.",
)
if status_code == 403:
- raise NonRecoverableNetworkError(
- "Received 403 - FORBIDDEN. Confirm your authentication credentials."
- )
+ return False, "403 codes are not retried"
# Request failed and server said NotImplemented. This isn't recoverable. Don't retry.
if status_code == 501:
- raise NonRecoverableNetworkError("Received code 501 from server.")
+ return False, "Received code 501 from server."
# Request failed and this method is not retryable. We only retry POST requests.
if not self._is_method_retryable(method):
@@ -400,8 +401,9 @@ def should_retry(self, method: str, status_code: int) -> Tuple[bool, str]:
and status_code not in self.status_forcelist
and status_code not in self.force_dangerous_codes
):
- raise UnsafeToRetryError(
- "ExecuteStatement command can only be retried for codes 429 and 503"
+ return (
+ False,
+ "ExecuteStatement command can only be retried for codes 429 and 503",
)
# Request failed with a dangerous code, was an ExecuteStatement, but user forced retries for this
diff --git a/src/databricks/sql/auth/thrift_http_client.py b/src/databricks/sql/auth/thrift_http_client.py
index 6273ab284..f0daae162 100644
--- a/src/databricks/sql/auth/thrift_http_client.py
+++ b/src/databricks/sql/auth/thrift_http_client.py
@@ -198,6 +198,12 @@ def flush(self):
self.message = self.__resp.reason
self.headers = self.__resp.headers
+ logger.info(
+ "HTTP Response with status code {}, message: {}".format(
+ self.code, self.message
+ )
+ )
+
@staticmethod
def basic_proxy_auth_headers(proxy):
if proxy is None or not proxy.username:
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy