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

Make Launcher thread safe

Browse Source
This commit is contained in:
Ken Hibino 2019-11-19 07:46:04 -08:00
parent e238d3835d
commit eb6a5032f6
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
asynq.go
View File

@ -15,6 +15,7 @@ import (
"math" "math"
"math/rand" "math/rand"
"strconv" "strconv"
"sync"
"time" "time"
"github.com/go-redis/redis/v7" "github.com/go-redis/redis/v7"
@ -105,8 +106,9 @@ func (c *Client) enqueue(msg *taskMessage, executeAt time.Time) error {
type Launcher struct { type Launcher struct {
rdb *redis.Client rdb *redis.Client
// running indicates whether the workes are currently running. // running indicates whether manager and poller are both running.
running bool running bool
mu sync.Mutex
poller *poller poller *poller
@ -135,6 +137,8 @@ type TaskHandler func(*Task) error
// Start starts the manager and poller. // Start starts the manager and poller.
func (l *Launcher) Start(handler TaskHandler) { func (l *Launcher) Start(handler TaskHandler) {
l.mu.Lock()
defer l.mu.Unlock()
if l.running { if l.running {
return return
} }
@ -147,6 +151,8 @@ func (l *Launcher) Start(handler TaskHandler) {
// Stop stops both manager and poller. // Stop stops both manager and poller.
func (l *Launcher) Stop() { func (l *Launcher) Stop() {
l.mu.Lock()
defer l.mu.Unlock()
if !l.running { if !l.running {
return return
} }