Make sure sending cancel request in StatementClient.close()

treasure-data · Dec 7, 2024 · cd74922 · cd74922
1 parent 55c3bdc
commit cd74922
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/trino/client/statement_client.rb b/lib/trino/client/statement_client.rb
@@ -262,9 +262,11 @@ def cancel_leaf_stage
     end
 
     def close
-      return unless running?
-
-      @state = :client_aborted
+      return if finished? || query_failed? || client_aborted?
+
+      if running?
+        @state = :client_aborted
+      end
 
       begin
         if uri = @results.next_uri