go-pkg/asynq
2
0
Fork 0
mirror of https://github.com/hibiken/asynq.git synced 2024-12-27 00:02:19 +08:00
Code Issues Projects Releases Wiki Activity

Make Run method stateful

Browse Source
This commit is contained in:
Ken Hibino 2019-11-17 21:21:32 -08:00
parent 62db9863fb
commit c6f482d4f8
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
asynq.go
View File

@ -109,6 +109,9 @@ type Workers struct {
// poolTokens is a counting semaphore to ensure the number of active workers // poolTokens is a counting semaphore to ensure the number of active workers
// does not exceed the limit. // does not exceed the limit.
poolTokens chan struct{} poolTokens chan struct{}
// running indicates whether the workes are currently running.
running bool
} }
// NewWorkers creates and returns a new Workers. // NewWorkers creates and returns a new Workers.
@ -125,6 +128,11 @@ type TaskHandler func(*Task) error
// Run starts the workers and scheduler with a given handler. // Run starts the workers and scheduler with a given handler.
func (w *Workers) Run(handler TaskHandler) { func (w *Workers) Run(handler TaskHandler) {
if w.running {
return
}
w.running = true
go w.pollDeferred() go w.pollDeferred()
for { for {