Make sure sending cancel request in StatementClient.close()

treasure-data · Dec 6, 2024 · 402ee02 · 402ee02
1 parent 55c3bdc
commit 402ee02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/trino/client/statement_client.rb b/lib/trino/client/statement_client.rb
@@ -262,7 +262,7 @@ def cancel_leaf_stage
     end
 
     def close
-      return unless running?
+      return if finished? || query_failed?
 
       @state = :client_aborted