瀏覽代碼

HUE-9291 [flink] Provide check_status results in a result attribute

Romain 5 年之前
父節點
當前提交
cb6d034be3

+ 2 - 2
desktop/core/src/desktop/js/apps/notebook2/execution/executable.js

@@ -289,12 +289,12 @@ export default class Executable {
             break;
           case EXECUTION_STATUS.streaming:
             if (window.WEB_SOCKETS_ENABLED) {
-              huePubSub.publish('editor.ws.query.fetch_result', queryStatus);
+              huePubSub.publish('editor.ws.query.fetch_result', queryStatus.result);
             } else {
               if (!this.result) {
                 this.result = new ExecutionResult(this, true);
               }
-              this.result.handleResultResponse(queryStatus);
+              this.result.handleResultResponse(queryStatus.result);
             }
           case EXECUTION_STATUS.running:
           case EXECUTION_STATUS.starting:

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

@@ -153,7 +153,7 @@ class FlinkSqlApi(Api):
           resp = self.db.fetch_status(session['id'], statement_id)
           if resp.get('status') == 'RUNNING':
             status = 'streaming'
-            response.update(self.fetch_result(notebook, snippet, n, False))
+            response['result'] = self.fetch_result(notebook, snippet, n, False)
           elif resp.get('status') == 'FINISHED':
             status = 'available'
           elif resp.get('status') == 'FAILED':