Merge pull request #395 from dbt-msft/add-timeout-support

Add support for timeout options

Author: sdebruyn

dbt/adapters/sqlserver/sql_server_connection_manager.py

@@ -377,7 +377,9 @@ def connect():
                 con_str_concat,
                 attrs_before=attrs_before,
                 autocommit=True,
+                timeout=credentials.login_timeout,
             )
+            handle.timeout = credentials.query_timeout
             logger.debug(f"Connected to db: {credentials.database}")
             return handle
 

dbt/adapters/sqlserver/sql_server_credentials.py

@@ -22,6 +22,8 @@ class SQLServerCredentials(Credentials):
     trust_cert: Optional[bool] = False  # default value in MS ODBC Driver 18 as well
     retries: int = 1
     schema_authorization: Optional[str] = None
+    login_timeout: Optional[int] = 0
+    query_timeout: Optional[int] = 0
 
     _ALIASES = {
         "user": "UID",
@@ -58,6 +60,8 @@ def _connection_keys(self):
             "encrypt",
             "trust_cert",
             "retries",
+            "login_timeout",
+            "query_timeout",
         )
 
     @property