address comments

pingcap · Apr 25, 2022 · 5834c9f · 5834c9f
1 parent db91789
commit 5834c9f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/br/pkg/lightning/restore/meta_manager.go b/br/pkg/lightning/restore/meta_manager.go
@@ -1087,13 +1087,15 @@ func (m *singleTaskMetaMgr) InitTask(ctx context.Context, source int64) error {
 func (m *singleTaskMetaMgr) CheckTasksExclusively(ctx context.Context, action func(tasks []taskMeta) ([]taskMeta, error)) error {
 	newTasks, err := action([]taskMeta{
 		{
-			taskID:      m.taskID,
-			status:      taskMetaStatusInitial,
-			sourceBytes: m.sourceBytes,
+			taskID:       m.taskID,
+			status:       taskMetaStatusInitial,
+			sourceBytes:  m.sourceBytes,
+			clusterAvail: m.clusterAvail,
 		},
 	})
 	for _, t := range newTasks {
 		if m.taskID == t.taskID {
+			m.sourceBytes = t.sourceBytes
 			m.clusterAvail = t.clusterAvail
 		}
 	}