Skip to content

Commit b1f6a0c

Browse files
committed
actually set timeout appropriately on boosted workers
Signed-off-by: Andrew Thornton <[email protected]>
1 parent 45ed06c commit b1f6a0c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

modules/queue/workerpool.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ func (p *WorkerPool) zeroBoost() {
9393
log.Warn("WorkerPool: %d (for %s) has zero workers - adding %d temporary workers for %s", p.qid, mq.Name, boost, p.boostTimeout)
9494

9595
start := time.Now()
96-
pid := mq.RegisterWorkers(boost, start, false, start, cancel, false)
96+
pid := mq.RegisterWorkers(boost, start, true, start.Add(p.boostTimeout), cancel, false)
9797
go func() {
9898
select {
9999
case <-ctx.Done():
@@ -150,7 +150,7 @@ func (p *WorkerPool) pushBoost(data Data) {
150150
log.Warn("WorkerPool: %d (for %s) Channel blocked for %v - adding %d temporary workers for %s, block timeout now %v", p.qid, mq.Name, ourTimeout, boost, p.boostTimeout, p.blockTimeout)
151151

152152
start := time.Now()
153-
pid := mq.RegisterWorkers(boost, start, false, start, cancel, false)
153+
pid := mq.RegisterWorkers(boost, start, true, start.Add(p.boostTimeout), cancel, false)
154154
go func() {
155155
<-ctx.Done()
156156
mq.RemoveWorkers(pid)

0 commit comments

Comments
 (0)