Struct StreamingAqeGuard 

pub struct StreamingAqeGuard;

Guards streaming plans from AQE rules that would change partition count.

Stateful streaming stages use keyed-distribution routing: the same key must always map to the same executor task for the entire job lifetime. AQE coalescing and repartitioning would change the partition count mid-job, orphaning all in-flight state.

Place this rule first in any AQE pipeline that includes coalescing or repartitioning rules. When the plan carries ExecutionKind::Streaming, all subsequent AQE rules that affect partitioning must be skipped.

Usage:

use krishiv_plan::optimizer::{AqeOptimizer, CoalesceRule, StreamingAqeGuard};
let mut aqe = AqeOptimizer::new();
aqe.add_guarded_rule(Box::new(CoalesceRule::new(64 * 1024 * 1024)));

Implementations

impl StreamingAqeGuard

pub fn plan_is_streaming(plan: &PhysicalPlan) -> bool

Returns true if the plan contains any streaming node that must not be subject to AQE partition-count changes.

P3.18: Walk the plan tree recursively so that hybrid batch/streaming plans are also detected. A plan is considered streaming if either its top-level ExecutionKind is Streaming or any of its nodes carries ExecutionKind::Streaming.