985722b102
Previously the global context was either accessed via gopts.ctx, stored in a local variable and then used within that function or sometimes both. This makes it very hard to follow which ctx or a wrapped version of it reaches which method. Thus just drop the context from the globalOptions struct and pass it explicitly to every command line handler method.
68 lines
2 KiB
Go
68 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/restic/restic/internal/restic"
|
|
)
|
|
|
|
// DeleteFiles deletes the given fileList of fileType in parallel
|
|
// it will print a warning if there is an error, but continue deleting the remaining files
|
|
func DeleteFiles(ctx context.Context, gopts GlobalOptions, repo restic.Repository, fileList restic.IDSet, fileType restic.FileType) {
|
|
_ = deleteFiles(ctx, gopts, true, repo, fileList, fileType)
|
|
}
|
|
|
|
// DeleteFilesChecked deletes the given fileList of fileType in parallel
|
|
// if an error occurs, it will cancel and return this error
|
|
func DeleteFilesChecked(ctx context.Context, gopts GlobalOptions, repo restic.Repository, fileList restic.IDSet, fileType restic.FileType) error {
|
|
return deleteFiles(ctx, gopts, false, repo, fileList, fileType)
|
|
}
|
|
|
|
// deleteFiles deletes the given fileList of fileType in parallel
|
|
// if ignoreError=true, it will print a warning if there was an error, else it will abort.
|
|
func deleteFiles(ctx context.Context, gopts GlobalOptions, ignoreError bool, repo restic.Repository, fileList restic.IDSet, fileType restic.FileType) error {
|
|
totalCount := len(fileList)
|
|
fileChan := make(chan restic.ID)
|
|
wg, ctx := errgroup.WithContext(ctx)
|
|
wg.Go(func() error {
|
|
defer close(fileChan)
|
|
for id := range fileList {
|
|
select {
|
|
case fileChan <- id:
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
|
|
bar := newProgressMax(!gopts.JSON && !gopts.Quiet, uint64(totalCount), "files deleted")
|
|
defer bar.Done()
|
|
// deleting files is IO-bound
|
|
workerCount := repo.Connections()
|
|
for i := 0; i < int(workerCount); i++ {
|
|
wg.Go(func() error {
|
|
for id := range fileChan {
|
|
h := restic.Handle{Type: fileType, Name: id.String()}
|
|
err := repo.Backend().Remove(ctx, h)
|
|
if err != nil {
|
|
if !gopts.JSON {
|
|
Warnf("unable to remove %v from the repository\n", h)
|
|
}
|
|
if !ignoreError {
|
|
return err
|
|
}
|
|
}
|
|
if !gopts.JSON && gopts.verbosity > 2 {
|
|
Verbosef("removed %v\n", h)
|
|
}
|
|
bar.Add(1)
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
err := wg.Wait()
|
|
return err
|
|
}
|