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

[fix](nereids)need add mark join slot to upper project in PullUpProjectUnderApply rule (#31408)

Browse Source
This commit is contained in:
starocean999
2024-02-28 22:31:29 +08:00
committed by yiguolei
parent 5e80a718fd
commit 4a5283b466
3 changed files with 7 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -65,6 +65,9 @@ public class PullUpProjectUnderApply extends OneRewriteRuleFactory {
"ScalarSubquery should only have one output column");
newProjects.add(project.getProjects().get(0));
}
if (apply.isMarkJoin()) {
newProjects.add(apply.getMarkJoinSlotReference().get());
}
return project.withProjectsAndChild(newProjects, newCorrelate);
}).toRule(RuleType.PULL_UP_PROJECT_UNDER_APPLY);
}

2
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/MarkJoinSlotReference.java
View File

@ -25,7 +25,7 @@ import com.google.common.collect.ImmutableList;
/**
* A special type of column that will be generated to replace the subquery when unnesting the subquery of MarkJoin.
*/
public class MarkJoinSlotReference extends SlotReference implements SlotNotFromChildren {
public class MarkJoinSlotReference extends SlotReference {
final boolean existsHasAgg;
public MarkJoinSlotReference(String name) {