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

Make sure to invoke CancelFunc in all cases

Browse Source
This commit is contained in:
Ken Hibino 2020-05-03 13:58:54 -07:00
parent 1c1474c55c
commit 24f2b64c6c
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
processor.go
View File

@ -188,15 +188,17 @@ func (p *processor) exec() {
<-p.sema /* release token */ <-p.sema /* release token */
}() }()
resCh := make(chan error, 1)
task := NewTask(msg.Type, msg.Payload)
ctx, cancel := createContext(msg) ctx, cancel := createContext(msg)
p.cancelations.Add(msg.ID.String(), cancel) p.cancelations.Add(msg.ID.String(), cancel)
go func() { defer func() {
resCh <- perform(ctx, task, p.handler) cancel()
p.cancelations.Delete(msg.ID.String()) p.cancelations.Delete(msg.ID.String())
}() }()
resCh := make(chan error, 1)
task := NewTask(msg.Type, msg.Payload)
go func() { resCh <- perform(ctx, task, p.handler) }()
select { select {
case <-p.quit: case <-p.quit:
// time is up, quit this worker goroutine. // time is up, quit this worker goroutine.