Function get_query_sql_for_scale_factor

benchmarks/src/tpch/mod.rs:148–182

Get the SQL statements from the specified query file using the provided scale factor for TPC-H substitutions such as Q11 FRACTION.

(
    query: usize,
    scale_factor: f64,
)

/// Get the SQL statements from the specified query file using the provided scale factor for
/// TPC-H substitutions such as Q11 FRACTION.
pub fn get_query_sql_for_scale_factor(
    query: usize,
    scale_factor: f64,
) -> Result<Vec<String>> {
    if !(scale_factor.is_finite() && scale_factor > 0.0) {
        return plan_err!(
            "invalid scale factor. Expected a positive finite value, got {scale_factor}"
        );
    }

    if query > 0 && query < 23 {
        let possibilities = vec![
            format!("queries/q{query}.sql"),
            format!("benchmarks/queries/q{query}.sql"),
        ];
        let mut errors = vec![];
        for filename in possibilities {
            match fs::read_to_string(&filename) {
                Ok(contents) => {
                    let contents = customize_query_sql(query, contents, scale_factor)?;
                    return Ok(contents
                        .split(';')
                        .map(|s| s.trim())
                        .filter(|s| !s.is_empty())
                        .map(|s| s.to_string())
                        .collect());
                }
                Err(e) => errors.push(format!("{filename}: {e}")),
            };
        }
        plan_err!("invalid query. Could not find query: {:?}", errors)
    } else {
        plan_err!("invalid query. Expected value between 1 and 22")
    }
}

fn customize_query_sql(
    query: usize,