feat: expose set/get reserved bytes to public API

Author: levydsa

libsql/src/connection.rs

@@ -39,6 +39,14 @@ pub(crate) trait Conn {
 
     async fn reset(&self);
 
+    fn set_reserved_bytes(&self, _reserved_bytes: i32) -> Result<()> {
+        Err(crate::Error::ReservedBytesNotSupported)
+    }
+
+    fn get_reserved_bytes(&self) -> Result<i32> {
+        Err(crate::Error::ReservedBytesNotSupported)
+    }
+
     fn enable_load_extension(&self, _onoff: bool) -> Result<()> {
         Err(crate::Error::LoadExtensionNotSupported)
     }
@@ -230,6 +238,14 @@ impl Connection {
         self.conn.reset().await
     }
 
+    pub fn set_reserved_bytes(&self, reserved_bytes: i32) -> Result<()> {
+        self.conn.set_reserved_bytes(reserved_bytes)
+    }
+
+    pub fn get_reserved_bytes(&self) -> Result<i32> {
+        self.conn.get_reserved_bytes()
+    }
+
     /// Enable loading SQLite extensions from SQL queries and Rust API.
     ///
     /// See [`load_extension`](Connection::load_extension) documentation for more details.

libsql/src/errors.rs

@@ -61,6 +61,8 @@ pub enum Error {
     Sync(crate::BoxError),
     #[error("WAL frame insert conflict")]
     WalConflict,
+    #[error("Reserved bytes not supported")]
+    ReservedBytesNotSupported,
 }
 
 #[cfg(feature = "hrana")]

libsql/src/local/impls.rs

@@ -81,6 +81,14 @@ impl Conn for LibsqlConnection {
 
     async fn reset(&self) {}
 
+    fn set_reserved_bytes(&self, reserved_bytes: i32) -> Result<()> {
+        self.conn.set_reserved_bytes(reserved_bytes)
+    }
+
+    fn get_reserved_bytes(&self) -> Result<i32> {
+        self.conn.get_reserved_bytes()
+    }
+
     fn enable_load_extension(&self, onoff: bool) -> Result<()> {
         self.conn.enable_load_extension(onoff)
     }

libsql/src/replication/connection.rs

@@ -542,6 +542,14 @@ impl Conn for RemoteConnection {
     }
 
     async fn reset(&self) {}
+
+    fn set_reserved_bytes(&self, reserved_bytes: i32) -> Result<()> {
+        self.local.set_reserved_bytes(reserved_bytes)
+    }
+
+    fn get_reserved_bytes(&self) -> Result<i32> {
+        self.local.get_reserved_bytes()
+    }
 }
 
 pub struct ColumnMeta {

libsql/src/sync/connection.rs

@@ -189,4 +189,12 @@ impl Conn for SyncedConnection {
     }
 
     async fn reset(&self) {}
+
+    fn set_reserved_bytes(&self, reserved_bytes: i32) -> Result<()> {
+        self.local.set_reserved_bytes(reserved_bytes)
+    }
+
+    fn get_reserved_bytes(&self) -> Result<i32> {
+        self.local.get_reserved_bytes()
+    }
 }