br: modify how to parse the add index job args (#47456) (#47472)

close #47451

br/pkg/restore/ingestrec/ingest_recorder.go

@@ -79,9 +79,15 @@ func (i *IngestRecorder) AddJob(job *model.Job) error {
 		return nil
 	}
 
-	var indexID int64 = 0
-	if err := job.DecodeArgs(&indexID); err != nil {
-		return errors.Trace(err)
+	allIndexIDs := make([]int64, 1)
+	// The job.Args is either `Multi-index: ([]int64, ...)`,
+	// or `Single-index: (int64, ...)`.
+	// TODO: it's better to use the public function to parse the
+	// job's Args.
+	if err := job.DecodeArgs(&allIndexIDs[0]); err != nil {
+		if err = job.DecodeArgs(&allIndexIDs); err != nil {
+			return errors.Trace(err)
+		}
 	}
 
 	tableindexes, exists := i.items[job.TableID]
@@ -92,9 +98,11 @@ func (i *IngestRecorder) AddJob(job *model.Job) error {
 
 	// the current information of table/index might be modified by other ddl jobs,
 	// therefore update the index information at last
-	tableindexes[indexID] = &IngestIndexInfo{
-		IsPrimary: job.Type == model.ActionAddPrimaryKey,
-		Updated:   false,
+	for _, indexID := range allIndexIDs {
+		tableindexes[indexID] = &IngestIndexInfo{
+			IsPrimary: job.Type == model.ActionAddPrimaryKey,
+			Updated:   false,
+		}
 	}
 
 	return nil

br/pkg/stream/rewrite_meta_rawkv.go

@@ -840,15 +840,16 @@ func (sr *SchemasReplace) deleteRange(job *model.Job) error {
 				zap.Int64("oldTableID", job.TableID))
 			return nil
 		}
-		var indexID int64
-		var ifExists bool
+		indexIDs := make([]int64, 1)
+		ifExists := make([]bool, 1)
 		var partitionIDs []int64
-		if err := job.DecodeArgs(&indexID, &ifExists, &partitionIDs); err != nil {
-			return errors.Trace(err)
+		if err := job.DecodeArgs(&indexIDs[0], &ifExists[0], &partitionIDs); err != nil {
+			if err = job.DecodeArgs(&indexIDs, &ifExists, &partitionIDs); err != nil {
+				return errors.Trace(err)
+			}
 		}
 
 		var elementID int64 = 1
-		indexIDs := []int64{indexID}
 
 		if len(partitionIDs) > 0 {
 			for _, oldPid := range partitionIDs {