ddl: fix bug that new partition will not set the placement policy aft…

…er truncated (pingcap#44043) (pingcap#44061) close pingcap#44031
lcwangchao · May 24, 2023 · 0535004 · 0535004
1 parent e189cd5
commit 0535004
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/ddl/partition.go b/ddl/partition.go
@@ -1883,6 +1883,16 @@ func onTruncateTablePartition(d *ddlCtx, t *meta.Meta, job *model.Job) (int64, e
 		return ver, errors.Trace(err)
 	}
 
+	tableBundle, err := placement.NewTableBundle(t, tblInfo)
+	if err != nil {
+		job.State = model.JobStateCancelled
+		return ver, errors.Trace(err)
+	}
+
+	if tableBundle != nil {
+		bundles = append(bundles, tableBundle)
+	}
+
 	err = infosync.PutRuleBundlesWithDefaultRetry(context.TODO(), bundles)
 	if err != nil {
 		job.State = model.JobStateCancelled

diff --git a/ddl/placement_policy_test.go b/ddl/placement_policy_test.go
@@ -1835,6 +1835,7 @@ func TestTruncateTablePartitionWithPlacement(t *testing.T) {
 		" PARTITION `p1` VALUES LESS THAN (1000) /*T![placement] PLACEMENT POLICY=`p2` */,\n" +
 		" PARTITION `p2` VALUES LESS THAN (10000) /*T![placement] PLACEMENT POLICY=`p3` */,\n" +
 		" PARTITION `p3` VALUES LESS THAN (100000))"))
+	checkExistTableBundlesInPD(t, dom, "test", "tp")
 }
 
 func TestTruncatePartitionGCWithPlacement(t *testing.T) {