rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-06-01 20:40:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

hack for testing asyncification: repurpose pagebench initial logical size trigger benchmark to induce unbounded concurrency

Browse Source
This commit is contained in:
Christian Schwarz
2024-01-11 11:31:38 +00:00
parent e3fa914252
commit 9ebb922832
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pageserver/pagebench/src/cmd/trigger_initial_size_calculation.rs
View File

@@ -58,10 +58,21 @@ async fn main_impl(args: Args) -> anyhow::Result<()> {
for tl in timelines {
let mgmt_api_client = Arc::clone(&mgmt_api_client);
js.spawn(async move {
let info = mgmt_api_client
.timeline_info(tl.tenant_id, tl.timeline_id, ForceAwaitLogicalSize::Yes)
.await
.unwrap();
// detach-attach to attach with emptied initial logical size cache
mgmt_api_client.tenant_reset(tl.tenant_id).await.unwrap();
let info = loop {
let res = mgmt_api_client
.timeline_info(tl.tenant_id, tl.timeline_id, ForceAwaitLogicalSize::Yes)
.await;
match res {
Ok(info) => break info,
Err(_) => {
tokio::time::sleep(std::time::Duration::from_secs(1).into()).await;
}
}
};
// Polling should not be strictly required here since we await
// for the initial logical size, however it's possible for the request