snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-26 12:05:05 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Add precommands to workloads

Browse Source
This commit is contained in:
Louis Dureuil 2024-05-07 12:23:36 +02:00
parent ba75d23bfe
commit c9a0b1a201
No known key found for this signature in database
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
xtask/src/bench/workload.rs
View File

@ -22,6 +22,8 @@ pub struct Workload {
pub run_count: u16, pub run_count: u16,
pub extra_cli_args: Vec<String>, pub extra_cli_args: Vec<String>,
pub assets: BTreeMap<String, Asset>, pub assets: BTreeMap<String, Asset>,
#[serde(default)]
pub precommands: Vec<super::command::Command>,
pub commands: Vec<super::command::Command>, pub commands: Vec<super::command::Command>,
} }
@ -37,6 +39,15 @@ async fn run_commands(
let report_folder = &args.report_folder; let report_folder = &args.report_folder;
let workload_name = &workload.name; let workload_name = &workload.name;
for batch in workload
.precommands
.as_slice()
.split_inclusive(|command| !matches!(command.synchronous, SyncMode::DontWait))
{
super::command::run_batch(meili_client, batch, &workload.assets, &args.asset_folder)
.await?;
}
std::fs::create_dir_all(report_folder) std::fs::create_dir_all(report_folder)
.with_context(|| format!("could not create report directory at {report_folder}"))?; .with_context(|| format!("could not create report directory at {report_folder}"))?;