From 4a5c1f262472f65b960461efd04888245e287e62 Mon Sep 17 00:00:00 2001 From: Rogee Date: Mon, 30 Dec 2024 17:38:50 +0800 Subject: [PATCH] feat: fix issues --- templates/project/app/jobs/demo_cron.go.tpl | 48 +++++++++++++++++++++ templates/project/app/jobs/demo_job.go.tpl | 2 +- 2 files changed, 49 insertions(+), 1 deletion(-) create mode 100644 templates/project/app/jobs/demo_cron.go.tpl diff --git a/templates/project/app/jobs/demo_cron.go.tpl b/templates/project/app/jobs/demo_cron.go.tpl new file mode 100644 index 0000000..684fab8 --- /dev/null +++ b/templates/project/app/jobs/demo_cron.go.tpl @@ -0,0 +1,48 @@ +package jobs + +import ( + "time" + + _ "git.ipao.vip/rogeecn/atom" + "git.ipao.vip/rogeecn/atom/contracts" + "github.com/riverqueue/river" + "github.com/sirupsen/logrus" +) + +var _ contracts.CronJob = (*CronJob)(nil) + +// @provider contracts.CronJob atom.GroupCronJob +type CronJob struct { + log *logrus.Entry `inject:"false"` +} + +func (cron *CronJob) Prepare() error { + cron.log = logrus.WithField("module", "cron") + return nil +} + +func (cron *CronJob) Description() string { + return "hello world cron job" +} + +// InsertOpts implements contracts.CronJob. +func (cron *CronJob) InsertOpts() *river.InsertOpts { + return nil +} + +// JobArgs implements contracts.CronJob. +func (cron *CronJob) JobArgs() river.JobArgs { + return SortArgs{ + Strings: []string{"a", "c", "b", "d"}, + } +} + +// Periodic implements contracts.CronJob. +func (cron *CronJob) Periodic() time.Duration { + return time.Second * 10 +} + +// RunOnStart implements contracts.CronJob. +func (cron *CronJob) RunOnStart() bool { + return true +} diff --git a/templates/project/app/jobs/demo_job.go.tpl b/templates/project/app/jobs/demo_job.go.tpl index f94edc0..f10aa71 100644 --- a/templates/project/app/jobs/demo_job.go.tpl +++ b/templates/project/app/jobs/demo_job.go.tpl @@ -25,7 +25,7 @@ type SortArgs struct { // InsertOpts implements JobArgsWithInsertOpts. func (s SortArgs) InsertOpts() InsertOpts { return InsertOpts{ - Queue: "high_priority", + Queue: QueueDefault, Priority: PriorityDefault, } }