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

…corrupted (#30190) close pingcap/tidb#30135
pingcap · Dec 20, 2021 · c069bc0 · c069bc0
1 parent 1427691
commit c069bc0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/restore/pipeline_items.go b/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))