From d632590eff8dedb31ecf01e6f610f098b7cfea48 Mon Sep 17 00:00:00 2001 From: saurabh gulati Date: Tue, 22 Feb 2022 08:19:34 +0100 Subject: [PATCH] stop query on the processing engine when user clicks STOP on SQL Lab (#18789) Co-authored-by: Saurabh Gulati --- superset/db_engine_specs/hive.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/superset/db_engine_specs/hive.py b/superset/db_engine_specs/hive.py index 51becd7705..611a98448b 100644 --- a/superset/db_engine_specs/hive.py +++ b/superset/db_engine_specs/hive.py @@ -335,6 +335,10 @@ class HiveEngineSpec(PrestoEngineSpec): 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()