first cut

This commit is contained in:
Binaek Sarkar
2022-01-25 14:42:02 +05:30
committed by Ken Hibino
parent 8bd70c6f84
commit d7ceb0c090
3 changed files with 21 additions and 3 deletions

View File

@@ -26,7 +26,8 @@ type processor struct {
logger *log.Logger
broker base.Broker
handler Handler
handler Handler
baseContextFn asynqcontext.BaseContext
queueConfig map[string]int
@@ -71,6 +72,7 @@ type processor struct {
type processorParams struct {
logger *log.Logger
broker base.Broker
baseCtxFn asynqcontext.BaseContext
retryDelayFunc RetryDelayFunc
isFailureFunc func(error) bool
syncCh chan<- *syncRequest
@@ -94,6 +96,7 @@ func newProcessor(params processorParams) *processor {
return &processor{
logger: params.logger,
broker: params.broker,
baseContextFn: params.baseCtxFn,
queueConfig: queues,
orderedQueues: orderedQueues,
retryDelayFunc: params.retryDelayFunc,
@@ -190,7 +193,7 @@ func (p *processor) exec() {
<-p.sema // release token
}()
ctx, cancel := asynqcontext.New(msg, deadline)
ctx, cancel := asynqcontext.New(p.baseContextFn(), msg, deadline)
p.cancelations.Add(msg.ID, cancel)
defer func() {
cancel()