parser, ddl: correct the job type of 'REORGANIZE PARTITION' (#42490) (#…

…42494)

close #42442

parser/model/ddl.go

@@ -758,7 +758,7 @@ func (job *Job) NotStarted() bool {
 // MayNeedReorg indicates that this job may need to reorganize the data.
 func (job *Job) MayNeedReorg() bool {
 	switch job.Type {
-	case ActionAddIndex, ActionAddPrimaryKey:
+	case ActionAddIndex, ActionAddPrimaryKey, ActionReorganizePartition:
 		return true
 	case ActionModifyColumn:
 		if len(job.CtxVars) > 0 {

parser/model/ddl_test.go

@@ -71,3 +71,34 @@ func TestBackfillMetaCodec(t *testing.T) {
 	bmRet.Decode(bmBytes)
 	require.Equal(t, bm, bmRet)
 }
+
+func TestMayNeedReorg(t *testing.T) {
+	//TODO(bb7133): add more test cases for different ActionType.
+	reorgJobTypes := []model.ActionType{
+		model.ActionReorganizePartition,
+		model.ActionAddIndex,
+		model.ActionAddPrimaryKey,
+	}
+	generalJobTypes := []model.ActionType{
+		model.ActionCreateTable,
+		model.ActionDropTable,
+	}
+	job := &model.Job{
+		ID:              100,
+		Type:            model.ActionCreateTable,
+		SchemaID:        101,
+		TableID:         102,
+		SchemaName:      "test",
+		TableName:       "t",
+		State:           model.JobStateDone,
+		MultiSchemaInfo: nil,
+	}
+	for _, jobType := range reorgJobTypes {
+		job.Type = jobType
+		require.True(t, job.MayNeedReorg())
+	}
+	for _, jobType := range generalJobTypes {
+		job.Type = jobType
+		require.False(t, job.MayNeedReorg())
+	}
+}