feat: add download from alioss job

This commit is contained in:
yanghao05
2025-04-17 20:42:57 +08:00
parent d961c1a4a5
commit 9323fb4230
6 changed files with 129 additions and 14 deletions

View File

@@ -2,8 +2,12 @@ package jobs
import (
"context"
"os"
"path/filepath"
"time"
"quyun/providers/ali"
. "github.com/riverqueue/river"
log "github.com/sirupsen/logrus"
_ "go.ipao.vip/atom"
@@ -14,8 +18,7 @@ import (
var _ contracts.JobArgs = (*WechatCallback)(nil)
type DownloadFromAliOSS struct {
Bucket string
Path string
Path string
}
func (s DownloadFromAliOSS) InsertOpts() InsertOpts {
@@ -33,6 +36,12 @@ var _ Worker[DownloadFromAliOSS] = (*DownloadFromAliOSSWorker)(nil)
// @provider(job)
type DownloadFromAliOSSWorker struct {
WorkerDefaults[DownloadFromAliOSS]
oss *ali.OSSClient
}
func (w *DownloadFromAliOSSWorker) NextRetry(job *Job[DownloadFromAliOSS]) time.Time {
return time.Now().Add(30 * time.Second)
}
func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFromAliOSS]) error {
@@ -41,9 +50,37 @@ func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFr
log.Infof("[Start] Working on job with strings: %+v", job.Args)
defer log.Infof("[End] Finished %s", job.Args.Kind())
dst := filepath.Join("/Users/rogee/Projects/self/quyun/backend/fixtures/oss/", job.Args.Path)
// check is path exist
_, err := os.Stat(dst)
if os.IsNotExist(err) {
log.Infof("File not exists: %s", dst)
// midir
err := os.MkdirAll(filepath.Dir(dst), os.ModePerm)
if err != nil {
log.Errorf("Error creating directory: %v", err)
return err
}
}
if err == nil {
log.Infof("File already exists: %s", dst)
err := os.Remove(dst)
if err != nil {
log.Errorf("Error removing file: %v", err)
return err
}
return nil
}
if err := w.oss.Download(ctx, job.Args.Path, dst); err != nil {
log.Errorf("Error downloading file: %v", err)
return err
}
log.Infof("Successfully downloaded file: %s", job.Args.Path)
return nil
}
func (w *DownloadFromAliOSSWorker) NextRetry(job *Job[DownloadFromAliOSS]) time.Time {
return time.Now().Add(30 * time.Second)
}