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

[opt](nereids) convert or to inpredicate and optimize inpredicate partition prune (#28316)

Browse Source
This commit is contained in:
starocean999
2023-12-24 21:42:03 +08:00
committed by GitHub
parent 145683ccdb
commit 1e44a4b145
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
fe/fe-core/src/main/java/org/apache/doris/nereids/rules/expression/rules/OneRangePartitionEvaluator.java
View File

@ -396,11 +396,12 @@ public class OneRangePartitionEvaluator
if (inPredicate.getCompareExpr() instanceof Slot
&& inPredicate.getOptions().stream().allMatch(Literal.class::isInstance)) {
Slot slot = (Slot) inPredicate.getCompareExpr();
ColumnRange unionLiteralRange = inPredicate.getOptions()
.stream()
.map(Literal.class::cast)
.map(ColumnRange::singleton)
.reduce(ColumnRange.empty(), ColumnRange::union);
ColumnRange unionLiteralRange = ColumnRange.empty();
ColumnRange slotRange = result.childrenResult.get(0).columnRanges.get(slot);
for (Expression expr : inPredicate.getOptions()) {
unionLiteralRange = unionLiteralRange.union(
slotRange.intersect(ColumnRange.singleton((Literal) expr)));
}
Map<Slot, ColumnRange> slotRanges = result.childrenResult.get(0).columnRanges;
result = intersectSlotRange(result, slotRanges, slot, unionLiteralRange);
}

1
fe/fe-core/src/main/java/org/apache/doris/nereids/rules/expression/rules/PartitionPruner.java
View File

@ -113,6 +113,7 @@ public class PartitionPruner extends DefaultExpressionRewriter<Void> {
partitionTableType))
.collect(ImmutableList.toImmutableList());
partitionPredicate = OrToIn.INSTANCE.rewrite(partitionPredicate, null);
PartitionPruner partitionPruner = new PartitionPruner(evaluators, partitionPredicate);
//TODO: we keep default partition because it's too hard to prune it, we return false in canPrune().
return partitionPruner.prune();