Method initialize

benchmarks/src/sql_benchmark.rs:117–155

Initializes the benchmark by executing `load` and `init` queries. Registers any required tables or sets up state in the provided `SessionContext` before running queries. This method is idempotent: calling it multiple times on the same instance returns immediately after the first successful initialization. # Errors Returns an error if any `load` or `init` query fails, or if the benchmark file do

(&mut self, ctx: &SessionContext)

    /// Returns an error if any `load` or `init` query fails, or if the
    /// benchmark file does not contain a `run` query.
    pub async fn initialize(&mut self, ctx: &SessionContext) -> Result<()> {
        if self.is_loaded {
            return Ok(());
        }

        let path = self.benchmark_path.to_string_lossy().into_owned();

        // validate there was a run query
        if !self.queries.contains_key(&QueryDirective::Run) {
            return Err(exec_datafusion_err!(
                "Invalid benchmark file: no \"run\" query specified: {path}"
            ));
        }

        // display any echo's
        self.echo.iter().for_each(|txt| println!("{txt}"));

        let load_queries = self.queries.get(&QueryDirective::Load);

        if let Some(queries) = load_queries {
            for query in queries {
                debug!("Executing load query {query}");
                ctx.sql(query).await?.collect().await?;
            }
        }

        let init_queries = self.queries.get(&QueryDirective::Init);

        if let Some(queries) = init_queries {
            for query in queries {
                debug!("Executing init query {query}");
                ctx.sql(query).await?.collect().await?;
            }
        }

        self.is_loaded = true;

        Ok(())
    }

    /// Executes the `assert` queries and compares actual results against
    /// expected values.