feat: fix jobs

This commit is contained in:
yanghao05
2025-04-22 14:49:22 +08:00
parent b1f090918d
commit 202e94c020

View File

@@ -2,7 +2,6 @@ package jobs
import ( import (
"context" "context"
"errors"
"os" "os"
"path/filepath" "path/filepath"
"time" "time"
@@ -75,7 +74,7 @@ func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFr
} }
} else { } else {
if st.Size() == media.Size { if st.Size() == media.Size {
return JobCancel(errors.New("file already downloaded")) return w.NextJob(media.Hash)
} else { } else {
// remove file // remove file
if err := os.Remove(dst); err != nil { if err := os.Remove(dst); err != nil {
@@ -93,15 +92,18 @@ func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFr
log.Infof("Successfully downloaded file: %s", media.Path) log.Infof("Successfully downloaded file: %s", media.Path)
if err := w.job.Add(&VideoCut{MediaHash: job.Args.MediaHash}); err != nil { return w.NextJob(media.Hash)
}
func (w *DownloadFromAliOSSWorker) NextJob(hash string) error {
if err := w.job.Add(&VideoCut{MediaHash: hash}); err != nil {
log.Errorf("Error adding job: %v", err) log.Errorf("Error adding job: %v", err)
return err return err
} }
if err := w.job.Add(&VideoExtractHeadImage{MediaHash: job.Args.MediaHash}); err != nil { if err := w.job.Add(&VideoExtractHeadImage{MediaHash: hash}); err != nil {
log.Errorf("Error adding job: %v", err) log.Errorf("Error adding job: %v", err)
return err return err
} }
return nil return nil
} }