From 61c20e08bdb81701be3444dedf6a2f86d827da31 Mon Sep 17 00:00:00 2001
From: Alex Koshelev <koshelev@meta.com>
Date: Fri, 14 Jun 2024 18:50:14 -0700
Subject: [PATCH] Log bytes on the sender side and receiver side

---
 ipa-core/src/query/runner/oprf_ipa.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ipa-core/src/query/runner/oprf_ipa.rs b/ipa-core/src/query/runner/oprf_ipa.rs
index e74c8b166..06763f7cd 100644
--- a/ipa-core/src/query/runner/oprf_ipa.rs
+++ b/ipa-core/src/query/runner/oprf_ipa.rs
@@ -70,12 +70,14 @@ where
         let sz = usize::from(query_size);
 
         let input = if config.plaintext_match_keys {
+            tracing::info!("starting polling the plaintext stream to get {sz} items");
             let mut v = RecordsStream::<OPRFIPAInputRow<BA8, BA3, BA20>, _>::new(input_stream)
                 .try_concat()
                 .await?;
             v.truncate(sz);
             v
         } else {
+            tracing::info!("starting polling the encrypted stream to get {query_size:?} items");
             LengthDelimitedStream::<EncryptedOprfReport<BA8, BA3, BA20, _>, _>::new(input_stream)
                 .map_err(Into::<Error>::into)
                 .map_ok(|enc_reports| {
@@ -111,6 +113,7 @@ where
                 .await?
         };
 
+        tracing::info!("ready for OPRF ipa, got {} items", input.len());
         let aws = config.attribution_window_seconds;
         match config.per_user_credit_cap {
             8 => oprf_ipa::<BA8, BA3, HV, BA20, 3, 256>(ctx, input, aws).await,