Metadata query speedup

databrickslabs · Sep 29, 2023 · abdb87d · abdb87d
1 parent 5f4ba17
commit abdb87d
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/discoverx/explorer.py b/discoverx/explorer.py
@@ -56,9 +56,20 @@ def _get_table_list_sql(self, catalogs: str, schemas: str, tables: str, columns:
             string: The SQL expression
         """
 
-        catalog_sql = f"""AND regexp_like(table_catalog, "^{catalogs.replace("*", ".*")}$")"""
-        schema_sql = f"""AND regexp_like(table_schema, "^{schemas.replace("*", ".*")}$")"""
-        table_sql = f"""AND regexp_like(table_name, "^{tables.replace("*", ".*")}$")"""
+        if "*" in catalogs:
+            catalog_sql = f"""AND regexp_like(table_catalog, "^{catalogs.replace("*", ".*")}$")"""
+        else:
+            catalog_sql = f"""AND table_catalog = "{catalogs}" """
+
+        if "*" in schemas:
+            schema_sql = f"""AND regexp_like(table_schema, "^{schemas.replace("*", ".*")}$")"""
+        else:
+            schema_sql = f"""AND table_schema = "{schemas}" """
+
+        if "*" in tables:
+            table_sql = f"""AND regexp_like(table_name, "^{tables.replace("*", ".*")}$")"""
+        else:
+            table_sql = f"""AND table_name = "{tables}" """
 
         if columns:
             match_any_col = "|".join([f'({c.replace("*", ".*")})' for c in columns])