소스 검색

HUE-4424 [librdbms] Support more than one database

Romain Rigaux 9 년 전
부모
커밋
cabc9ff38b
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      desktop/libs/notebook/src/notebook/connectors/rdbms.py

+ 2 - 0
desktop/libs/notebook/src/notebook/connectors/rdbms.py

@@ -48,6 +48,7 @@ class RdbmsApi(Api):
     query_server = dbms.get_query_server_config(server=self.interpreter)
     db = dbms.get(self.user, query_server)
 
+    db.use(snippet['database'])
     table = db.execute_statement(snippet['statement'])  # TODO: execute statement stub in Rdbms
 
     data = list(table.rows())
@@ -166,6 +167,7 @@ class Assist():
     return self.db.get_databases()
 
   def get_tables(self, database, table_names=[]):
+    self.db.use(database)
     return self.db.get_tables(database, table_names)
 
   def get_columns(self, database, table):