瀏覽代碼

HUE-8740 [sql] Enable sqlalchemy result streaming

(cherry picked from commit 3749a2929d9b68fc57490c2e7c6b0d074d0a1915)
jdesjean 6 年之前
父節點
當前提交
49db0ab1bc
共有 1 個文件被更改,包括 3 次插入3 次删除
  1. 3 3
      desktop/libs/notebook/src/notebook/connectors/sqlalchemyapi.py

+ 3 - 3
desktop/libs/notebook/src/notebook/connectors/sqlalchemyapi.py

@@ -115,7 +115,7 @@ class SqlAlchemyApi(Api):
     if not self.engine:
       self.engine = self._create_engine()
     connection = self.engine.connect()
-    result = connection.execute(snippet['statement'])
+    result = connection.execution_options(stream_results=True).execute(snippet['statement'])
     cache = {
       'connection': connection,
       'result': result,
@@ -217,7 +217,7 @@ class SqlAlchemyApi(Api):
     guid = uuid.uuid4().hex
 
     connection = self.engine.connect()
-    result = connection.execute(snippet['statement'])
+    result = connection.execution_options(stream_results=True).execute(snippet['statement'])
 
     CONNECTION_CACHE[guid] = {
       'connection': connection,
@@ -340,7 +340,7 @@ class Assist():
     statement = "SELECT %s FROM `%s`.`%s` LIMIT %d" % (column, database, table, 100)
     connection = self.engine.connect()
     try:
-      result = connection.execute(statement)
+      result = connection.execution_options(stream_results=True).execute(statement)
       return result.cursor.description, result.fetchall()
     finally:
       connection.close()