54 lines
1.1 KiB
Go
Executable File
54 lines
1.1 KiB
Go
Executable File
package jobs
|
|
|
|
import (
|
|
"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 := &DemoCronJob{}
|
|
if err := obj.Prepare(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
container.Later(func() error { return __job.AddPeriodicJobs(obj) })
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
if err := container.Container.Provide(func(
|
|
__job *job.Job,
|
|
) (contracts.Initial, error) {
|
|
obj := &DemoJobWorker{}
|
|
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 := &OrderRefundJobWorker{}
|
|
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return obj, nil
|
|
}, atom.GroupInitial); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|