From 022597d2f92d43f3a0a382db9c779a66e9f6c76f Mon Sep 17 00:00:00 2001 From: rogeecn Date: Mon, 17 Mar 2025 11:48:47 +0800 Subject: [PATCH] feat: update job --- templates/project/providers/job/provider.go.tpl | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/templates/project/providers/job/provider.go.tpl b/templates/project/providers/job/provider.go.tpl index 866c2f6..4c8f4e0 100644 --- a/templates/project/providers/job/provider.go.tpl +++ b/templates/project/providers/job/provider.go.tpl @@ -40,7 +40,7 @@ func Provide(opts ...opt.Option) error { container.AddCloseAble(dbPool.Close) pool := riverpgxv5.New(dbPool) - queue := &Job{Workers: workers, Driver: pool, ctx: ctx} + queue := &Job{Workers: workers, driver: pool, ctx: ctx, periodicJobs: make(map[string]rivertype.PeriodicJobHandle), jobs: make(map[string]*rivertype.JobInsertResult)} container.AddCloseAble(queue.Close) return queue, nil @@ -48,11 +48,13 @@ func Provide(opts ...opt.Option) error { } type Job struct { - ctx context.Context Workers *river.Workers - Driver *riverpgxv5.Driver - l sync.Mutex + l sync.Mutex + + ctx context.Context + driver *riverpgxv5.Driver + client *river.Client[pgx.Tx] periodicJobs map[string]rivertype.PeriodicJobHandle jobs map[string]*rivertype.JobInsertResult