Эх сурвалжийг харах

HUE-4616 [editor] Only select the snippet DB when executing the first statement

Romain Rigaux 9 жил өмнө
parent
commit
5e73a61ff7

+ 2 - 1
desktop/libs/notebook/src/notebook/connectors/hiveserver2.py

@@ -218,7 +218,8 @@ class HS2Api(Api):
     query = self._prepare_hql_query(snippet, statement['statement'], session)
 
     try:
-      db.use(query.database)
+      if statement.get('statement_id') == 0:
+        db.use(query.database)
       handle = db.client.query(query)
     except QueryServerException, ex:
       raise QueryError(ex.message, handle=statement)

+ 1 - 1
desktop/libs/notebook/src/notebook/connectors/rdbms.py

@@ -48,7 +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'])
+    db.use(snippet['database']) # TODO: only do the use on the first statement in a multi query
     table = db.execute_statement(snippet['statement'])  # TODO: execute statement stub in Rdbms
 
     data = list(table.rows())