From e615a9f4a8ab7b923980937262e35435934862ff Mon Sep 17 00:00:00 2001
From: Ertugrul Aypek <ertugrul.aypek@toolsforhumanity.com>
Date: Thu, 26 Sep 2024 10:28:27 +0200
Subject: [PATCH] more debug logs

---
 iris-mpc-gpu/src/server/actor.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/iris-mpc-gpu/src/server/actor.rs b/iris-mpc-gpu/src/server/actor.rs
index 64195283f..6d499ac84 100644
--- a/iris-mpc-gpu/src/server/actor.rs
+++ b/iris-mpc-gpu/src/server/actor.rs
@@ -55,6 +55,7 @@ impl ServerActorHandle {
         &mut self,
         batch: BatchQuery,
     ) -> impl Future<Output = ServerJobResult> {
+        tracing::info!("Submitting batch query to the Actor job queue");
         let (tx, rx) = oneshot::channel();
         let job = ServerJob {
             batch,
@@ -449,6 +450,7 @@ impl ServerActor {
             .preprocess_db(&mut self.right_mask_db_slices, &self.current_db_sizes);
     }
 
+    #[tracing::instrument(skip(self, return_channel))]
     fn process_batch_query(
         &mut self,
         batch: BatchQuery,
@@ -877,6 +879,7 @@ impl ServerActor {
         Ok(())
     }
 
+    #[tracing::instrument(skip(self, compact_device_queries, compact_device_sums, events, eye_db))]
     fn compare_query_against_db_and_self(
         &mut self,
         compact_device_queries: &DeviceCompactQuery,
@@ -1160,6 +1163,7 @@ impl ServerActor {
         }
     }
 
+    #[tracing::instrument(skip(self, valid_entries))]
     fn sync_batch_entries(&mut self, valid_entries: &[bool]) -> eyre::Result<Vec<bool>> {
         let mut buffer = self
             .device_manager
@@ -1193,6 +1197,7 @@ impl ServerActor {
         Ok(valid_merged)
     }
 
+    #[tracing::instrument(skip(self))]
     fn prepare_deletion_shares(&self) -> eyre::Result<(DeviceCompactQuery, DeviceCompactSums)> {
         let (dummy_code_share, dummy_mask_share) = get_dummy_shares_for_deletion(self.party_id);
         let compact_query = {