SNOW-1468259 Add connection timeout and read timeout to storage client request (#1988)

* Add connection timeout and read timeout to storage client request

* Add changelog entry

Author: sfc-gh-stan

DESCRIPTION.md

@@ -9,6 +9,7 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
 # Release Notes
 
 - v3.12.0(TBD)
+  - Set default connection timeout of 10 seconds and socket read timeout of 10 minutes for HTTP calls in file transfer.
   - Optimized `to_pandas()` performance by fully parallel downloading logic.
   - Fixed a bug that specifying client_session_keep_alive_heartbeat_frequency in snowflake-sqlalchemy could crash the connector
 

src/snowflake/connector/constants.py

@@ -399,6 +399,7 @@ class IterUnit(Enum):
     TABLE_UNIT = "table"
 
 
+# File Transfer
 # Amazon S3 multipart upload limits
 # https://docs.aws.amazon.com/AmazonS3/latest/userguide/qfacts.html
 S3_DEFAULT_CHUNK_SIZE = 8 * 1024**2
@@ -410,6 +411,10 @@ class IterUnit(Enum):
 S3_CHUNK_SIZE = 8388608  # boto3 default
 AZURE_CHUNK_SIZE = 4 * megabyte
 
+# https://requests.readthedocs.io/en/latest/user/advanced/#timeouts
+REQUEST_CONNECTION_TIMEOUT = 10
+REQUEST_READ_TIMEOUT = 600
+
 DAY_IN_SECONDS = 60 * 60 * 24
 
 # TODO: all env variables definitions should be here

src/snowflake/connector/storage_client.py

@@ -19,7 +19,13 @@
 
 import OpenSSL
 
-from .constants import HTTP_HEADER_CONTENT_ENCODING, FileHeader, ResultStatus
+from .constants import (
+    HTTP_HEADER_CONTENT_ENCODING,
+    REQUEST_CONNECTION_TIMEOUT,
+    REQUEST_READ_TIMEOUT,
+    FileHeader,
+    ResultStatus,
+)
 from .encryption_util import EncryptionMetadata, SnowflakeEncryptionUtil
 from .errors import RequestExceedMaxRetryError
 from .file_util import SnowflakeFileUtil
@@ -280,6 +286,7 @@ def _send_request_with_retry(
         while self.retry_count[retry_id] < self.max_retry:
             cur_timestamp = self.credentials.timestamp
             url, rest_kwargs = get_request_args()
+            rest_kwargs["timeout"] = (REQUEST_CONNECTION_TIMEOUT, REQUEST_READ_TIMEOUT)
             try:
                 if conn:
                     with conn._rest._use_requests_session(url) as session: