bench: improve polling

geofmureithi · Jun 26, 2024 · b0af55c · b0af55c
1 parent 7bbde7f
commit b0af55c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benches/storages.rs b/benches/storages.rs
@@ -51,7 +51,7 @@ macro_rules! define_bench {
                             for _i in 0..s {
                                 let _ = s1.push(TestJob).await;
                             }
-                            while s1.len().await.unwrap_or(-1) != 0 {
+                            while counter.0.load(Ordering::Relaxed) != s && s1.len().await.unwrap_or(-1) != 0 {
                                 interval.tick().await;
                             }
                             counter.0.store(0, Ordering::Relaxed);