From 2d6763882e65d87aed223853024592355d5d8d72 Mon Sep 17 00:00:00 2001 From: Christian Schwarz Date: Thu, 12 Sep 2024 10:30:33 +0000 Subject: [PATCH] pagebench: fake queue depth of 10 --- pageserver/client/src/page_service.rs | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/pageserver/client/src/page_service.rs b/pageserver/client/src/page_service.rs index f9507fc47a..9701ca6ff9 100644 --- a/pageserver/client/src/page_service.rs +++ b/pageserver/client/src/page_service.rs @@ -142,11 +142,16 @@ impl PagestreamClient { ) -> anyhow::Result { let req = PagestreamFeMessage::GetPage(req); let req: bytes::Bytes = req.serialize(); - // let mut req = tokio_util::io::ReaderStream::new(&req); - let mut req = tokio_stream::once(Ok(req)); - self.copy_both.send_all(&mut req).await?; + for i in 0..10 { + let mut req = tokio_stream::once(Ok(req.clone())); + self.copy_both.send_all(&mut req).await?; + } + for i in 0..9 { + let next: Option> = self.copy_both.next().await; + let next: bytes::Bytes = next.unwrap()?; + } let next: Option> = self.copy_both.next().await; let next: bytes::Bytes = next.unwrap()?;