fix(ctx): consider ctx done

iyear · Sep 15, 2022 · 216ad34 · 216ad34
1 parent 7139f42
commit 216ad34
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/pkg/downloader/downloader.go b/pkg/downloader/downloader.go
@@ -88,6 +88,12 @@ func (d *Downloader) Download(ctx context.Context, limit int) error {
 }
 
 func (d *Downloader) download(ctx context.Context, item *Item) error {
+ select {
+ case <-ctx.Done():
+ return ctx.Err()
+ default:
+ }
+
  tracker := prog.AppendTracker(d.pw, formatter, item.Name, item.Size)
  filename := fmt.Sprintf("%s%s", utils.FS.GetNameWithoutExt(item.Name), TempExt)
  path := filepath.Join(consts.DownloadPath, filename)

diff --git a/pkg/uploader/uploader.go b/pkg/uploader/uploader.go
@@ -87,6 +87,12 @@ func (u *Uploader) upload(ctx context.Context, item *Item) error {
  _ = R.Close()
  }(item.R)
 
+ select {
+ case <-ctx.Done():
+ return ctx.Err()
+ default:
+ }
+
  tracker := prog.AppendTracker(u.pw, formatter, item.Name, item.Size)
 
  up := uploader.NewUploader(u.client).