Method scan_query_dir

benchmarks/src/benchmark_runner/suite.rs:181–220

Recursively scans a directory and appends valid query benchmark files to the provided collection.

(&self, dir: &Path, queries: &mut Vec<SuiteQuery>)

    /// Recursively scans a directory and appends valid query benchmark files to
    /// the provided collection.
    fn scan_query_dir(&self, dir: &Path, queries: &mut Vec<SuiteQuery>) -> Result<()> {
        let mut entries = read_dir_entries(dir, "benchmark query directory")?;
        entries.sort_by_key(|entry| entry.file_name());

        for entry in entries {
            let path = entry.path();
            let file_type = entry.file_type().map_err(|e| {
                DataFusionError::External(
                    format!(
                        "failed to read benchmark query entry type {}: {e}",
                        path.display()
                    )
                    .into(),
                )
            })?;

            if file_type.is_dir() {
                self.scan_query_dir(&path, queries)?;
                continue;
            }

            if path
                .extension()
                .is_none_or(|extension| extension != "benchmark")
            {
                continue;
            }

            let file_name = entry.file_name().to_string_lossy().into_owned();
            if let Some(id) = parse_query_file_name(&file_name) {
                queries.push(SuiteQuery {
                    id,
                    file_name,
                    path,
                });
            }
        }

        Ok(())
    }

    /// Validates suite metadata that cannot be enforced by TOML deserialization.
    fn validate(&self) -> Result<()> {