Function optimize_oneshot

src/adapter/src/optimize.rs:226–240

Runs the shared one-shot optimization pipeline for a single optimizer. This factors out the (otherwise duplicated) HIR ⇒ local MIR ⇒ resolve ⇒ global LIR sequence that is common to the `SELECT`/`EXPLAIN` and `COPY TO` paths. The `resolve` closure attaches the timestamp/session/stats context to the local plan; it is path-specific only in the concrete plan type it operates on.

(
    optimizer: &mut O,
    raw_expr: HirRelationExpr,
    resolve: impl FnOnce(LocalPlan) -> ResolvedPlan,
)

/// the local plan; it is path-specific only in the concrete plan type it
/// operates on.
pub(crate) fn optimize_oneshot<O, LocalPlan, ResolvedPlan, GlobalPlan>(
    optimizer: &mut O,
    raw_expr: HirRelationExpr,
    resolve: impl FnOnce(LocalPlan) -> ResolvedPlan,
) -> Result<GlobalPlan, OptimizerError>
where
    O: Optimize<HirRelationExpr, To = LocalPlan> + Optimize<ResolvedPlan, To = GlobalPlan>,
{
    // HIR ⇒ MIR lowering and MIR optimization (local).
    let local_mir_plan = optimizer.catch_unwind_optimize(raw_expr)?;
    // Attach resolved context required to continue the pipeline.
    let resolved_mir_plan = resolve(local_mir_plan);
    // MIR optimization (global), MIR ⇒ LIR lowering, and LIR optimization (global).
    optimizer.catch_unwind_optimize(resolved_mir_plan)
}

// Optimizer configuration
// -----------------------