libsql: Add Statement::interrupt()

Author: penberg

libsql-sys/src/statement.rs

@@ -82,6 +82,10 @@ impl Statement {
         unsafe { crate::ffi::sqlite3_step(self.raw_stmt) }
     }
 
+    pub fn interrupt(&self) {
+        unsafe { crate::ffi::libsql_stmt_interrupt(self.raw_stmt) }
+    }
+
     pub fn reset(&self) -> std::ffi::c_int {
         unsafe { crate::ffi::sqlite3_reset(self.raw_stmt) }
     }

libsql/src/hrana/hyper.rs

@@ -205,6 +205,12 @@ impl crate::statement::Stmt for crate::hrana::Statement<HttpSender> {
         self.run(params).await
     }
 
+    fn interrupt(&mut self) -> crate::Result<()> {
+        Err(crate::Error::Misuse(
+            "interrupt is not supported for remote connections".to_string(),
+        ))
+    }
+
     fn reset(&mut self) {}
 
     fn parameter_count(&self) -> usize {

libsql/src/local/impls.rs

@@ -120,6 +120,10 @@ impl Stmt for LibsqlStmt {
         stmt.run(&params)
     }
 
+    fn interrupt(&mut self) -> Result<()> {
+        self.0.interrupt()
+    }
+
     fn reset(&mut self) {
         self.0.reset();
     }

libsql/src/local/statement.rs

@@ -134,6 +134,12 @@ impl Statement {
         }
     }
 
+    /// Interrupt the statement.
+    pub fn interrupt(&self) -> Result<()> {
+        self.inner.interrupt();
+        Ok(())
+    }
+
     /// Reset the prepared statement to initial state for reuse.
     pub fn reset(&self) {
         self.inner.reset();

libsql/src/replication/connection.rs

@@ -732,6 +732,12 @@ impl Stmt for RemoteStatement {
         Ok(())
     }
 
+    fn interrupt(&mut self) -> Result<()> {
+        Err(Error::Misuse(
+            "interrupt is not supported for remote connections".to_string(),
+        ))
+    }
+
     fn reset(&mut self) {}
 
     fn parameter_count(&self) -> usize {

libsql/src/statement.rs

@@ -14,6 +14,8 @@ pub(crate) trait Stmt {
 
     async fn run(&mut self, params: &Params) -> Result<()>;
 
+    fn interrupt(&mut self) -> Result<()>;
+
     fn reset(&mut self);
 
     fn parameter_count(&self) -> usize;
@@ -60,6 +62,11 @@ impl Statement {
         Ok(())
     }
 
+    /// Interrupt the statement.
+    pub fn interrupt(&mut self) -> Result<()> {
+        self.inner.interrupt()
+    }
+
     /// Execute a query that returns the first [`Row`].
     ///
     /// # Errors

libsql/src/sync/statement.rs

@@ -40,6 +40,10 @@ impl Stmt for SyncedStatement {
         result
     }
 
+    fn interrupt(&mut self) -> Result<()> {
+        self.inner.interrupt()
+    }
+
     fn reset(&mut self) {
         self.inner.reset()
     }