bit1001/tdl
1
1
Fork 0
mirror of https://github.com/bit1001/tdl.git synced 2025-01-06 11:56:41 +08:00
Code Issues Projects Releases Packages Wiki Activity

fix(downloader): write files concurrently

Browse source
This commit is contained in:
iyear 2022-09-04 11:02:50 +08:00
parent f85aa491cb
commit 8b8c4ecb7a
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pkg/downloader/write_at.go
View file

@ -3,15 +3,20 @@ package downloader
import ( import (
"github.com/jedib0t/go-pretty/v6/progress" "github.com/jedib0t/go-pretty/v6/progress"
"os" "os"
"sync"
) )
// writeAt wrapper for file to use progress bar // writeAt wrapper for file to use progress bar
type writeAt struct { type writeAt struct {
mu sync.Mutex
f *os.File f *os.File
tracker *progress.Tracker tracker *progress.Tracker
} }
func (w *writeAt) WriteAt(p []byte, off int64) (int, error) { func (w *writeAt) WriteAt(p []byte, off int64) (int, error) {
w.mu.Lock()
defer w.mu.Unlock()
at, err := w.f.WriteAt(p, off) at, err := w.f.WriteAt(p, off)
if err != nil { if err != nil {
w.tracker.MarkAsErrored() w.tracker.MarkAsErrored()