use criterion::{black_box, criterion_group, criterion_main, Criterion};
use partiql_parser::{Parser, ParserResult};
use std::time::Duration;

const Q_STAR: &str = "SELECT *";

const Q_ION: &str = "SELECT a FROM `{'a':1,  'b':1}`";

const Q_GROUP: &str = "SELECT g FROM data GROUP BY a AS x, b + c AS y, foo(d) AS z GROUP AS g";

const Q_COMPLEX: &str = r#"
            SELECT (
                SELECT numRec, data
                FROM delta_full_transactions.deltas delta,
                (
                    SELECT foo(u.id), bar(review), rindex
                    FROM delta.data as u CROSS JOIN UNPIVOT u.reviews as review AT rindex
                ) as data,
                delta.numRec as numRec
            )
            AS deltas FROM SOURCE_VIEW_DELTA_FULL_TRANSACTIONS delta_full_transactions
            "#;

const Q_COMPLEX_FEXPR: &str = r#"
            SELECT (
                SELECT numRec, data
                FROM delta_full_transactions.deltas delta,
                (
                    SELECT cast(trim(u.id) as VARCHAR(20)), substring(review from 2 for 5), rindex
                    FROM delta.data as u CROSS JOIN UNPIVOT u.reviews as review AT rindex
                ) as data,
                delta.numRec as numRec
            )
            AS deltas FROM SOURCE_VIEW_DELTA_FULL_TRANSACTIONS delta_full_transactions
            "#;

fn parse_bench(c: &mut Criterion) {
    fn parse(text: &str) -> ParserResult {
        Parser::default().parse(text)
    }
    c.bench_function("parse-simple", |b| b.iter(|| parse(black_box(Q_STAR))));
    c.bench_function("parse-ion", |b| b.iter(|| parse(black_box(Q_ION))));
    c.bench_function("parse-group", |b| b.iter(|| parse(black_box(Q_GROUP))));
    c.bench_function("parse-complex", |b| b.iter(|| parse(black_box(Q_COMPLEX))));
    c.bench_function("parse-complex-fexpr", |b| {
        b.iter(|| parse(black_box(Q_COMPLEX_FEXPR)))
    });
}

criterion_group! {
    name = parse;
    config = Criterion::default().measurement_time(Duration::new(5, 0));
    targets = parse_bench
}

criterion_main!(parse);