go-pkg/asynq
2
0
Fork 0
mirror of https://github.com/hibiken/asynq.git synced 2024-11-10 11:31:58 +08:00
Code Issues Projects Releases Wiki Activity

Fix bug around releasing semaphore token

Browse Source
This commit is contained in:
Ken Hibino 2019-11-18 07:42:26 -08:00
parent c6f482d4f8
commit 3daef02632
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
asynq.go
View File

@ -154,9 +154,10 @@ func (w *Workers) Run(handler TaskHandler) {
log.Printf("[Servere Error] could not parse json encoded message %s: %v", data, err)
continue
}
w.poolTokens <- struct{}{} // acquire a token
t := &Task{Type: msg.Type, Payload: msg.Payload}
w.poolTokens <- struct{}{} // acquire token
go func(task *Task) {
defer func() { <-w.poolTokens }() // release token
err := handler(task)
if err != nil {
if msg.Retried >= msg.Retry {
@ -178,7 +179,6 @@ func (w *Workers) Run(handler TaskHandler) {
}
}
<-w.poolTokens // release the token
}(t)
}
}