database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[fix](nereids): fix PlanPostProcessor use visitor (#35244)

Browse Source 
(cherry picked from commit 46e004a358b9e13adb492d376f77e4317e558a6a)
This commit is contained in:
jakevin
2024-05-23 11:55:16 +08:00
committed by jackwener
parent e962a7309b
commit 0b440685d9
2 changed files with 4 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fe/fe-core/src/main/java/org/apache/doris/nereids/processor/post/AddOffsetIntoDistribute.java
View File

@ -31,11 +31,12 @@ import org.apache.doris.nereids.trees.plans.physical.PhysicalLimit;
public class AddOffsetIntoDistribute extends PlanPostProcessor {
@Override
public Plan visitPhysicalLimit(PhysicalLimit<? extends Plan> limit, CascadesContext context) {
limit = (PhysicalLimit<? extends Plan>) super.visit(limit, context);
if (limit.getPhase().isLocal() || limit.getOffset() == 0) {
return limit;
}
return new PhysicalDistribute<>(DistributionSpecGather.INSTANCE,
limit.withLimit(limit.getLimit() + limit.getOffset()));
limit.withLimit(limit.getLimit() + limit.getOffset())).copyStatsAndGroupIdFrom(limit);
}
}

10
fe/fe-core/src/main/java/org/apache/doris/nereids/processor/post/PushDownFilterThroughProject.java
View File

@ -19,7 +19,6 @@ package org.apache.doris.nereids.processor.post;
import org.apache.doris.nereids.CascadesContext;
import org.apache.doris.nereids.trees.plans.Plan;
import org.apache.doris.nereids.trees.plans.physical.AbstractPhysicalPlan;
import org.apache.doris.nereids.trees.plans.physical.PhysicalFilter;
import org.apache.doris.nereids.trees.plans.physical.PhysicalProject;
import org.apache.doris.nereids.util.ExpressionUtils;
@ -30,15 +29,10 @@ import org.apache.doris.nereids.util.ExpressionUtils;
public class PushDownFilterThroughProject extends PlanPostProcessor {
@Override
public Plan visitPhysicalFilter(PhysicalFilter<? extends Plan> filter, CascadesContext context) {
filter = (PhysicalFilter<? extends Plan>) super.visit(filter, context);
Plan child = filter.child();
if (!(child instanceof PhysicalProject)) {
Plan newChild = child.accept(this, context);
if (newChild == child) {
return filter;
} else {
return ((AbstractPhysicalPlan) filter.withChildren(child.accept(this, context)))
.copyStatsAndGroupIdFrom(filter);
}
return filter;
}
PhysicalProject<? extends Plan> project = (PhysicalProject<? extends Plan>) child;