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

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

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() {