Function process_replacements_with_env

benchmarks/src/sql_benchmark.rs:1359–1401

(
    input: &str,
    replacement_map: &HashMap<String, String>,
    get_env: impl Fn(&str) -> Option<String>,
)

}

fn process_replacements_with_env(
    input: &str,
    replacement_map: &HashMap<String, String>,
    get_env: impl Fn(&str) -> Option<String>,
) -> Result<String> {
    debug!("processing replacements for line '{input}'");

    // handle ${VAR:-default|true value|false value} syntax
    let replacement = |caps: &regex::Captures| -> Result<String> {
        let key = &caps[1];
        let default = caps.get(2).map(|m| m.as_str().to_string());
        let true_val = &caps[3];
        let false_val = &caps[4];

        let value = lookup_replacement_value(key, replacement_map, &get_env).or(default);

        match value {
            Some(v) if v.eq_ignore_ascii_case("true") => Ok(true_val.to_string()),
            Some(_) => Ok(false_val.to_string()),
            None => Err(exec_datafusion_err!("Missing value for key '{key}'")),
        }
    };
    let input = replace_all(&TRUE_FALSE_REPLACEMENT_RE, input, replacement)?;

    // handle ${KEY} and ${KEY:-default}`
    let replacement = |caps: &regex::Captures| -> Result<String> {
        let key = &caps[1];
        let default = caps.get(2);

        if let Some(v) = lookup_replacement_value(key, replacement_map, &get_env) {
            return Ok(v.to_string());
        }

        // use default if it was set
        if let Some(def) = default {
            Ok(def.as_str().to_string())
        } else {
            Err(exec_datafusion_err!("Missing value for key '{key}'"))
        }
    };

    replace_all(&VARIABLE_REPLACEMENT_RE, &input, replacement)
}

fn lookup_replacement_value(
    key: &str,