rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: update cronjob

Browse Source
This commit is contained in:
Rogee
2025-03-15 21:23:41 +08:00
parent b7fcf2c3e6
commit aeae94a7d2
4 changed files with 30 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
templates/project/providers/job/provider.go.tpl
View File

@@ -113,19 +113,23 @@ func (q *Job) StopAndCancel(ctx context.Context) error {
return client.StopAndCancel(ctx)
}
func (q *Job) AddPeriodicJobs(job contracts.CronJob) (rivertype.PeriodicJobHandle, error) {
func (q *Job) AddPeriodicJobs(job contracts.CronJob) (map[string]rivertype.PeriodicJobHandle, error) {
client, err := q.Client()
if err != nil {
return 0, err
return nil, err
}
return client.PeriodicJobs().Add(river.NewPeriodicJob(
job.Periodic(),
func() (river.JobArgs, *river.InsertOpts) {
return job.JobArgs(), lo.ToPtr(job.InsertOpts())
},
&river.PeriodicJobOpts{
RunOnStart: job.RunOnStart(),
},
)), nil
handles := make(map[string]rivertype.PeriodicJobHandle)
for _, job := range job.Args() {
handles[job.Kind] = client.PeriodicJobs().Add(river.NewPeriodicJob(
job.PeriodicInterval,
func() (river.JobArgs, *river.InsertOpts) {
return job.Arg, lo.ToPtr(job.Arg.InsertOpts())
},
&river.PeriodicJobOpts{
RunOnStart: job.RunOnStart,
},
))
}
return handles, nil
}