rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-01-17 10:22:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

pagebench: getpage: WIP: when auto-discovering timelines, add ability to limit

Browse Source
This commit is contained in:
Christian Schwarz
2023-12-13 14:05:50 +00:00
parent 14b5b6d72f
commit 7ac6f54ee9
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pageserver/pagebench/src/getpage_latest_lsn.rs
View File

@@ -36,6 +36,8 @@ pub(crate) struct Args {
runtime: Option<humantime::Duration>,
#[clap(long)]
per_target_rate_limit: Option<usize>,
#[clap(long)]
limit_to_first_n_targets: Option<usize>,
targets: Option<Vec<TenantTimelineId>>,
}
@@ -213,23 +215,26 @@ async fn main_impl(
async move {
(
tenant_id,
mgmt_api_client.list_timelines(tenant_id).await.unwrap(),
mgmt_api_client.tenant_details(tenant_id).await.unwrap(),
)
}
});
}
while let Some(res) = js.join_next().await {
let (tenant_id, tl_infos) = res.unwrap();
for tl in tl_infos {
let (tenant_id, details) = res.unwrap();
for timeline_id in details.timelines {
timelines.push(TenantTimelineId {
tenant_id,
timeline_id: tl.timeline_id,
timeline_id,
});
}
}
}
info!("timelines:\n{:?}", timelines);
info!("number of timelines:\n{:?}", timelines.len());
let mut js = JoinSet::new();
for timeline in &timelines {