[Fix](MV) query not hit partition in original planner (apache#38565)

this issue seems introduced by apache#21533

Fix issue that query not hit partition in original planner, which will
cause serve performance degradation.

fe/fe-core/src/main/java/org/apache/doris/planner/HashDistributionPruner.java

@@ -24,6 +24,7 @@
 import org.apache.doris.catalog.Column;
 import org.apache.doris.catalog.PartitionKey;
 import org.apache.doris.common.Config;
+import org.apache.doris.qe.ConnectContext;
 
 import com.google.common.collect.Lists;
 import com.google.common.collect.Sets;
@@ -79,10 +80,13 @@ public Collection<Long> prune(int columnId, PartitionKey hashKey, int complex) {
  return Lists.newArrayList(bucketsList.get((int) ((hashValue & 0xffffffff) % hashMod)));
  }
  Column keyColumn = distributionColumns.get(columnId);
- String columnName = isBaseIndexSelected ? keyColumn.getName()
- : org.apache.doris.nereids.rules.rewrite.mv.AbstractSelectMaterializedIndexRule
+ String columnName = keyColumn.getName();
+ if (!isBaseIndexSelected && ConnectContext.get().getSessionVariable().isEnableNereidsPlanner()) {
+ columnName = org.apache.doris.nereids.rules.rewrite.mv.AbstractSelectMaterializedIndexRule
  .normalizeName(
  CreateMaterializedViewStmt.mvColumnBuilder(keyColumn.getName()));
+ }
+
  PartitionColumnFilter filter = distributionColumnFilters.get(columnName);
  if (null == filter) {
  // no filter in this column, no partition Key

regression-test/suites/rollup_p0/test_rollup_prune_tablet.groovy

@@ -0,0 +1,52 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements. See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership. The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License. You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied. See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+suite("test_rollup_prune_tablet") {
+ sql "SET enable_nereids_planner=false"
+
+ sql "DROP TABLE IF EXISTS test_prune_tablet_t2"
+ sql """
+ CREATE TABLE `test_prune_tablet_t2` (
+ `id` INT NULL,
+ `c1` BOOLEAN NULL,
+ `id2` BIGINT NULL,
+ `id3` BIGINT NULL,
+ `id4` BIGINT NULL
+ ) distributed by hash(id) BUCKETS 16 properties('replication_num'='1');
+ """
+
+ createMV( """
+ alter table test_prune_tablet_t2 add rollup example_rollup_index(c1,id); 
+ """)
+
+ sql """insert into test_prune_tablet_t2 values
+ (1,0,1,1,1),
+ (2,0,2,2,2),
+ (3,0,3,3,3),
+ (4,0,4,4,4),
+ (5,0,5,5,5),
+ (6,0,6,6,6),
+ (7,0,7,7,7);"""
+
+ explain {
+ sql("select id,sum(c1) from test_prune_tablet_t2 where c1=0 and id = 3 group by id;")
+ contains "example_rollup_index"
+ contains "tablets=1/16"
+ }
+
+ sql "DROP TABLE IF EXISTS test_prune_tablet_t2"
+}