improve logging (#536)

* improve logging

* use u64 consistently for batch sizes

iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs

@@ -238,9 +238,9 @@ async fn send_batch_and_wait_for_ack(
     let batch_size = batch.len();
     // Send the batch size to all servers
     let (batch_size_result_a, batch_size_result_b, batch_size_result_c) = (
-        server1.write_u8(batch_size as u8),
-        server2.write_u8(batch_size as u8),
-        server3.write_u8(batch_size as u8),
+        server1.write_u64(batch_size as u64),
+        server2.write_u64(batch_size as u64),
+        server3.write_u64(batch_size as u64),
     )
         .join()
         .await;

iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs

@@ -128,8 +128,8 @@ async fn main() -> eyre::Result<()> {
     for batch_num in 0..num_batches + 1 {
         tracing::info!("Processing batch {} of size: {}", batch_num, batch_size1);
         let start_time = Instant::now();
-        let batch_size_1_message = client_stream1.read_u8().await?;
-        let batch_size_2_message = client_stream2.read_u8().await?;
+        let batch_size_1_message = client_stream1.read_u64().await?;
+        let batch_size_2_message = client_stream2.read_u64().await?;
 
         if batch_size_1_message != batch_size_2_message {
             bail!(

iris-mpc-upgrade/src/lib.rs

@@ -143,14 +143,14 @@ impl<S: NewIrisShareSink> IrisCodeUpgrader<S> {
             *a = a.wrapping_add(*b);
         }
         let duration = start_time.elapsed();
-        tracing::info!("Computed iris codes STEP DURATION: {:.2?}", duration);
+        tracing::debug!("Computed iris codes STEP DURATION: {:.2?}", duration);
 
         let start_time = std::time::Instant::now();
         self.iris_sink
             .store_code_mask_share(id, &result, &mask)
             .await?;
         let duration = start_time.elapsed();
-        tracing::info!("Stored iris codes STEP DURATION: {:.2?}", duration);
+        tracing::debug!("Stored iris codes STEP DURATION: {:.2?}", duration);
         Ok(())
     }
 }