[regression](nereids) add test case for partition prune

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

@@ -222,8 +222,12 @@ public EvaluateRangeResult visit(Expression expr, EvaluateRangeInput context) {
         if (expr.getDataType() instanceof BooleanType && !(expr instanceof Literal)
                 && result.childrenResult.stream().anyMatch(childResult ->
                 childResult.columnRanges.values().stream().anyMatch(ColumnRange::isEmptyRange))) {
+            // this assumes that for expression: func(A)
+            // if A reject partition, then func(A) reject partition.
+            // implement visitFunc for Func if Func does not satisfy the above assumption.
             return new EvaluateRangeResult(BooleanLiteral.FALSE, result.columnRanges, result.childrenResult);
         }
+        // assumption: for func(A), if A accept range (n, m), then func(A) accept range (n, m).
         return result;
     }
 

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

@@ -1239,8 +1239,11 @@ public String getNodeExplainString(String prefix, TExplainLevel detailLevel) {
             output.append(getRuntimeFilterExplainString(false));
         }
 
-        output.append(prefix).append(String.format("partitions=%s/%s, tablets=%s/%s", selectedPartitionNum,
-                olapTable.getPartitions().size(), selectedTabletsNum, totalTabletsNum));
+        String selectedPartitions = getSelectedPartitionIds().stream().sorted()
+                .map(id -> olapTable.getPartition(id).getName())
+                .collect(Collectors.joining(","));
+        output.append(prefix).append(String.format("partitions=%s/%s (%s), tablets=%s/%s", selectedPartitionNum,
+                olapTable.getPartitions().size(), selectedPartitions, selectedTabletsNum, totalTabletsNum));
         // We print up to 3 tablet, and we print "..." if the number is more than 3
         if (scanTabletIds.size() > 3) {
             List<Long> firstTenTabletIds = scanTabletIds.subList(0, 3);

regression-test/data/performance_p0/redundant_conjuncts.out

@@ -13,7 +13,7 @@ PLAN FRAGMENT 0
   0:VOlapScanNode
      TABLE: default_cluster:regression_test_performance_p0.redundant_conjuncts(redundant_conjuncts), PREAGGREGATION: OFF. Reason: No AggregateInfo
      PREDICATES: `k1` = 1
-     partitions=0/1, tablets=0/0, tabletList=
+     partitions=0/1 (), tablets=0/0, tabletList=
      cardinality=0, avgRowSize=8.0, numNodes=1
      pushAggOp=NONE
 
@@ -31,7 +31,7 @@ PLAN FRAGMENT 0
   0:VOlapScanNode
      TABLE: default_cluster:regression_test_performance_p0.redundant_conjuncts(redundant_conjuncts), PREAGGREGATION: OFF. Reason: No AggregateInfo
      PREDICATES: `k1` = 1 OR `k1` = 2
-     partitions=0/1, tablets=0/0, tabletList=
+     partitions=0/1 (), tablets=0/0, tabletList=
      cardinality=0, avgRowSize=8.0, numNodes=1
      pushAggOp=NONE
 

regression-test/suites/nereids_rules_p0/partition_prune/test_multi_range_partition.groovy

@@ -0,0 +1,241 @@
+// 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_multi_range_partition") {
+    String db = context.config.getDbNameByFile(context.file)
+    sql "use ${db}"
+    sql "SET enable_nereids_planner=true"
+    sql "SET enable_fallback_to_original_planner=false"
+    sql "set partition_pruning_expand_threshold=10;"
+    sql "drop table if exists pt"
+    sql """
+        CREATE TABLE `pt` (
+        `k1` int(11) NULL COMMENT "",
+        `k2` int(11) NULL COMMENT "",
+        `k3` int(11) NULL COMMENT ""
+        ) 
+        PARTITION BY RANGE(`k1`, `k2`)
+        (PARTITION p1 VALUES LESS THAN ("3", "1"),
+        PARTITION p2 VALUES [("3", "1"), ("7", "10")),
+        PARTITION p3 VALUES [("7", "10"), ("10", "15")))
+        DISTRIBUTED BY HASH(`k1`) BUCKETS 10
+        PROPERTIES ('replication_num' = '1');
+        """
+    sql "insert into pt values (7, 0, 0);"
+    sql "insert into pt(k1) values (7);"
+    sql "insert into pt(k1) values (0);"
+    sql "insert into pt values (7, 11, 0);"
+
+    // basic test
+    explain{
+        sql "select * from pt where k1=7;"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain{
+        sql "select * from pt where k1=10;"
+        contains "partitions=1/3 (p3)"
+    }
+
+    explain{
+        sql "select * from pt where k1=11;"
+        contains "VEMPTYSET"
+    }
+
+    explain{
+        sql "select * from pt where k1=-1;"
+        contains "partitions=1/3 (p1)"
+    }
+
+    // ===============function(part_key)===================
+    explain{
+        sql "select * from pt where 2*k1=20; --漏裁 p1"
+        contains "partitions=2/3 (p1,p3)"
+    }
+    explain{
+        sql "select * from pt where 2*(k1+1)=22; --漏裁 p1"
+        contains "partitions=2/3 (p1,p3)"
+    }
+    explain{
+        sql "select * from pt where sin(k1)=0"
+        contains "artitions=3/3 (p1,p2,p3)"
+    }
+
+    // BUG: p1 missed
+    explain{
+        sql "select * from pt where sin(k1)<>0"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    // ============= in predicate ======================
+    explain{
+        sql "select * from pt where k1 in (7, 8);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain{
+        sql "select * from pt where k1 in (15, 18);"
+        contains "VEMPTYSET"
+    }
+
+    // =========== is null ===================
+    explain{
+        sql "select * from pt where k1 is null"
+        contains "partitions=1/3 (p1)"
+    }
+
+    // BUG: p1 missed
+    explain{
+        sql "select * from pt where k1 is not null"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    //======== the second partition key =========
+    explain{
+        sql "select * from pt where k1=7 and (k1>k2);"
+        contains "partitions=1/3 (p2)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and not (k1>k2);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    // p3 NOT pruned
+    explain {
+        sql "select * from pt where k1=7 and (k1<k2);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and k1=k2"
+        contains "partitions=1/3 (p2)"
+    }
+
+    //BUG: p2 missed
+    explain {
+        sql "select * from pt where k1=7 and k1<>k2"
+        contains "partitions=1/3 (p3)"
+    }
+
+    //p3 NOT pruned
+    explain {
+        sql "select * from pt where k1=7 and (k1 > cast(k2 as bigint));"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    //BUG: p2 missed
+    explain {
+        sql "select * from pt where k1=7 and not (k2 is null);"
+        contains "VEMPTYSET"
+    }
+
+    //p3 NOT pruned
+    explain {
+        sql "select * from pt where k1=7 and not (k2 is not null);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    //BUG: p2 missed
+    explain {
+        sql "select * from pt where k1=7 and k2 not in (1, 2);"
+        contains "partitions=1/3 (p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and k2 in (1, 12);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    //BUG: p2,p3 pruned
+    explain {
+        sql "select * from pt where k1=7 and k2 not in (1, 12)"
+        contains "VEMPTYSET"
+    }
+
+    explain {
+        sql" select * from pt where k1=7 and k2 in (0);"
+        contains "partitions=1/3 (p2)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and k2 in (11)"
+        contains "partitions=1/3 (p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and k2 in (null);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and k2 not in (null);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=10 and k2 in (30);"
+        contains "VEMPTYSET"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and k1 > k3;"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and k1 <> k3;"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k2 in (null);"
+        contains "partitions=3/3 (p1,p2,p3)"
+    }
+
+    // p1/p2/p3 NOT pruned
+    explain {
+        sql "select * from pt where k2 not in (null)"
+        contains "partitions=3/3 (p1,p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k2 in (0)"
+        contains "partitions=3/3 (p1,p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k2 > 100"
+        contains "partitions=3/3 (p1,p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k2 < -1"
+        contains "partitions=3/3 (p1,p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and (k3 is null)"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+    explain {
+        sql "select * from pt where k1=7 and not (k3 is null);"
+        contains "partitions=2/3 (p2,p3)"
+    }
+
+}