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

feat: update job

Browse Source
This commit is contained in:
rogeecn
2025-03-17 11:48:47 +08:00
parent 60b5539061
commit 022597d2f9
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -40,7 +40,7 @@ func Provide(opts ...opt.Option) error {
container.AddCloseAble(dbPool.Close) container.AddCloseAble(dbPool.Close)
pool := riverpgxv5.New(dbPool) 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) container.AddCloseAble(queue.Close)
return queue, nil return queue, nil
@@ -48,11 +48,13 @@ func Provide(opts ...opt.Option) error {
} }
type Job struct { type Job struct {
ctx context.Context
Workers *river.Workers Workers *river.Workers
Driver *riverpgxv5.Driver
l sync.Mutex l sync.Mutex
ctx context.Context
driver *riverpgxv5.Driver
client *river.Client[pgx.Tx] client *river.Client[pgx.Tx]
periodicJobs map[string]rivertype.PeriodicJobHandle periodicJobs map[string]rivertype.PeriodicJobHandle
jobs map[string]*rivertype.JobInsertResult jobs map[string]*rivertype.JobInsertResult