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

[fix](nereids) PushdownAliasThroughJoin may lost required columns from parent node (#25835)

Browse Source
This commit is contained in:
starocean999
2023-10-26 11:03:41 +08:00
committed by GitHub
parent 31d2a9a4f5
commit f322ad277d
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -37,6 +37,7 @@ import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;
/**
* Pushdown Alias (inside must be Slot) through Join.
@ -112,9 +113,11 @@ public class PushdownAliasThroughJoin extends OneRewriteRuleFactory {
}
private List<NamedExpression> createNewOutput(List<Slot> oldOutput,
Map<Expression, List<NamedExpression>> aliasMap) {
List<NamedExpression> output = oldOutput.stream()
.flatMap(slot -> aliasMap.getOrDefault(slot, Collections.singletonList(slot)).stream())
Map<Expression, List<NamedExpression>> aliasMap) {
// we should keep all original outputs and add new alias in the output list
// because the upper node may require both col#1 and col#1 as colAlias#2
List<NamedExpression> output = Stream.concat(oldOutput.stream(), oldOutput.stream()
.flatMap(slot -> aliasMap.getOrDefault(slot, Collections.emptyList()).stream()))
.collect(Collectors.toList());
return output;
}