planner: Using converted value in newBatchPointGetPlan, fix for partitioned table and IN (#57011) (#59325)

close #54746

Author: ti-chi-bot

cmd/explaintest/r/integration_partition.result

@@ -0,0 +1,19 @@
+drop table if exists t;
+CREATE TABLE `t` (`tenant_id` bigint(20) NOT NULL DEFAULT '0',`order_id` bigint(20) NOT NULL DEFAULT '0',UNIQUE KEY `uk_ten_ord` (`order_id`, `tenant_id`))  PARTITION BY HASH(`tenant_id`) PARTITIONS 32;
+INSERT INTO t(tenant_id, order_id) VALUES (123, 456);
+select * from t where (tenant_id, order_id) in (('123','456'));
+tenant_id	order_id
+123	456
+select * from t where (tenant_id, order_id) in (('0123','456'));
+tenant_id	order_id
+123	456
+drop table t;
+CREATE TABLE `t` (`tenant_id` bigint(20) NOT NULL DEFAULT '0',`order_id` bigint(20) NOT NULL DEFAULT '0',UNIQUE KEY `uk_ten_ord` (`order_id`, `tenant_id`));
+INSERT INTO t(tenant_id, order_id) VALUES (123, 456);
+select * from t where (tenant_id, order_id) in (('123','456'));
+tenant_id	order_id
+123	456
+select * from t where (tenant_id, order_id) in (('0123','456'));
+tenant_id	order_id
+123	456
+drop table t;

cmd/explaintest/t/integration_partition.test

@@ -0,0 +1,14 @@
+# TestIssue54746
+drop table if exists t;
+CREATE TABLE `t` (`tenant_id` bigint(20) NOT NULL DEFAULT '0',`order_id` bigint(20) NOT NULL DEFAULT '0',UNIQUE KEY `uk_ten_ord` (`order_id`, `tenant_id`))  PARTITION BY HASH(`tenant_id`) PARTITIONS 32;
+INSERT INTO t(tenant_id, order_id) VALUES (123, 456);
+select * from t where (tenant_id, order_id) in (('123','456'));
+select * from t where (tenant_id, order_id) in (('0123','456'));
+
+drop table t;
+CREATE TABLE `t` (`tenant_id` bigint(20) NOT NULL DEFAULT '0',`order_id` bigint(20) NOT NULL DEFAULT '0',UNIQUE KEY `uk_ten_ord` (`order_id`, `tenant_id`));
+INSERT INTO t(tenant_id, order_id) VALUES (123, 456);
+select * from t where (tenant_id, order_id) in (('123','456'));
+select * from t where (tenant_id, order_id) in (('0123','456'));
+
+drop table t;

planner/core/point_get_plan.go

@@ -808,8 +808,8 @@ func newBatchPointGetPlan(
 					if dval == nil {
 						return nil
 					}
-					values[permIndex] = innerX.Datum
-					pairs = append(pairs, nameValuePair{colName: whereColNames[index], value: innerX.Datum})
+					values[permIndex] = *dval
+					pairs = append(pairs, nameValuePair{colName: whereColNames[index], value: *dval})
 				case *driver.ParamMarkerExpr:
 					con, err := expression.ParamMarkerExpression(ctx, innerX, true)
 					if err != nil {
@@ -823,12 +823,12 @@ func newBatchPointGetPlan(
 					if dval == nil {
 						return nil
 					}
-					values[permIndex] = innerX.Datum
+					values[permIndex] = *dval
 					valuesParams[permIndex] = con
 					if initTypes {
 						indexTypes[permIndex] = &colInfos[index].FieldType
 					}
-					pairs = append(pairs, nameValuePair{colName: whereColNames[index], value: innerX.Datum})
+					pairs = append(pairs, nameValuePair{colName: whereColNames[index], value: *dval})
 				default:
 					return nil
 				}