use std::time::Duration;

use bytes::Bytes;
use criterion::{
    black_box, criterion_group, criterion_main, measurement::WallTime, BatchSize, BenchmarkGroup,
    Criterion, SamplingMode,
};
use vrl::datadog_grok::filters::keyvalue::{self, KeyValueFilter};
use vrl::value::Value;

fn make_filter() -> KeyValueFilter {
    let quotes = vec![('"', '"'), ('\'', '\''), ('<', '>')];
    KeyValueFilter {
        re_pattern: keyvalue::regex_from_config(
            "=",
            r"\w.\-_@",
            quotes.clone(),
            ("|".to_string(), "|".to_string()),
        )
        .unwrap(),
        quotes: vec![('"', '"'), ('\'', '\''), ('<', '>')],
    }
}

fn apply_filter_bench(c: &mut Criterion) {
    let mut group: BenchmarkGroup<WallTime> =
        c.benchmark_group("datadog_grok::filters::keyvalue::apply_filter");
    group.sampling_mode(SamplingMode::Auto);

    group.bench_function("apply_filter key=valueStr", move |b| {
        b.iter_batched(
            || (Value::Bytes(Bytes::from("key=valueStr")), make_filter()),
            |(value, filter): (Value, KeyValueFilter)| {
                let result = black_box(filter.apply_filter(&value)).unwrap();
                let object = result.as_object().unwrap();
                assert_eq!(object.len(), 1);
            },
            BatchSize::SmallInput,
        )
    });

    group.bench_function("apply_filter key1=value1|key2=value2", move |b| {
        b.iter_batched(
            || {
                (
                    Value::Bytes(Bytes::from("key1=value1|key2=value2")),
                    make_filter(),
                )
            },
            |(value, filter): (Value, KeyValueFilter)| {
                let result = black_box(filter.apply_filter(&value)).unwrap();
                let object = result.as_object().unwrap();
                assert_eq!(object.len(), 2);
            },
            BatchSize::SmallInput,
        )
    });
}

criterion_group!(
    name = benches;
    config = Criterion::default()
        .warm_up_time(Duration::from_secs(5))
        .measurement_time(Duration::from_secs(120))
        // degree of noise to ignore in measurements, here 1%
        .noise_threshold(0.01)
        // likelihood of noise registering as difference, here 5%
        .significance_level(0.05)
        // likelihood of capturing the true runtime, here 95%
        .confidence_level(0.95)
        // total number of bootstrap resamples, higher is less noisy but slower
        .nresamples(100_000)
        // total samples to collect within the set measurement time
        .sample_size(150);
    targets = apply_filter_bench
);
criterion_main!(benches);