Add cancel to other functions

Signed-off-by: Bagus Nurtomo <bagus.nurtomo@pm.me>
starburstdata · Jan 16, 2025 · b49c048 · b49c048
1 parent 9c0f9ba
commit b49c048
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/drivers/starburst/src/metabase/driver/implementation/execute.clj b/drivers/starburst/src/metabase/driver/implementation/execute.clj
@@ -250,7 +250,8 @@
           rs)
           (catch Throwable e (handle-execution-error e))))
     (setMaxRows [nb] (.setMaxRows stmt nb))
-    (close [] (.close stmt))))
+    (close [] (.close stmt))
+    (cancel [] (.cancel stmt))))
 
 (defmethod sql-jdbc.execute/prepared-statement :starburst
   [driver ^Connection conn ^String sql params]
@@ -295,7 +296,8 @@
             (finally (remove-impersonation conn))))
       (getResultSet [] (.getResultSet stmt))
       (setMaxRows [nb] (.setMaxRows stmt nb))
-      (close [] (.close stmt)))))
+      (close [] (.close stmt))
+      (cancel [] (.cancel stmt)))))
 
 ;;; +----------------------------------------------------------------------------------------------------------------+
 ;;; |                                          Prepared Statement Substitutions                                      |