Expose push_batch_size config through Builder

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

Author: Piotr Jastrzebski

libsql/src/database/builder.rs

@@ -107,6 +107,7 @@ impl Builder<()> {
                     connector: None,
                     read_your_writes: true,
                     remote_writes: false,
+                    push_batch_size: 0,
                 },
             }
         }
@@ -524,6 +525,7 @@ cfg_sync! {
         connector: Option<crate::util::ConnectorService>,
         remote_writes: bool,
         read_your_writes: bool,
+        push_batch_size: u32,
     }
 
     impl Builder<SyncedDatabase> {
@@ -543,6 +545,11 @@ cfg_sync! {
             self
         }
 
+        pub fn set_push_batch_size(mut self, v: u32) -> Builder<SyncedDatabase> {
+            self.inner.push_batch_size = v;
+            self
+        }
+
         /// Provide a custom http connector that will be used to create http connections.
         pub fn connector<C>(mut self, connector: C) -> Builder<SyncedDatabase>
         where
@@ -570,6 +577,7 @@ cfg_sync! {
                 connector,
                 remote_writes,
                 read_your_writes,
+                push_batch_size,
             } = self.inner;
 
             let path = path.to_str().ok_or(crate::Error::InvalidUTF8Path)?.to_owned();
@@ -596,6 +604,10 @@ cfg_sync! {
             )
             .await?;
 
+            if push_batch_size > 0 {
+                db.sync_ctx.as_ref().unwrap().lock().await.set_push_batch_size(push_batch_size);
+            }
+
             Ok(Database {
                 db_type: DbType::Offline {
                     db,

libsql/src/sync.rs

@@ -120,6 +120,10 @@ impl SyncContext {
         Ok(me)
     }
 
+    pub fn set_push_batch_size(&mut self, push_batch_size: u32) {
+        self.push_batch_size = push_batch_size;
+    }
+
     #[tracing::instrument(skip(self))]
     pub(crate) async fn pull_one_frame(
         &mut self,