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

feat: add context cancellation handling in saveFileWithRetry function

Browse Source
This commit is contained in:
krau
2025-02-27 22:07:41 +08:00
parent be6444cf96
commit 7015081a84
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/utils.go
+8
View File
@@ -21,11 +21,19 @@ import (
func saveFileWithRetry(task *types.Task, taskStorage storage.Storage, localFilePath string) error { func saveFileWithRetry(task *types.Task, taskStorage storage.Storage, localFilePath string) error {
for i := 0; i <= config.Cfg.Retry; i++ { for i := 0; i <= config.Cfg.Retry; i++ {
if err := task.Ctx.Err(); err != nil {
return fmt.Errorf("context canceled while saving file: %w", err)
}
if err := taskStorage.Save(task.Ctx, localFilePath, task.StoragePath); err != nil { if err := taskStorage.Save(task.Ctx, localFilePath, task.StoragePath); err != nil {
if i == config.Cfg.Retry { if i == config.Cfg.Retry {
return fmt.Errorf("failed to save file: %w", err) return fmt.Errorf("failed to save file: %w", err)
} }
logger.L.Errorf("Failed to save file: %s, retrying...", err) logger.L.Errorf("Failed to save file: %s, retrying...", err)
select {
case <-task.Ctx.Done():
return fmt.Errorf("context canceled during retry delay: %w", task.Ctx.Err())
case <-time.After(time.Duration(i*500) * time.Millisecond):
}
continue continue
} }
return nil return nil