add warnings when unable to parse sql query

datahub-project · Oct 23, 2024 · ff211af · ff211af
1 parent b0c8c08
commit ff211af
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_entities.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_entities.py
@@ -185,8 +185,12 @@ def _get_affected_tables(self) -> str:
         return ""
 
     def get_raw_query(self, sql_query: str) -> str:
-        parsed = parse_one(sql_query)
-        return parsed.sql(comments=False)
+        try:
+            parsed = parse_one(sql_query)
+            return parsed.sql(comments=False)
+        except Exception as e:
+            logger.warning(e)
+            return sql_query
 
 
 class DremioDataset: