br: add error handling for group context cancel when restore file is …

…corrupted (#30190) close #30135
pingcap · Dec 20, 2021 · e1fb2f5 · e1fb2f5
1 parent 24d970f
commit e1fb2f5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/br/pkg/restore/pipeline_items.go b/br/pkg/restore/pipeline_items.go
@@ -360,6 +360,7 @@ func (b *tikvSender) restoreWorker(ctx context.Context, ranges <-chan drainResul
 			eg.Go(func() error {
 				e := b.client.RestoreFiles(ectx, files, r.result.RewriteRules, b.updateCh)
 				if e != nil {
+					r.done()
 					return e
 				}
 				log.Info("restore batch done", rtree.ZapRanges(r.result.Ranges))