u8 => u64

worldcoin · Oct 15, 2024 · 225b6a2 · 225b6a2
1 parent 6bfdaa5
commit 225b6a2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs b/iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs
@@ -92,9 +92,9 @@ async fn main() -> eyre::Result<()> {
     server2.write_u64(end).await?;
     server3.write_u64(end).await?;
 
-    server1.write_u8(args.batch_size).await?;
-    server2.write_u8(args.batch_size).await?;
-    server3.write_u8(args.batch_size).await?;
+    server1.write_u64(args.batch_size).await?;
+    server2.write_u64(args.batch_size).await?;
+    server3.write_u64(args.batch_size).await?;
 
     server1.flush().await?;
     server2.flush().await?;

diff --git a/iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs b/iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs
@@ -108,8 +108,8 @@ async fn main() -> eyre::Result<()> {
         );
     }
 
-    let batch_size1 = client_stream1.read_u8().await?;
-    let batch_size2 = client_stream2.read_u8().await?;
+    let batch_size1 = client_stream1.read_u64().await?;
+    let batch_size2 = client_stream2.read_u64().await?;
 
     if batch_size1 != batch_size2 {
         bail!(
@@ -120,7 +120,7 @@ async fn main() -> eyre::Result<()> {
     }
 
     let num_elements = end1.checked_sub(start1).unwrap();
-    let num_batches = num_elements / u64::from(batch_size1);
+    let num_batches = num_elements / batch_size1;
     tracing::info!("Batch size: {}, num batches: {}", batch_size1, num_batches);
 
     let mut batch = Vec::new();

diff --git a/iris-mpc-upgrade/src/config.rs b/iris-mpc-upgrade/src/config.rs
@@ -88,7 +88,7 @@ pub struct UpgradeClientConfig {
     pub party_id: u8,
 
     #[clap(long)]
-    pub batch_size: u8,
+    pub batch_size: u64,
 
     #[clap(long)]
     pub eye: Eye,