br: check the unchecked error of initialization of dml iterator in pi…

…tr (#45854) close #45853
pingcap · Aug 7, 2023 · 52680cd · 52680cd
1 parent 325a8b8
commit 52680cd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/br/pkg/task/stream.go b/br/pkg/task/stream.go
@@ -1269,6 +1269,9 @@ func restoreStream(
 	updateRewriteRules(rewriteRules, schemasReplace)
 
 	logFilesIter, err := client.LoadDMLFiles(ctx)
+	if err != nil {
+		return errors.Annotate(err, "failed to initialize the dml iterator")
+	}
 	pd := g.StartProgress(ctx, "Restore KV Files", int64(dataFileCount), !cfg.LogProgress)
 	err = withProgress(pd, func(p glue.Progress) error {
 		return client.RestoreKVFiles(ctx, rewriteRules, logFilesIter, cfg.PitrBatchCount, cfg.PitrBatchSize, updateStats, p.IncBy)