use crate::common::{bootstrap_graph, run_large_ingestion_benchmarks};
use criterion::{
    criterion_group, criterion_main, AxisScale, Criterion, PlotConfiguration, Throughput,
};

mod common;

pub fn parameterized(c: &mut Criterion) {
    let vertices_exponents = 1..6;

    let vertices = vertices_exponents.map(|exp| 10usize.pow(exp));
    let mut ingestion_group = c.benchmark_group("ingestion-num_vertices");
    ingestion_group.plot_config(PlotConfiguration::default().summary_scale(AxisScale::Logarithmic));
    for num_vertices in vertices {
        let make_graph = || bootstrap_graph(4, num_vertices);
        ingestion_group.throughput(Throughput::Elements(num_vertices as u64));
        ingestion_group.sample_size(10);
        ingestion_group.warm_up_time(std::time::Duration::from_secs(1));
        run_large_ingestion_benchmarks(&mut ingestion_group, make_graph, Some(num_vertices));
    }
    ingestion_group.finish();
}

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