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

[enhancement](Nereids): extract group plan in struct info node (#27939)

Browse Source
This commit is contained in:
谢健
2023-12-04 19:46:40 +08:00
committed by GitHub
parent 4c9bf98dcd
commit 8e2961858e
2 changed files with 40 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
fe/fe-core/src/main/java/org/apache/doris/nereids/jobs/joinorder/hypergraph/node/StructInfoNode.java
View File

@ -19,8 +19,11 @@ package org.apache.doris.nereids.jobs.joinorder.hypergraph.node;
import org.apache.doris.nereids.jobs.joinorder.hypergraph.Edge;
import org.apache.doris.nereids.jobs.joinorder.hypergraph.HyperGraph;
import org.apache.doris.nereids.trees.plans.GroupPlan;
import org.apache.doris.nereids.trees.plans.Plan;
import com.google.common.collect.ImmutableList;
import java.util.ArrayList;
import java.util.List;
@ -32,7 +35,7 @@ public class StructInfoNode extends AbstractNode {
private List<HyperGraph> graphs = new ArrayList<>();
public StructInfoNode(int index, Plan plan, List<Edge> edges) {
super(plan, index, edges);
super(extractPlan(plan), index, edges);
}
public StructInfoNode(int index, Plan plan) {
@ -44,6 +47,18 @@ public class StructInfoNode extends AbstractNode {
this.graphs = graphs;
}
private static Plan extractPlan(Plan plan) {
if (plan instanceof GroupPlan) {
//TODO: Note mv can be in logicalExpression, how can we choose it
plan = ((GroupPlan) plan).getGroup().getLogicalExpressions().get(0)
.getPlan();
}
List<Plan> children = plan.children().stream()
.map(StructInfoNode::extractPlan)
.collect(ImmutableList.toImmutableList());
return plan.withChildren(children);
}
public boolean needToFlat() {
return !graphs.isEmpty();
}