[fix](nereids)only enable colocate scan for one phase global parttion…

… topn in some condition
apache · Nov 6, 2023 · 40ccd49 · 40ccd49
1 parent 2ba2cd2
commit 40ccd49
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...e-core/src/main/java/org/apache/doris/nereids/glue/translator/PhysicalPlanTranslator.java b/...e-core/src/main/java/org/apache/doris/nereids/glue/translator/PhysicalPlanTranslator.java
@@ -85,6 +85,7 @@
 import org.apache.doris.nereids.trees.plans.AggMode;
 import org.apache.doris.nereids.trees.plans.AggPhase;
 import org.apache.doris.nereids.trees.plans.JoinType;
+import org.apache.doris.nereids.trees.plans.PartitionTopnPhase;
 import org.apache.doris.nereids.trees.plans.Plan;
 import org.apache.doris.nereids.trees.plans.PreAggStatus;
 import org.apache.doris.nereids.trees.plans.physical.AbstractPhysicalJoin;
@@ -1546,8 +1547,8 @@ public PlanFragment visitPhysicalPartitionTopN(PhysicalPartitionTopN<? extends P
  addPlanRoot(inputFragment, partitionSortNode, partitionTopN);
  // in pipeline engine, we use parallel scan by default, but it broke the rule of data distribution
  // we need turn of parallel scan to ensure to get correct result.
- // TODO: nereids forbid all parallel scan under PhysicalSetOperation temporary
- if (findOlapScanNodesByPassExchangeAndJoinNode(inputFragment.getPlanRoot())) {
+ if (partitionTopN.getPhase() == PartitionTopnPhase.ONE_PHASE_GLOBAL_PTOPN
+  && findOlapScanNodesByPassExchangeAndJoinNode(inputFragment.getPlanRoot())) {
  inputFragment.setHasColocatePlanNode(true);
  }
  return inputFragment;