1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

use criterion::BenchmarkId;
use criterion::Criterion;
use criterion::{criterion_group, criterion_main};

// 此结构体告诉 Criterion.rs 使用 "futures" crate 的 current-thread 执行器
use criterion::async_executor::FuturesExecutor;

// 待基准测试的异步函数
async fn do_something(size: usize) {
    // 用 size 执行某些异步操作
}

fn from_elem(c: &mut Criterion) {
    let size: usize = 1024;

    c.bench_with_input(BenchmarkId::new("input_example", size), &size, |b, &s| {
        // 调用 `to_async` 将 bencher 转换为异步模式。
        // 计时循环与常规 bencher 相同。
        b.to_async(FuturesExecutor).iter(|| do_something(s));
    });
}

criterion_group!(benches, from_elem);
criterion_main!(benches);