From db3333eecb3257e4f059f63a913d404e9c6a1e84 Mon Sep 17 00:00:00 2001
From: Christian Schwarz <christian@neon.tech>
Date: Thu, 21 Mar 2024 10:51:42 +0000
Subject: [PATCH] yield after ever redo execution

---
 pageserver/benches/bench_walredo.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pageserver/benches/bench_walredo.rs b/pageserver/benches/bench_walredo.rs
index 68902b991b..e98457a6ff 100644
--- a/pageserver/benches/bench_walredo.rs
+++ b/pageserver/benches/bench_walredo.rs
@@ -145,6 +145,9 @@ async fn client(
     for _ in 0..n_redos {
         let page = redo_work.execute(&mgr).await.unwrap();
         assert_eq!(page.remaining(), 8192);
+        // The real pageserver will rarely if ever do 2 walredos in a row without
+        // yielding to the executor.
+        tokio::task::yield_now().await;
     }
     start.elapsed()
 }