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

[feature](Nereids): eliminate useless project (#34611)

Browse Source
This commit is contained in:
jakevin
2024-05-13 10:17:29 +08:00
committed by yiguolei
parent e2614d453a
commit b3b848f862
3 changed files with 19 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -40,9 +40,16 @@ import java.util.stream.Collectors;
})
public class ColumnPruningPostProcessor extends PlanPostProcessor {
@Override
public PhysicalProject visitPhysicalProject(PhysicalProject<? extends Plan> project, CascadesContext ctx) {
public Plan visitPhysicalProject(PhysicalProject<? extends Plan> project, CascadesContext ctx) {
Plan child = project.child();
Plan newChild = child.accept(this, ctx);
if (project.isAllSlots()) {
Set<Slot> projects = project.getProjects().stream().map(Slot.class::cast).collect(Collectors.toSet());
Set<Slot> outputSet = newChild.getOutputSet();
if (outputSet.equals(projects)) {
return ((AbstractPhysicalPlan) newChild).copyStatsAndGroupIdFrom(project);
}
}
if (newChild instanceof AbstractPhysicalJoin) {
AbstractPhysicalJoin<? extends Plan, ? extends Plan> join = (AbstractPhysicalJoin) newChild;
Plan left = join.left();
@ -91,7 +98,7 @@ public class ColumnPruningPostProcessor extends PlanPostProcessor {
}
if (newLeft != left || newRight != right) {
return (PhysicalProject) project.withChildren(join.withChildren(newLeft, newRight));
return project.withChildren(join.withChildren(newLeft, newRight));
} else {
return project;
}

10
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/algebra/Project.java
View File

@ -117,4 +117,14 @@ public interface Project {
return expr;
});
}
/** isAllSlots */
default boolean isAllSlots() {
for (NamedExpression project : getProjects()) {
if (!project.isSlot()) {
return false;
}
}
return true;
}
}

10
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/logical/LogicalProject.java
View File

@ -114,16 +114,6 @@ public class LogicalProject<CHILD_TYPE extends Plan> extends LogicalUnary<CHILD_
return excepts;
}
/** isAllSlots */
public boolean isAllSlots() {
for (NamedExpression project : projects) {
if (!project.isSlot()) {
return false;
}
}
return true;
}
@Override
public List<Slot> computeOutput() {
Builder<Slot> slots = ImmutableList.builderWithExpectedSize(projects.size());