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

fix bug in or expansion (#34851)

Browse Source
This commit is contained in:
谢健
2024-05-14 19:39:33 +08:00
committed by GitHub
parent ff22128013
commit c35fef631b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fe/fe-core/src/main/java/org/apache/doris/nereids/rules/rewrite/OrExpansion.java
View File

@ -107,7 +107,7 @@ public class OrExpansion extends DefaultPlanRewriter<OrExpandsionContext> implem
@Override
public Plan visitLogicalCTEAnchor(
LogicalCTEAnchor<? extends Plan, ? extends Plan> anchor, OrExpandsionContext ctx) {
Plan child1 = this.visit(anchor.child(0), ctx);
Plan child1 = anchor.child(0).accept(this, ctx);
// Consumer's CTE must be child of the cteAnchor in this case:
// anchor
// +-producer1
@ -121,7 +121,7 @@ public class OrExpansion extends DefaultPlanRewriter<OrExpandsionContext> implem
// +-consumer2 join consumer3
OrExpandsionContext consumerContext =
new OrExpandsionContext(ctx.statementContext, ctx.cascadesContext);
Plan child2 = this.visit(anchor.child(1), consumerContext);
Plan child2 = anchor.child(1).accept(this, consumerContext);
for (int i = consumerContext.cteProducerList.size() - 1; i >= 0; i--) {
LogicalCTEProducer<? extends Plan> producer = consumerContext.cteProducerList.get(i);
child2 = new LogicalCTEAnchor<>(producer.getCteId(), producer, child2);
@ -131,7 +131,7 @@ public class OrExpansion extends DefaultPlanRewriter<OrExpandsionContext> implem
@Override
public Plan visitLogicalJoin(LogicalJoin<? extends Plan, ? extends Plan> join, OrExpandsionContext ctx) {
join = (LogicalJoin<? extends Plan, ? extends Plan>) this.visit(join, ctx);
join = (LogicalJoin<? extends Plan, ? extends Plan>) join.accept(this, ctx);
if (join.isMarkJoin() || !JoinUtils.shouldNestedLoopJoin(join)) {
return join;
}