me/neoq
1
0
Fork 0
mirror of https://github.com/acaloiaro/neoq synced 2026-07-21 18:29:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add public jobs.WithJobContext()

Browse source
This commit is contained in:
Adriano Caloiaro 2024-06-07 07:17:55 -07:00 committed by Adriano Caloiaro
parent 37b6732ad0
commit a6f7fbcd36
4 changed files with 8 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
backends/memory/memory_backend.go
View file

@ -321,7 +321,7 @@ func (m *MemBackend) scheduleFutureJobs(ctx context.Context) {
} }
func (m *MemBackend) handleJob(ctx context.Context, job *jobs.Job, h handler.Handler) (err error) { func (m *MemBackend) handleJob(ctx context.Context, job *jobs.Job, h handler.Handler) (err error) {
ctx = withJobContext(ctx, job) ctx = jobs.WithJobContext(ctx, job)
m.logger.Debug( m.logger.Debug(
"handling job", "handling job",
@ -381,8 +381,3 @@ func (m *MemBackend) removeFutureJob(jobID int64) {
m.futureJobs.Delete(job.ID) m.futureJobs.Delete(job.ID)
} }
} }
// withJobContext creates a new context with the Job set
func withJobContext(ctx context.Context, j *jobs.Job) context.Context {
return context.WithValue(ctx, internal.JobCtxVarKey, j)
}

7
backends/postgres/postgres_backend.go
View file

@ -885,7 +885,7 @@ func (p *PgBackend) handleJob(ctx context.Context, jobID string) (err error) {
return return
} }
ctx = withJobContext(ctx, job) ctx = jobs.WithJobContext(ctx, job)
ctx = context.WithValue(ctx, txCtxVarKey, tx) ctx = context.WithValue(ctx, txCtxVarKey, tx)
if job.Deadline != nil && job.Deadline.Before(time.Now().UTC()) { if job.Deadline != nil && job.Deadline.Before(time.Now().UTC()) {
@ -1068,11 +1068,6 @@ func (p *PgBackend) acquire(ctx context.Context) (conn *pgxpool.Conn, err error)
} }
} }
// withJobContext creates a new context with the Job set
func withJobContext(ctx context.Context, j *jobs.Job) context.Context {
return context.WithValue(ctx, internal.JobCtxVarKey, j)
}
func GetPQConnectionString(connectionString string) (string, error) { func GetPQConnectionString(connectionString string) (string, error) {
pgxCfg, err := pgx.ParseConfig(connectionString) pgxCfg, err := pgx.ParseConfig(connectionString)
if err != nil { if err != nil {

7
backends/redis/redis_backend.go
View file

@ -242,7 +242,7 @@ func (b *RedisBackend) Start(_ context.Context, h handler.Handler) (err error) {
RunAfter: ti.NextProcessAt, RunAfter: ti.NextProcessAt,
} }
ctx = withJobContext(ctx, job) ctx = jobs.WithJobContext(ctx, job)
err = handler.Exec(ctx, h) err = handler.Exec(ctx, h)
if err != nil { if err != nil {
b.logger.Error("error handling job", slog.Any("error", err)) b.logger.Error("error handling job", slog.Any("error", err))
@ -349,8 +349,3 @@ func (b *RedisBackend) Shutdown(_ context.Context) {
b.client.Close() b.client.Close()
b.server.Shutdown() b.server.Shutdown()
} }
// withJobContext creates a new context with the Job set
func withJobContext(ctx context.Context, j *jobs.Job) context.Context {
return context.WithValue(ctx, internal.JobCtxVarKey, j)
}

5
jobs/jobs.go
View file

@ -75,6 +75,11 @@ func FingerprintJob(j *Job) (err error) {
return return
} }
// WithJobContext adds a job to the provided context
func WithJobContext(ctx context.Context, job *Job) context.Context {
return context.WithValue(ctx, internal.JobCtxVarKey, job)
}
// FromContext fetches the job from a context if the job context variable is set // FromContext fetches the job from a context if the job context variable is set
func FromContext(ctx context.Context) (j *Job, err error) { func FromContext(ctx context.Context) (j *Job, err error) {
var ok bool var ok bool