GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-08-08 14:03:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve error handling during file download in stream mode

Browse Source
This commit is contained in:
krau
2025-07-28 14:37:08 +08:00
parent 9e5042bda1
commit 03907f2d32
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
core/batchtftask/execute.go
View File

@@ -61,10 +61,12 @@ func (t *Task) processElement(ctx context.Context, elem TaskElement) error {
t.Progress.OnProgress(ctx, t) t.Progress.OnProgress(ctx, t)
}) })
errg.Go(func() error { errg.Go(func() error {
defer pw.Close()
logger.Info("Starting file download in stream mode") logger.Info("Starting file download in stream mode")
_, err := tfile.NewDownloader(elem.File).Stream(uploadCtx, wr) _, err := tfile.NewDownloader(elem.File).Stream(uploadCtx, wr)
if closeErr := pw.CloseWithError(err); closeErr != nil { if err != nil {
logger.Errorf("Failed to close pipe writer: %v", closeErr) logger.Errorf("Failed to download file: %v", err)
pw.CloseWithError(err)
} }
return err return err
}) })

6
core/tftask/stream.go
View File

@@ -21,10 +21,12 @@ func executeStream(ctx context.Context, task *Task) error {
}) })
wr := newWriter(ctx, pw, task.Progress, task) wr := newWriter(ctx, pw, task.Progress, task)
errg.Go(func() error { errg.Go(func() error {
defer pw.Close()
logger.Info("Starting file download in stream mode") logger.Info("Starting file download in stream mode")
_, err := tfile.NewDownloader(task.File).Stream(uploadCtx, wr) _, err := tfile.NewDownloader(task.File).Stream(uploadCtx, wr)
if closeErr := pw.CloseWithError(err); closeErr != nil { if err != nil {
logger.Errorf("Failed to close pipe writer: %v", closeErr) logger.Errorf("Failed to download file: %v", err)
pw.CloseWithError(err)
} }
return err return err
}) })