Update dremio_sql_queries.py

datahub-project · Oct 10, 2024 · e4ad9a3 · e4ad9a3
1 parent bd2727e
commit e4ad9a3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_sql_queries.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_sql_queries.py
@@ -118,6 +118,8 @@ class DremioSQLQueries:
             COLUMN_SIZE
         FROM
             INFORMATION_SCHEMA.COLUMNS
+        WHERE
+            LOCATE('{container_name}', LOWER(TABLE_SCHEMA)) = 1
         ) C
         ON
             CONCAT(REPLACE(REPLACE(REPLACE(V.PATH, ', ', '.'), '[', ''), ']', '')) =
@@ -220,6 +222,8 @@ class DremioSQLQueries:
             COLUMN_SIZE
         FROM
             INFORMATION_SCHEMA.COLUMNS
+        WHERE
+            LOCATE('{container_name}', LOWER(TABLE_SCHEMA)) = 1
         ) C
         ON
             CONCAT(REPLACE(REPLACE(REPLACE(V.PATH, ', ', '.'), '[', ''), ']', '')) =