Method process_query_directive

benchmarks/src/sql_benchmark.rs:785–860

(
        bench: &mut SqlBenchmark,
        reader: &mut BenchmarkFileReader,
        line: &mut String,
        splits: &[&str],
    )

    }

    fn process_query_directive(
        bench: &mut SqlBenchmark,
        reader: &mut BenchmarkFileReader,
        line: &mut String,
        splits: &[&str],
    ) -> Result<()> {
        let directive = QueryDirective::parse(splits[0]).ok_or_else(|| {
            exec_datafusion_err!("Invalid query directive: {}", splits[0])
        })?;

        if directive == QueryDirective::Run && bench.queries.contains_key(&directive) {
            return Err(exec_datafusion_err!(
                "Multiple calls to run in the same benchmark file"
            ));
        }

        line.clear();

        // Read the query body until a blank line or EOF.
        let mut query = String::new();
        let mut reader_result = reader.read_line(line);

        loop {
            match reader_result {
                Some(Ok(_)) => {
                    if is_comment_line(line) {
                        // comment, ignore
                    } else if is_blank_line(line) {
                        break;
                    } else {
                        query.push_str(line);
                        query.push('\n');
                    }
                }
                Some(Err(e)) => return Err(e),
                None => break,
            }

            // Clear the line buffer for the next iteration.
            line.clear();
            reader_result = reader.read_line(line);
        }

        // Optional file parameter.
        if splits.len() > 1 && !splits[1].is_empty() {
            if !query.trim().is_empty() {
                return Err(exec_datafusion_err!(
                    "{}",
                    reader.format_exception(&format!(
                        "{} directive must use either a query file or inline SQL, not both",
                        directive.as_str()
                    ))
                ));
            }

            debug!("Processing {} file: {}", splits[0], splits[1]);

            let query_file = fs::read_to_string(splits[1]).map_err(|e| {