use std::fs;

use criterion::{black_box, criterion_group, criterion_main, Criterion};
use rstorage::{
    storage::{Config, DiskConfig, PartitionConfig, Storage},
    Bounds, DataPoint, EncodeStrategy, Row,
};

fn criterion_benchmark(c: &mut Criterion) {
    c.bench_function("insert", |b| {
        b.iter(|| {
            let data_path = String::from("./test_bench_data");
            let storage = Storage::new(Config {
                partition: PartitionConfig {
                    duration: 10000,
                    hot_partitions: 2,
                    max_partitions: 10,
                },
                disk: Some(DiskConfig {
                    data_path: data_path.clone(),
                    encode_strategy: EncodeStrategy::CSV,
                }),
                insert_bounds: Some(Bounds {
                    past: Some(100),
                    future: None,
                }),
                sweep_interval: None,
            })
            .unwrap();
            let num_metrics = 100_000;
            let metric = "hello";
            for i in 0..num_metrics {
                storage
                    .insert(&Row {
                        metric,
                        data_point: DataPoint {
                            timestamp: i,
                            value: 123.0,
                        },
                    })
                    .unwrap()
            }
            #[allow(unused_must_use)]
            {
                fs::remove_dir_all(data_path);
            }
        });
    });

    c.bench_function("select", |b| {
        let data_path = String::from("./test_bench_data");
        let storage = Storage::new(Config {
            partition: PartitionConfig {
                duration: 1_000,
                hot_partitions: 2,
                max_partitions: 10,
            },
            disk: Some(DiskConfig {
                data_path: data_path.clone(),
                encode_strategy: EncodeStrategy::CSV,
            }),
            insert_bounds: Some(Bounds {
                past: Some(100),
                future: None,
            }),
            sweep_interval: None,
        })
        .unwrap();
        let num_metrics = 10_000;
        let metric = "hello";
        for i in 0..num_metrics {
            black_box(
                storage
                    .insert(&Row {
                        metric,
                        data_point: DataPoint {
                            timestamp: i,
                            value: 123.0,
                        },
                    })
                    .unwrap(),
            )
        }

        let num_selects = 10;
        b.iter(|| {
            for _ in 0..num_selects {
                black_box(storage.select(metric, 0, 100_000).unwrap());
            }
        });

        #[allow(unused_must_use)]
        {
            fs::remove_dir_all(data_path);
        }
    });
}

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