fix: no auth subscription must only wait for the expected bundle ID.

jito-labs · Nov 18, 2024 · 0e3e862 · 0e3e862
1 parent 51fb639
commit 0e3e862
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/searcher_client/src/lib.rs b/searcher_client/src/lib.rs
@@ -1,7 +1,4 @@
-use std::{
-    sync::Arc,
-    time::{Duration, Instant},
-};
+use std::{future, sync::Arc, time::{Duration, Instant}};
 
 use futures_util::StreamExt;
 use jito_protos::{
@@ -120,7 +117,9 @@ where
     let mut time_left = 5000;
     while let Ok(Some(Ok(results))) = timeout(
         Duration::from_millis(time_left),
-        bundle_results_subscription.next(),
+        bundle_results_subscription
+            .filter(|bundle_result| future::ready(matches!(bundle_result, Ok(bundle) if bundle.bundle_id == uuid)))
+            .next(),
     )
     .await
     {