stop query on the processing engine when user clicks STOP on SQL Lab

apache · Feb 17, 2022 · 5805fb3 · 5805fb3
1 parent f5f25ff
commit 5805fb3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/superset/db_engine_specs/hive.py b/superset/db_engine_specs/hive.py
@@ -335,6 +335,10 @@ def handle_cursor(  # pylint: disable=too-many-locals
         job_id = None
         query_id = query.id
         while polled.operationState in unfinished_states:
+            # Queries don't terminate when user clicks the STOP button on SQL LAB.
+            # Refresh session so that the `query.status` modified in stop_query in
+            # views/core.py is reflected here.
+            session.refresh(query)
             query = session.query(type(query)).filter_by(id=query_id).one()
             if query.status == QueryStatus.STOPPED:
                 cursor.cancel()