ion-sound/benches/statement.rs

use criterion::*;
use ion_shell::parser::{StatementSplitter, Terminator};
use itertools::Itertools;

const TEXT: &[u8] = include_bytes!("test.ion");

fn criterion_benchmark(c: &mut Criterion) {
    let stmts = TEXT
        .iter()
        .copied()
        .batching(|lines| Terminator::new(lines).terminate())
        .collect::<Vec<_>>();
    c.bench(
        "statement-splitter-throughput",
        ParameterizedBenchmark::new(
            "statement",
            |b, script| {
                b.iter(|| {
                    script.iter().flat_map(|cmd| StatementSplitter::new(cmd)).collect::<Vec<_>>()
                })
            },
            vec![stmts],
        )
        .throughput(|script| Throughput::Bytes(script.len() as u64)),
    );
}

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