Function load_benchmarks

benchmarks/benches/sql.rs:264–295

Loads all benchmark files in the `sql_benchmarks` directory. For each file ending with `.benchmark` it creates a new `SqlBenchmark` instance.

(
    args: &EnvParser,
    ctx: &SessionContext,
    path: &str,
)

/// For each file ending with `.benchmark` it creates a new
/// `SqlBenchmark` instance.
async fn load_benchmarks(
    args: &EnvParser,
    ctx: &SessionContext,
    path: &str,
) -> Result<BTreeMap<String, Vec<SqlBenchmark>>> {
    let mut benches = BTreeMap::new();
    let mut paths = Vec::new();

    list_files(path, &mut |path: &str| {
        if path.ends_with(".benchmark") {
            paths.push(path.to_string());
        }
    });

    for path in paths {
        debug!("Loading benchmark from {path}");

        let benchmark = SqlBenchmark::new(ctx, &path, &*SQL_BENCHMARK_DIRECTORY).await?;
        let entries = benches
            .entry(benchmark.group().to_string())
            .or_insert(vec![]);

        entries.push(benchmark);
    }

    benches = filter_benchmarks(args, benches);
    benches.iter_mut().for_each(|(_, benchmarks)| {
        benchmarks.sort_by(|b1, b2| b1.name().cmp(b2.name()))
    });

    Ok(benches)
}

fn filter_benchmarks(
    args: &EnvParser,