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

[Bug](partition) fix npe when prune partition with not exist partition column in mv #31860

Browse Source
This commit is contained in:
Pxl
2024-03-06 20:47:06 +08:00
committed by yiguolei
parent 5c1c73e5bb
commit 3716f8a171
3 changed files with 79 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -17,6 +17,7 @@
package org.apache.doris.nereids.rules.rewrite;
import org.apache.doris.catalog.Column;
import org.apache.doris.catalog.OlapTable;
import org.apache.doris.catalog.PartitionInfo;
import org.apache.doris.catalog.PartitionItem;
@ -62,10 +63,16 @@ public class PruneOlapScanPartition extends OneRewriteRuleFactory {
.collect(Collectors.toMap(slot -> slot.getName().toLowerCase(), Function.identity()));
PartitionInfo partitionInfo = table.getPartitionInfo();
List<Slot> partitionSlots = partitionInfo.getPartitionColumns()
.stream()
.map(column -> scanOutput.get(column.getName().toLowerCase()))
.collect(Collectors.toList());
List<Slot> partitionSlots = new ArrayList<>();
for (Column column : partitionInfo.getPartitionColumns()) {
Slot slot = scanOutput.get(column.getName().toLowerCase());
if (slot == null) {
return filter;
} else {
partitionSlots.add(slot);
}
}
List<Long> manuallySpecifiedPartitions = scan.getManuallySpecifiedPartitions();
Map<Long, PartitionItem> idToPartitions;