ProjectHentai/alist
6
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(task): allow retry canceled (#7852)

Browse Source
This commit is contained in:
KirCute_ECT
2025-01-27 20:18:10 +08:00
committed by GitHub
parent 23f3178f39
commit d5ec998699
6 changed files with 31 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/fs/archive.go
View File

@ -50,6 +50,7 @@ func (t *ArchiveDownloadTask) GetStatus() string {
}
func (t *ArchiveDownloadTask) Run() error {
t.ReinitCtx()
t.ClearEndTime()
t.SetStartTime(time.Now())
defer func() { t.SetEndTime(time.Now()) }()
@ -144,6 +145,7 @@ func (t *ArchiveContentUploadTask) GetStatus() string {
}
func (t *ArchiveContentUploadTask) Run() error {
t.ReinitCtx()
t.ClearEndTime()
t.SetStartTime(time.Now())
defer func() { t.SetEndTime(time.Now()) }()
@ -235,7 +237,9 @@ func (t *ArchiveContentUploadTask) RunWithNextTaskCallback(f func(nextTsk *Archi
func (t *ArchiveContentUploadTask) Cancel() {
t.TaskExtension.Cancel()
t.deleteSrcFile()
if !conf.Conf.Tasks.AllowRetryCanceled {
t.deleteSrcFile()
}
}
func (t *ArchiveContentUploadTask) deleteSrcFile() {

1
internal/fs/copy.go
View File

@ -39,6 +39,7 @@ func (t *CopyTask) GetStatus() string {
}
func (t *CopyTask) Run() error {
t.ReinitCtx()
t.ClearEndTime()
t.SetStartTime(time.Now())
defer func() { t.SetEndTime(time.Now()) }()