libsql: Add Connection::busy_timeout() API

Author: penberg

libsql/src/connection.rs

@@ -2,6 +2,7 @@ use std::collections::VecDeque;
 use std::fmt;
 use std::path::Path;
 use std::sync::Arc;
+use std::time::Duration;
 
 use crate::params::{IntoParams, Params};
 use crate::rows::Rows;
@@ -23,6 +24,8 @@ pub(crate) trait Conn {
 
     fn interrupt(&self) -> Result<()>;
 
+    fn busy_timeout(&self, timeout: Duration) -> Result<()>;
+
     fn is_autocommit(&self) -> bool;
 
     fn changes(&self) -> u64;
@@ -192,6 +195,10 @@ impl Connection {
         self.conn.interrupt()
     }
 
+    pub fn busy_timeout(&self, timeout: Duration) -> Result<()> {
+        self.conn.busy_timeout(timeout)
+    }
+
     /// Check weather libsql is in `autocommit` or not.
     pub fn is_autocommit(&self) -> bool {
         self.conn.is_autocommit()

libsql/src/hrana/hyper.rs

@@ -16,6 +16,7 @@ use http::{HeaderValue, StatusCode};
 use hyper::body::HttpBody;
 use std::io::ErrorKind;
 use std::sync::Arc;
+use std::time::Duration;
 
 use super::StmtResultRows;
 
@@ -168,6 +169,11 @@ impl Conn for HttpConnection<HttpSender> {
         Ok(())
     }
 
+    fn busy_timeout(&self, _timeout: Duration) -> crate::Result<()> {
+        // Busy timeout is a no-op for remote connections.
+        Ok(())
+    }
+
     fn is_autocommit(&self) -> bool {
         self.is_autocommit()
     }
@@ -362,6 +368,11 @@ impl Conn for HranaStream<HttpSender> {
         Ok(())
     }
 
+    fn busy_timeout(&self, _timeout: Duration) -> crate::Result<()> {
+        // Busy timeout is a no-op for remote connections.
+        Ok(())
+    }
+
     fn is_autocommit(&self) -> bool {
         false // for streams this method is callable only when we're within explicit transaction
     }

libsql/src/local/connection.rs

@@ -3,6 +3,7 @@
 use crate::local::rows::BatchedRows;
 use crate::params::Params;
 use crate::{connection::BatchRows, errors};
+use std::time::Duration;
 
 use super::{Database, Error, Result, Rows, RowsFuture, Statement, Transaction};
 
@@ -361,6 +362,11 @@ impl Connection {
         Ok(())
     }
 
+    pub fn busy_timeout(&self, timeout: Duration) -> Result<()> {
+        unsafe { ffi::sqlite3_busy_timeout(self.raw, timeout.as_millis() as i32) };
+        Ok(())
+    }
+
     pub fn is_autocommit(&self) -> bool {
         unsafe { ffi::sqlite3_get_autocommit(self.raw) != 0 }
     }

libsql/src/local/impls.rs

@@ -1,5 +1,6 @@
 use std::sync::Arc;
 use std::{fmt, path::Path};
+use std::time::Duration;
 
 use crate::connection::BatchRows;
 use crate::{
@@ -58,6 +59,10 @@ impl Conn for LibsqlConnection {
         self.conn.interrupt()
     }
 
+    fn busy_timeout(&self, timeout: Duration) -> Result<()> {
+        self.conn.busy_timeout(timeout)
+    }
+
     fn is_autocommit(&self) -> bool {
         self.conn.is_autocommit()
     }

libsql/src/replication/connection.rs

@@ -1,5 +1,6 @@
 // TODO(lucio): Move this to `remote/mod.rs`
 
+use std::time::Duration;
 use std::str::FromStr;
 use std::sync::Arc;
 use std::sync::atomic::AtomicU64;
@@ -508,6 +509,11 @@ impl Conn for RemoteConnection {
         Ok(())
     }
 
+    fn busy_timeout(&self, _timeout: Duration) -> Result<()> {
+        // Busy timeout is a no-op for remote connections.
+        Ok(())
+    }
+
     fn is_autocommit(&self) -> bool {
         self.is_state_init()
     }

libsql/src/sync/connection.rs

@@ -8,6 +8,7 @@ use crate::{
     BatchRows, Error, Result, Statement, Transaction, TransactionBehavior,
 };
 use std::sync::Arc;
+use std::time::Duration;
 use tokio::sync::Mutex;
 
 use super::{statement::SyncedStatement, transaction::SyncedTx};
@@ -100,6 +101,10 @@ impl Conn for SyncedConnection {
         Ok(())
     }
 
+    fn busy_timeout(&self, timeout: Duration) -> Result<()> {
+        self.remote.busy_timeout(timeout)
+    }
+
     fn is_autocommit(&self) -> bool {
         self.remote.is_autocommit()
     }