actually set timeout appropriately on boosted workers

Signed-off-by: Andrew Thornton <art27@cantab.net>
zeripath · May 1, 2021 · b1f6a0c · b1f6a0c
1 parent 45ed06c
commit b1f6a0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/queue/workerpool.go b/modules/queue/workerpool.go
@@ -93,7 +93,7 @@ func (p *WorkerPool) zeroBoost() {
 		log.Warn("WorkerPool: %d (for %s) has zero workers - adding %d temporary workers for %s", p.qid, mq.Name, boost, p.boostTimeout)
 
 		start := time.Now()
-		pid := mq.RegisterWorkers(boost, start, false, start, cancel, false)
+		pid := mq.RegisterWorkers(boost, start, true, start.Add(p.boostTimeout), cancel, false)
 		go func() {
 			select {
 			case <-ctx.Done():
@@ -150,7 +150,7 @@ func (p *WorkerPool) pushBoost(data Data) {
 				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)
 
 				start := time.Now()
-				pid := mq.RegisterWorkers(boost, start, false, start, cancel, false)
+				pid := mq.RegisterWorkers(boost, start, true, start.Add(p.boostTimeout), cancel, false)
 				go func() {
 					<-ctx.Done()
 					mq.RemoveWorkers(pid)