set msg attribute names to all

worldcoin · Sep 9, 2024 · dbac940 · dbac940
1 parent f64e8d0
commit dbac940
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/iris-mpc/src/bin/client.rs b/iris-mpc/src/bin/client.rs
@@ -139,6 +139,7 @@ async fn main() -> eyre::Result<()> {
             // Receive responses
             let msg = results_sqs_client
                 .receive_message()
+                .message_attribute_names("All")
                 .max_number_of_messages(1)
                 .queue_url(response_queue_url.clone())
                 .send()

diff --git a/iris-mpc/src/bin/server.rs b/iris-mpc/src/bin/server.rs
@@ -124,10 +124,12 @@ async fn receive_batch(
     let mut handles = vec![];
     let mut msg_counter = 0;
 
+    tracing::info!("Setting message attribute names to All in receive_batch");
     while msg_counter < *CURRENT_BATCH_SIZE.lock().unwrap() {
         let rcv_message_output = client
             .receive_message()
             .message_attribute_names("All")
+            .set_message_attribute_names(Some(vec!["All".to_string()]))
             .max_number_of_messages(1)
             .queue_url(queue_url)
             .send()