strip comment when sql statment start with SET (#89)

cloudera · Dec 5, 2022 · 149c7ad · 149c7ad
1 parent f293ba9
commit 149c7ad
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/dbt/adapters/hive/connections.py b/dbt/adapters/hive/connections.py
@@ -351,6 +351,15 @@ def add_query(
                 additional_info = {}
                 logger.debug(f"Unable to get query header {ex}")
 
+        strip_sql = sql.split("*/")
+        if len(strip_sql) > 1:
+            strip_sql = strip_sql[1]
+        else:
+            strip_sql = strip_sql[0]
+
+        if strip_sql.strip().lower().startswith("set"):
+            sql = strip_sql
+
         with self.exception_handler(sql):
             if abridge_sql_log:
                 log_sql = "{}...".format(sql[:512])