C bindings: Add libsql_reset

Signed-off-by: Piotr Jastrzebski <[email protected]>

bindings/c/include/libsql.h

@@ -56,6 +56,8 @@ void libsql_close(libsql_database_t db);
 
 int libsql_connect(libsql_database_t db, libsql_connection_t *out_conn, const char **out_err_msg);
 
+int libsql_reset(libsql_connection_t conn, const char **out_err_msg);
+
 void libsql_disconnect(libsql_connection_t conn);
 
 int libsql_prepare(libsql_connection_t conn, const char *sql, libsql_stmt_t *out_stmt, const char **out_err_msg);

bindings/c/src/lib.rs

@@ -216,6 +216,20 @@ pub unsafe extern "C" fn libsql_connect(
     0
 }
 
+#[no_mangle]
+pub unsafe extern "C" fn libsql_reset(
+    conn: libsql_connection_t,
+    out_err_msg: *mut *const std::ffi::c_char,
+) -> std::ffi::c_int {
+    if conn.is_null() {
+        set_err_msg("Null connection".to_string(), out_err_msg);
+        return 1;
+    }
+    let conn = conn.get_ref();
+    RT.block_on(conn.reset());
+    0
+}
+
 #[no_mangle]
 pub unsafe extern "C" fn libsql_disconnect(conn: libsql_connection_t) {
     if conn.is_null() {

libsql/src/connection.rs

@@ -21,6 +21,8 @@ pub(crate) trait Conn {
     async fn changes(&self) -> u64;
 
     async fn last_insert_rowid(&self) -> i64;
+
+    async fn reset(&self);
 }
 
 /// A connection to some libsql database, this can be a remote one or a local one.
@@ -111,4 +113,8 @@ impl Connection {
     pub async fn last_insert_rowid(&self) -> i64 {
         self.conn.last_insert_rowid().await
     }
+
+    pub async fn reset(&self) {
+        self.conn.reset().await
+    }
 }

libsql/src/hrana/hyper.rs

@@ -157,6 +157,10 @@ impl Conn for HttpConnection<HttpSender> {
     async fn last_insert_rowid(&self) -> i64 {
         self.last_insert_rowid()
     }
+
+    async fn reset(&self) {
+        self.current_stream().reset().await;
+    }
 }
 
 #[async_trait::async_trait]
@@ -284,4 +288,8 @@ impl Conn for HranaStream<HttpSender> {
     async fn last_insert_rowid(&self) -> i64 {
         self.last_insert_rowid()
     }
+
+    async fn reset(&self) {
+        self.reset().await;
+    }
 }

libsql/src/hrana/stream.rs

@@ -266,6 +266,10 @@ where
     pub fn is_autocommit(&self) -> bool {
         self.inner.is_autocommit.load(Ordering::SeqCst)
     }
+
+    pub async fn reset(&self) {
+        (*self.inner).stream.lock().await.reset();
+    }
 }
 
 #[derive(Debug)]

libsql/src/local/impls.rs

@@ -59,6 +59,8 @@ impl Conn for LibsqlConnection {
     async fn last_insert_rowid(&self) -> i64 {
         self.conn.last_insert_rowid()
     }
+
+    async fn reset(&self) {}
 }
 
 impl Drop for LibsqlConnection {

libsql/src/replication/connection.rs

@@ -343,6 +343,8 @@ impl Conn for RemoteConnection {
     async fn last_insert_rowid(&self) -> i64 {
         self.inner.lock().last_insert_rowid
     }
+
+    async fn reset(&self) {}
 }
 
 pub struct ColumnMeta {