Skip to content

Fix infinite hang when Dremio job is cancelled (CANCELED vs CANCELLED) #322

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
shanmugapriya5c wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Fix infinite hang when Dremio job is cancelled (CANCELED vs CANCELLED) #322

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
fd66a6a
Fixed job status for CANCELED
shanmugapriya5c Jun 10, 2026
400f1fe
Fix infinite hang on cancelled Dremio jobs by handling the CANCELED j…
shanmugapriya5c Jun 10, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions CHANGELOG.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
- Fixed CI MinIO client installation by following download redirects.
- Suppressed Dremio Cloud folder-creation errors when a folder already exists.
- Avoided creating existing folders during schema setup.
- Fix infinite hang on cancelled Dremio jobs by handling the CANCELED job state (single L)

# dbt-dremio v1.10.0

Expand Down
11 changes: 9 additions & 2 deletions dbt/adapters/dremio/api/cursor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ def _initialize(self):
def _populate_rowcount(self):
if self.closed:
raise Exception("CursorClosed")
# keep checking job status until status is one of COMPLETE, CANCELLED or FAILED
# keep checking job status until status is one of COMPLETED, CANCELED or FAILED
# map job results to AdapterResponse
job_id = self._job_id

Expand All @@ -132,7 +132,14 @@ def _populate_rowcount(self):
error_message = job_status_response["errorMessage"]
raise Exception(f"ERROR: {error_message}")

if job_status_state == "CANCELLED":
if job_status_state in ("CANCELED", "CANCELLED"):
cancellation_reason = job_status_response.get(
"cancellationReason"
)
if cancellation_reason:
raise Exception(
f"Job was cancelled: {cancellation_reason}"
)
raise Exception("Job was cancelled")

if job_status_state == "COMPLETED":
Expand Down