planner/core: support range partition pruning for 'in' expression (#17210)

[sre-bot]

cherry-pick #17210 to release-3.0

---

What problem does this PR solve?

create table t1 (a int)  partition by range (a) (
    partition p0 values less than (10),
    partition p1 values less than (20),
    partition p2 values less than (30),
    partition p3 values less than (40),
    partition p4 values less than MAXVALUE
);

explain select * from t1 where a in (20, 21, 22, 23, 24, 25);
+------------------------+----------+-----------+------------------------+---------------------------------------+
| id                     | estRows  | task      | access object          | operator info                         |
+------------------------+----------+-----------+------------------------+---------------------------------------+
| Union_11               | 300.00   | root      |                        |                                       |
| ├─TableReader_14       | 60.00    | root      |                        | data:Selection_13                     |
| │ └─Selection_13       | 60.00    | cop[tikv] |                        | in(test.t1.a, 20, 21, 22, 23, 24, 25) |
| │   └─TableFullScan_12 | 10000.00 | cop[tikv] | table:t1, partition:p0 | keep order:false, stats:pseudo        |
| ├─TableReader_17       | 60.00    | root      |                        | data:Selection_16                     |
| │ └─Selection_16       | 60.00    | cop[tikv] |                        | in(test.t1.a, 20, 21, 22, 23, 24, 25) |
| │   └─TableFullScan_15 | 10000.00 | cop[tikv] | table:t1, partition:p1 | keep order:false, stats:pseudo        |
| ├─TableReader_20       | 60.00    | root      |                        | data:Selection_19                     |
| │ └─Selection_19       | 60.00    | cop[tikv] |                        | in(test.t1.a, 20, 21, 22, 23, 24, 25) |
| │   └─TableFullScan_18 | 10000.00 | cop[tikv] | table:t1, partition:p2 | keep order:false, stats:pseudo        |
| ├─TableReader_23       | 60.00    | root      |                        | data:Selection_22                     |
| │ └─Selection_22       | 60.00    | cop[tikv] |                        | in(test.t1.a, 20, 21, 22, 23, 24, 25) |
| │   └─TableFullScan_21 | 10000.00 | cop[tikv] | table:t1, partition:p3 | keep order:false, stats:pseudo        |
| └─TableReader_26       | 60.00    | root      |                        | data:Selection_25                     |
|   └─Selection_25       | 60.00    | cop[tikv] |                        | in(test.t1.a, 20, 21, 22, 23, 24, 25) |
|     └─TableFullScan_24 | 10000.00 | cop[tikv] | table:t1, partition:p4 | keep order:false, stats:pseudo        |
+------------------------+----------+-----------+------------------------+---------------------------------------+

Problem Summary:

Our current partition pruning doesn't handle the in expression.

What is changed and how it works?

Proposal: xxx

What's Changed:

Support range partition pruning for 'in' expression

How it Works:

x in (1, 2) can be treat as x = 1 or x = 2 ...

Related changes

• Need to cherry-pick to the release branch

Check List

Tests

• Integration test

Release note

• Support range partition pruning for 'in' expression

[sre-bot]

/run-all-tests

[imtbkcat]

@tiancaiamao please fix conflict

[tiancaiamao]

/run-all-tests

[tiancaiamao]

/run-all-tests tidb-test=pr/1059

[tiancaiamao]

/run-all-tests tidb-test=pr/1059

[XuHuaiyu]

LGTM

[imtbkcat]

LGTM