78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package jobs
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"quyun/app/models"
|
|
"quyun/providers/ali"
|
|
"quyun/providers/job"
|
|
|
|
. "github.com/riverqueue/river"
|
|
log "github.com/sirupsen/logrus"
|
|
_ "go.ipao.vip/atom"
|
|
"go.ipao.vip/atom/contracts"
|
|
)
|
|
|
|
var _ contracts.JobArgs = (*WechatCallback)(nil)
|
|
|
|
type ExtractHeadImageFromVideo struct {
|
|
MediaID int64 `json:"media_id"`
|
|
}
|
|
|
|
func (s ExtractHeadImageFromVideo) InsertOpts() InsertOpts {
|
|
return InsertOpts{
|
|
Queue: QueueDefault,
|
|
Priority: PriorityDefault,
|
|
}
|
|
}
|
|
|
|
func (s ExtractHeadImageFromVideo) Kind() string { return "extract_head_image_from_video" }
|
|
func (a ExtractHeadImageFromVideo) UniqueID() string { return a.Kind() }
|
|
|
|
var _ Worker[ExtractHeadImageFromVideo] = (*ExtractHeadImageFromVideoWorker)(nil)
|
|
|
|
// @provider(job)
|
|
type ExtractHeadImageFromVideoWorker struct {
|
|
WorkerDefaults[ExtractHeadImageFromVideo]
|
|
|
|
oss *ali.OSSClient
|
|
job *job.Job
|
|
}
|
|
|
|
func (w *ExtractHeadImageFromVideoWorker) NextRetry(job *Job[ExtractHeadImageFromVideo]) time.Time {
|
|
return time.Now().Add(30 * time.Second)
|
|
}
|
|
|
|
func (w *ExtractHeadImageFromVideoWorker) Work(ctx context.Context, job *Job[ExtractHeadImageFromVideo]) error {
|
|
log := log.WithField("job", job.Args.Kind())
|
|
|
|
log.Infof("[Start] Working on job with strings: %+v", job.Args)
|
|
defer log.Infof("[End] Finished %s", job.Args.Kind())
|
|
|
|
media, err := models.Medias.GetByID(ctx, job.Args.MediaID)
|
|
if err != nil {
|
|
log.Errorf("Error getting media by ID: %v", err)
|
|
return JobCancel(err)
|
|
}
|
|
_ = media
|
|
|
|
// TODO
|
|
|
|
// path := "/Users/rogee/Projects/self/quyun/backend/fixtures/oss/"
|
|
// dst := filepath.Join(path, media.Path)
|
|
|
|
// // use ffmpeg to extract audio from video
|
|
// audioPath := filepath.Join(path, media.Hash+".mp3")
|
|
|
|
// cmd := exec.Command("ffmpeg", "-i", dst, audioPath)
|
|
// if err := cmd.Run(); err != nil {
|
|
// log.Errorf("Error extracting audio: %v", err)
|
|
// return err
|
|
// }
|
|
|
|
// log.Infof("Successfully extracted audio to: %s", audioPath)
|
|
|
|
return nil
|
|
}
|