bug: Sessions should stay open even if snowflake is idle for hours (#272)

visch · pre-commit-ci[bot] · web-flow · commit 5d883378f810 · 2024-10-10T13:24:10.000-05:00
Fixes #270 --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
diff --git a/target_snowflake/connector.py b/target_snowflake/connector.py
@@ -218,6 +218,7 @@ def create_engine(self) -> Engine:
             "session_parameters": {
                 "QUOTED_IDENTIFIERS_IGNORE_CASE": "TRUE",
             },
+            "client_session_keep_alive": True,  # See https://github.com/snowflakedb/snowflake-connector-python/issues/218
         }
         if self.auth_method == SnowflakeAuthMethod.KEY_PAIR:
             connect_args["private_key"] = self.get_private_key()

Original file line number	Diff line number	Diff line change
`@@ -218,6 +218,7 @@ def create_engine(self) -> Engine:`
`218`	`218`	`"session_parameters": {`
`219`	`219`	`"QUOTED_IDENTIFIERS_IGNORE_CASE": "TRUE",`
`220`	`220`	`},`
	`221`	`+ "client_session_keep_alive": True, # See https://github.com/snowflakedb/snowflake-connector-python/issues/218`
`221`	`222`	`}`
`222`	`223`	`if self.auth_method == SnowflakeAuthMethod.KEY_PAIR:`
`223`	`224`	`connect_args["private_key"] = self.get_private_key()`