62 lines
1.9 KiB
Rust
62 lines
1.9 KiB
Rust
//! 10,000 task breakdown benchmarks - construction vs execution
|
|
|
|
use criterion::{BatchSize, Criterion};
|
|
use dagx::DagRunner;
|
|
use dagx_test::task_fn;
|
|
|
|
pub fn bench_10k_breakdown(c: &mut Criterion) {
|
|
let rt = tokio::runtime::Runtime::new().unwrap();
|
|
let mut group = c.benchmark_group("10k_tasks");
|
|
group.measurement_time(std::time::Duration::from_secs(10));
|
|
|
|
// a) Adding 10k tasks (construction only)
|
|
group.bench_function("construction", |b| {
|
|
b.iter(|| {
|
|
let mut dag = DagRunner::new();
|
|
for i in 0..10_000 {
|
|
dag.add_task(task_fn::<(), _, _>(move |_: ()| i));
|
|
}
|
|
});
|
|
});
|
|
|
|
// b) Full lifecycle: add 10k tasks + run them
|
|
group.bench_function("full_execution", |b| {
|
|
b.iter(|| {
|
|
rt.block_on(async {
|
|
let mut dag = DagRunner::new();
|
|
for i in 0..10_000 {
|
|
dag.add_task(task_fn::<(), _, _>(move |_: ()| i));
|
|
}
|
|
let _output = dag
|
|
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
|
.await
|
|
.unwrap();
|
|
})
|
|
});
|
|
});
|
|
|
|
// c) Execution only (pre-built DAG, just run)
|
|
group.bench_function("execution_only", |b| {
|
|
b.iter_batched(
|
|
|| {
|
|
let mut dag = DagRunner::new();
|
|
for i in 0..10_000 {
|
|
dag.add_task(task_fn::<(), _, _>(move |_: ()| i));
|
|
}
|
|
dag
|
|
},
|
|
|dag| {
|
|
rt.block_on(async {
|
|
let _output = dag
|
|
.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
|
.await
|
|
.unwrap();
|
|
})
|
|
},
|
|
BatchSize::LargeInput,
|
|
);
|
|
});
|
|
|
|
group.finish();
|
|
}
|