Skip to content

Commit 5b94291

Browse files
committed
add timeout parameters
1 parent b5ac6c1 commit 5b94291

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

dbt/adapters/sqlserver/sql_server_connection_manager.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,9 @@ def connect():
377377
con_str_concat,
378378
attrs_before=attrs_before,
379379
autocommit=True,
380+
timeout=credentials.login_timeout,
380381
)
382+
handle.timeout = credentials.query_timeout
381383
logger.debug(f"Connected to db: {credentials.database}")
382384
return handle
383385

dbt/adapters/sqlserver/sql_server_credentials.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@ class SQLServerCredentials(Credentials):
2222
trust_cert: Optional[bool] = False # default value in MS ODBC Driver 18 as well
2323
retries: int = 1
2424
schema_authorization: Optional[str] = None
25+
login_timeout: Optional[int] = None
26+
query_timeout: Optional[int] = None
2527

2628
_ALIASES = {
2729
"user": "UID",
@@ -58,6 +60,8 @@ def _connection_keys(self):
5860
"encrypt",
5961
"trust_cert",
6062
"retries",
63+
"login_timeout",
64+
"query_timeout",
6165
)
6266

6367
@property

0 commit comments

Comments
 (0)