fix: list schema = flink list database

getindata · Mar 14, 2023 · a69e843 · a69e843
1 parent e3550f9
commit a69e843
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/dbt/adapters/flink/impl.py b/dbt/adapters/flink/impl.py
@@ -109,7 +109,7 @@ def get_relation(self, database: str, schema: str, identifier: str) -> Optional[
         return rel
 
     def list_schemas(self, database: str) -> List[str]:
-        return self.connections.show_catalogs()
+        return self.connections._show_databases(database)
 
     @classmethod
     def quote(cls, identifier: str) -> str:

diff --git a/flink/sqlgateway/schema_operations.py b/flink/sqlgateway/schema_operations.py
@@ -97,8 +97,10 @@ def show_databases(session: SqlGatewaySession, catalog: str) -> List[str]:
                 need_switch_catalog = True
 
         if need_switch_catalog:
-            if SchemaOperation.use_catalog_database(session, catalog):
-                raise RuntimeError(f"fail query database in {catalog}, please check catalog {catalog} exists")
+            try:
+                SchemaOperation.use_catalog_database(session, catalog)
+            except Exception as e:
+                raise RuntimeError(f"fail list database in {catalog}, please check {catalog} exists in this session")
         dbs = SchemaOperation.show_xxx(session, sql_show_databases)
 
         # switch back database if necessary