Merge pull request #3 from pconstantinou/bug/set-context

Set job in context before calling updates
pconstantinou · Mar 25, 2024 · f84f7cb · f84f7cb
2 parents 4a13c41 + 951f8ca
commit f84f7cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/backends/postgres/postgres_backend.go b/backends/postgres/postgres_backend.go
@@ -924,6 +924,9 @@ func (p *PgBackend) handleJob(ctx context.Context, jobID string) (err error) {
 		return
 	}
 
+	ctx = withJobContext(ctx, job)
+	ctx = context.WithValue(ctx, txCtxVarKey, tx)
+
 	if job.RunAfter.After(time.Now()) {
 		p.logger.Error("Running a job too early.",
 			slog.String("queue", job.Queue),
@@ -940,9 +943,6 @@ func (p *PgBackend) handleJob(ctx context.Context, jobID string) (err error) {
 		return
 	}
 
-	ctx = withJobContext(ctx, job)
-	ctx = context.WithValue(ctx, txCtxVarKey, tx)
-
 	// check if the job is being retried and increment retry count accordingly
 	if job.Status != internal.JobStatusNew {
 		job.Retries++