135 lines
2.7 KiB
Go
Executable File
135 lines
2.7 KiB
Go
Executable File
package jobs
|
|
|
|
import (
|
|
"quyun/v2/providers/ali"
|
|
"quyun/v2/providers/app"
|
|
"quyun/v2/providers/job"
|
|
|
|
"github.com/riverqueue/river"
|
|
"go.ipao.vip/atom"
|
|
"go.ipao.vip/atom/container"
|
|
"go.ipao.vip/atom/contracts"
|
|
"go.ipao.vip/atom/opt"
|
|
)
|
|
|
|
func Provide(opts ...opt.Option) error {
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
) (contracts.Initial, error) {
|
|
obj := &BalancePayNotifyWorker{}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
app *app.Config,
|
|
job *job.Job,
|
|
oss *ali.OSSClient,
|
|
) (contracts.Initial, error) {
|
|
obj := &DownloadFromAliOSSWorker{
|
|
app: app,
|
|
job: job,
|
|
oss: oss,
|
|
}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
app *app.Config,
|
|
job *job.Job,
|
|
oss *ali.OSSClient,
|
|
) (contracts.Initial, error) {
|
|
obj := &PublishDraftPostsWorker{
|
|
app: app,
|
|
job: job,
|
|
oss: oss,
|
|
}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
) (contracts.Initial, error) {
|
|
obj := &RemoveFileWorker{}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
app *app.Config,
|
|
job *job.Job,
|
|
) (contracts.Initial, error) {
|
|
obj := &VideoCutWorker{
|
|
app: app,
|
|
job: job,
|
|
}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
app *app.Config,
|
|
job *job.Job,
|
|
oss *ali.OSSClient,
|
|
) (contracts.Initial, error) {
|
|
obj := &VideoExtractHeadImageWorker{
|
|
app: app,
|
|
job: job,
|
|
oss: oss,
|
|
}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
app *app.Config,
|
|
job *job.Job,
|
|
oss *ali.OSSClient,
|
|
) (contracts.Initial, error) {
|
|
obj := &VideoStoreShortWorker{
|
|
app: app,
|
|
job: job,
|
|
oss: oss,
|
|
}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|