spatialbench/src/q_and_a/mod.rs

//! TPC-H Queries and Answers.
//!
//! This module exposes a bundled query and answer tuple that makes it
//! easier to work with them in benchmark contexts.
pub mod answers_sf1;
pub mod queries;

/// QueryAndAnswer is a struct that contains a TPC-H query and its expected answer.
pub struct QueryAndAnswer(
    &'static str, // The TPC-H query as a string
    &'static str, // The expected answer as a string
);

impl QueryAndAnswer {
    /// Creates a new QueryAndAnswer instance.
    pub fn new(num: i32, scale_factor: f64) -> Result<Self, String> {
        match (num, scale_factor) {
            (1..=22, 1.) => Ok(QueryAndAnswer(
                queries::query(num).unwrap(),
                answers_sf1::answer(num).unwrap(),
            )),
            _ => Err(format!("Invalid TPC-H query number: {} the answers are only available for queries (1 to 22) and a scale factor of 1.0", num)),
        }
    }

    /// Returns the query string.
    pub fn query(&self) -> &str {
        self.0
    }

    /// Returns the expected answer string.
    pub fn answer(&self) -> &str {
        self.1
    }
}