use eccodes::FallibleStreamingIterator;
use std::path::Path;

use criterion::{black_box, criterion_group, criterion_main, Criterion};
use eccodes::codes_handle::{CodesHandle, ProductKind};

pub fn key_reading(c: &mut Criterion) {
    //prepare the variables for benchmark
    let file_path = Path::new("./data/iceland.grib");
    let product_kind = ProductKind::GRIB;

    let mut handle = CodesHandle::new_from_file(file_path, product_kind).unwrap();

    let msg = handle.next().unwrap().unwrap();

    c.bench_function("long reading", |b| {
        b.iter(|| msg.read_key_dynamic(black_box("dataDate")).unwrap())
    });

    c.bench_function("double reading", |b| {
        b.iter(|| {
            msg.read_key_dynamic(black_box("jDirectionIncrementInDegrees"))
                .unwrap()
        })
    });

    c.bench_function("double array reading", |b| {
        b.iter(|| msg.read_key_dynamic(black_box("values")).unwrap())
    });

    c.bench_function("string reading", |b| {
        b.iter(|| msg.read_key_dynamic(black_box("name")).unwrap())
    });

    c.bench_function("bytes reading", |b| {
        b.iter(|| msg.read_key_dynamic(black_box("section1Padding")).unwrap())
    });

    c.bench_function("missing nul-byte termination reading", |b| {
        b.iter(|| {
            msg.read_key_dynamic(black_box("experimentVersionNumber"))
                .unwrap()
        })
    });

    c.bench_function("problematic key reading", |b| {
        b.iter(|| {
            msg.read_key_dynamic(black_box("zero"))
                .unwrap_or_else(|_| msg.read_key_dynamic(black_box("zeros")).unwrap())
        })
    });
}

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