(cherry-pick) fix: Check maintains_input_order in EnforceDistribution (#35) (#37)

rkrishn7 · web-flow · commit da6be6fd2dde · 2026-03-24T15:14:50.000+08:00
diff --git a/datafusion/physical-optimizer/src/enforce_distribution.rs b/datafusion/physical-optimizer/src/enforce_distribution.rs
@@ -1472,7 +1472,7 @@ pub fn ensure_distribution(
                                 &plan,
                                 &child.plan,
                             );
-                        if !streaming_benefit {
+                        if !streaming_benefit && !maintains {
                             child =
                                 replace_order_preserving_variants(child, false)?.0;
                         }

Original file line number	Diff line number	Diff line change
`@@ -1472,7 +1472,7 @@ pub fn ensure_distribution(`
`1472`	`1472`	`&plan,`
`1473`	`1473`	`&child.plan,`
`1474`	`1474`	`);`
`1475`		`- if !streaming_benefit {`
	`1475`	`+ if !streaming_benefit && !maintains {`
`1476`	`1476`	`child =`
`1477`	`1477`	`replace_order_preserving_variants(child, false)?.0;`
`1478`	`1478`	`}`