mirrors/WiiUDownloader
1
0
Fork 0
mirror of https://github.com/Xpl0itU/WiiUDownloader.git synced 2025-05-27 13:34:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use ctxio package for context aware downloads

Browse source
This commit is contained in:
Xpl0itU 2024-04-14 12:37:25 +02:00
parent 770d8e45f9
commit a30fb575b2
3 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
downloader.go
View file

@ -12,6 +12,7 @@ import (
"strings"
"time"
ctxio "github.com/jbenet/go-context/io"
"golang.org/x/sync/errgroup"
"golang.org/x/sync/semaphore"
)
@ -82,7 +83,9 @@ func downloadFileWithSemaphore(ctx context.Context, progressReporter ProgressRep
progressReporter.SetTotalDownloadedForFile(basePath, 0)
writerProgress := newWriterProgress(file, progressReporter, basePath)
_, err = io.Copy(writerProgress, resp.Body)
writerProgressWithContext := ctxio.NewWriter(ctx, writerProgress)
bodyReaderWithContext := ctxio.NewReader(ctx, resp.Body)
_, err = io.Copy(writerProgressWithContext, bodyReaderWithContext)
if err != nil {
file.Close()
resp.Body.Close()