fix(sql lab): View result button is not showing consistently (#20164)

apache · Jun 3, 2022 · 9205785 · 9205785
1 parent a020f75
commit 9205785
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/superset-frontend/src/SqlLab/reducers/sqlLab.js b/superset-frontend/src/SqlLab/reducers/sqlLab.js
@@ -340,6 +340,12 @@ export default function sqlLabReducer(state = {}, action) {
         errorMessage: null,
         cached: false,
       };
+
+      const resultsKey = action?.results?.query?.resultsKey;
+      if (resultsKey) {
+        alts.resultsKey = resultsKey;
+      }
+
       return alterInObject(state, 'queries', action.query, alts);
     },
     [actions.QUERY_FAILED]() {

diff --git a/superset/sql_lab.py b/superset/sql_lab.py
@@ -548,6 +548,7 @@ def execute_sql_statements(  # pylint: disable=too-many-arguments, too-many-loca
 
     if store_results and results_backend:
         key = str(uuid.uuid4())
+        payload["query"]["resultsKey"] = key
         logger.info(
             "Query %s: Storing results in results backend, key: %s", str(query_id), key
         )