use criterion::{black_box, criterion_group, criterion_main, Criterion};
use squeal::*;

fn generate() -> String {
        let result = Query {
        select: Select::new(Columns::Selected(vec!["a", "b"])),
        from: "table",
        where_clause: Some(
            Term::Condition(
                Box::new(Term::Atom("a")),
                Op::O("<>"),
                Box::new(Term::Atom("b")),
            )
        ),
        group_by: Some(vec!["a", "b"]),
        having: Some(Having::new(
            Term::Condition(
                Box::new(Term::Atom("a")),
                Op::O("<>"),
                Box::new(Term::Atom("b")),
            )
        )),
        order_by: Some(OrderBy{columns: vec![ OrderedColumn::Asc("a"),
                                              OrderedColumn::Desc("b")]}),
        limit: Some(19),
        offset: Some(10),
        }.sql();
    result
}

fn fluent_generation() -> String {
    let mut q = Q("the table");

    let result = q.select(vec!["a", "sum(b)"])
        .from("the_table")
        .where_(Term::Condition(
            Box::new(Term::Atom("a")),
            Op::O("<>"),
            Box::new(Term::Atom("b")),
        ))
        .group_by(vec!["a"])
        .having(Term::Condition(
            Box::new(Term::Atom("a")),
            Op::O("<>"),
            Box::new(Term::Atom("b")),
        ))
        .order_by(vec![OrderedColumn::Asc("a")])
        .limit(19)
        .offset(10).build();
    result.sql()
}

fn criterion_benchmark(c: &mut Criterion) {
    c.bench_function("generate", |b| b.iter(|| { generate(); } ));
    c.bench_function("fluent generation", |b| b.iter(|| { fluent_generation(); } ));
}

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