use std::time::Duration;

use heed::EnvOpenOptions;
use milli::Index;
use criterion::{criterion_group, criterion_main, BenchmarkId};

fn bench_search(c: &mut criterion::Criterion) {
    let database = "books-4cpu.mmdb";
    let queries = [
        "minogue kylie",
        "minogue kylie live",
    ];

    std::fs::create_dir_all(database).unwrap();
    let env = EnvOpenOptions::new()
        .map_size(100 * 1024 * 1024 * 1024) // 100 GB
        .max_readers(10)
        .max_dbs(5)
        .open(database).unwrap();

    let index = Index::new(&env).unwrap();

    let mut group = c.benchmark_group("search");
    group.sample_size(10);
    group.measurement_time(Duration::from_secs(12));

    for query in &queries {
        group.bench_with_input(BenchmarkId::from_parameter(query), &query, |b, &query| {
            b.iter(|| {
                let rtxn = env.read_txn().unwrap();
                let _documents_ids = index.search(&rtxn).query(*query).execute().unwrap();
            });
        });
    }


    group.finish();
}

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