kazanzhy commented on code in PR #27763:
URL: https://github.com/apache/airflow/pull/27763#discussion_r1026947118
##########
airflow/providers/snowflake/hooks/snowflake.py:
##########
@@ -321,6 +320,14 @@ def set_autocommit(self, conn, autocommit: Any) -> None:
def get_autocommit(self, conn):
return getattr(conn, "autocommit_mode", False)
+ @staticmethod
+ def split_sql_string(sql: str) -> list[str]:
+ split_statements_tuple = util_text.split_statements(StringIO(sql))
+ return [sql_string for sql_string, _ in split_statements_tuple if
sql_string]
+
+ def _update_query_ids(self, cursor) -> None:
+ self.query_ids.append(cursor.sfqid)
+
def run(
Review Comment:
This method still implemented here to keep Snowflake's `split_statements:
bool = True` default parameter
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
