129 lines
3.9 KiB
Rust
129 lines
3.9 KiB
Rust
use std::future::Future;
|
|
|
|
use async_executor::Executor;
|
|
use criterion::{criterion_group, criterion_main, Criterion};
|
|
use futures_lite::{future, prelude::*};
|
|
|
|
const TASKS: usize = 300;
|
|
const STEPS: usize = 300;
|
|
const LIGHT_TASKS: usize = 25_000;
|
|
|
|
static EX: Executor<'_> = Executor::new();
|
|
|
|
fn run(f: impl FnOnce(), multithread: bool) {
|
|
let limit = if multithread { num_cpus::get() } else { 1 };
|
|
|
|
let (s, r) = async_channel::bounded::<()>(1);
|
|
easy_parallel::Parallel::new()
|
|
.each(0..limit, |_| future::block_on(EX.run(r.recv())))
|
|
.finish(move || {
|
|
let _s = s;
|
|
f()
|
|
});
|
|
}
|
|
|
|
fn create(c: &mut Criterion) {
|
|
c.bench_function("executor::create", |b| {
|
|
b.iter(|| {
|
|
let ex = Executor::new();
|
|
let task = ex.spawn(async {});
|
|
future::block_on(ex.run(task));
|
|
})
|
|
});
|
|
}
|
|
|
|
fn running_benches(c: &mut Criterion) {
|
|
for (group_name, multithread) in [("single_thread", false), ("multi_thread", true)].iter() {
|
|
let mut group = c.benchmark_group(group_name.to_string());
|
|
|
|
group.bench_function("executor::spawn_one", |b| {
|
|
run(
|
|
|| {
|
|
b.iter(|| {
|
|
future::block_on(async { EX.spawn(async {}).await });
|
|
});
|
|
},
|
|
*multithread,
|
|
);
|
|
});
|
|
|
|
group.bench_function("executor::spawn_many_local", |b| {
|
|
run(
|
|
|| {
|
|
b.iter(move || {
|
|
future::block_on(async {
|
|
let mut tasks = Vec::new();
|
|
for _ in 0..LIGHT_TASKS {
|
|
tasks.push(EX.spawn(async {}));
|
|
}
|
|
for task in tasks {
|
|
task.await;
|
|
}
|
|
});
|
|
});
|
|
},
|
|
*multithread,
|
|
);
|
|
});
|
|
|
|
group.bench_function("executor::spawn_recursively", |b| {
|
|
#[allow(clippy::manual_async_fn)]
|
|
fn go(i: usize) -> impl Future<Output = ()> + Send + 'static {
|
|
async move {
|
|
if i != 0 {
|
|
EX.spawn(async move {
|
|
let fut = go(i - 1).boxed();
|
|
fut.await;
|
|
})
|
|
.await;
|
|
}
|
|
}
|
|
}
|
|
|
|
run(
|
|
|| {
|
|
b.iter(move || {
|
|
future::block_on(async {
|
|
let mut tasks = Vec::new();
|
|
for _ in 0..TASKS {
|
|
tasks.push(EX.spawn(go(STEPS)));
|
|
}
|
|
for task in tasks {
|
|
task.await;
|
|
}
|
|
});
|
|
});
|
|
},
|
|
*multithread,
|
|
);
|
|
});
|
|
|
|
group.bench_function("executor::yield_now", |b| {
|
|
run(
|
|
|| {
|
|
b.iter(move || {
|
|
future::block_on(async {
|
|
let mut tasks = Vec::new();
|
|
for _ in 0..TASKS {
|
|
tasks.push(EX.spawn(async move {
|
|
for _ in 0..STEPS {
|
|
future::yield_now().await;
|
|
}
|
|
}));
|
|
}
|
|
for task in tasks {
|
|
task.await;
|
|
}
|
|
});
|
|
});
|
|
},
|
|
*multithread,
|
|
);
|
|
});
|
|
}
|
|
}
|
|
|
|
criterion_group!(benches, create, running_benches);
|
|
|
|
criterion_main!(benches);
|