stop query on the processing engine when user clicks STOP on SQL Lab (#18789)

Co-authored-by: Saurabh Gulati <saurabh.gulati@fedex.com>
This commit is contained in:
saurabh gulati 2022-02-22 08:19:34 +01:00 committed by GitHub
parent 535634310b
commit d632590eff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions

View File

@ -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()