Struct AqeOptimizer 

pub struct AqeOptimizer { /* private fields */ }

AQE optimizer that automatically skips partition-changing rules for streaming plans.

Rules added via add_guarded_rule are not applied when StreamingAqeGuard::plan_is_streaming returns true. Rules added via add_rule always run regardless of execution kind — use this for rules that are safe on streaming plans (e.g., pure statistics collection).

Implementations

impl AqeOptimizer

pub fn new() -> Self

Create an empty AQE optimizer backed by StaticCostModel.

pub fn with_cost_model(self, model: Arc<dyn CostModel>) -> Self

Replace the default cost model with a custom implementation.

pub fn add_rule(&mut self, rule: Box<dyn AqeRule>)

Add a rule that always runs, including on streaming plans.

pub fn add_guarded_rule(&mut self, rule: Box<dyn AqeRule>)

Add a rule that is skipped when the plan is a streaming plan.

Use this for coalescing, repartitioning, and any other AQE rule that changes partition count or assignment.

pub fn apply( &self, plan: PhysicalPlan, stats: &[RuntimeStats], ) -> OptimizerResult<(PhysicalPlan, Vec<String>)>

Apply all applicable rules given per-stage runtime statistics.

When stats is empty the cost model is used to synthesise a single RuntimeStats entry (from the logical plan cost estimate) so that rules that need size information can still make a first-pass decision.

Returns the (possibly rewritten) plan and the names of rules that fired.

Trait Implementations

impl Default for AqeOptimizer

fn default() -> Self

Returns the “default value” for a type.