From def9b7c01dd2b3a17643ee0214ff2a36090c3b52 Mon Sep 17 00:00:00 2001 From: Ruihang Xia Date: Wed, 9 Jul 2025 08:17:55 +0800 Subject: [PATCH] fix: expand on conditional commutative as well Signed-off-by: Ruihang Xia Signed-off-by: discord9 --- src/query/src/dist_plan/analyzer.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/query/src/dist_plan/analyzer.rs b/src/query/src/dist_plan/analyzer.rs index b3c5a0024d..55797398f5 100644 --- a/src/query/src/dist_plan/analyzer.rs +++ b/src/query/src/dist_plan/analyzer.rs @@ -186,6 +186,7 @@ impl PlanRewriter { Commutativity::PartialCommutative => { if let Some(plan) = partial_commutative_transformer(plan) { self.update_column_requirements(&plan); + self.expand_on_next_call = true; self.stage.push(plan) } } @@ -194,6 +195,7 @@ impl PlanRewriter { && let Some(plan) = transformer(plan) { self.update_column_requirements(&plan); + self.expand_on_next_call = true; self.stage.push(plan) } } @@ -202,7 +204,7 @@ impl PlanRewriter { && let Some(transformer_actions) = transformer(plan) { debug!( - "PlanRewriter: transformed plan: {:#?}\n from {plan}", + "PlanRewriter: transformed plan: {:?}\n from {plan}", transformer_actions.extra_parent_plans ); if let Some(last_stage) = transformer_actions.extra_parent_plans.last() {