use criterion::{black_box, criterion_group, criterion_main, Criterion};
use tiny_dynamo::{Const, Credentials, Request, Table, DB};

fn get_item(db: DB) -> Result<Request, Box<dyn std::error::Error>> {
    db.get_item_req("test")
}

fn put_item(db: DB) -> Result<Request, Box<dyn std::error::Error>> {
    db.put_item_req("test", "value")
}

fn criterion_benchmark(c: &mut Criterion) {
    c.bench_function("get_item", |b| {
        b.iter(|| {
            get_item(black_box(DB::new(
                Credentials::new("test", "test"),
                Table::new(
                    "test",
                    "key",
                    "value",
                    "us-east-1".parse()?,
                    Some("http://localhost:8000".into()),
                ),
                Const(200, "".into()),
            )))
        })
    });

    c.bench_function("put_item", |b| {
        b.iter(|| {
            put_item(black_box(DB::new(
                Credentials::new("test", "test"),
                Table::new(
                    "test",
                    "key",
                    "value",
                    "us-east-1".parse()?,
                    Some("http://localhost:8000".into()),
                ),
                Const(200, "".into()),
            )))
        })
    });
}

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