2
0
mirror of https://github.com/hibiken/asynq.git synced 2026-05-01 17:35:51 +08:00

Move client and launcher to its own files

This commit is contained in:
Ken Hibino
2019-11-19 21:19:46 -08:00
parent 85a04cbabb
commit e9069bfb47
3 changed files with 98 additions and 92 deletions

61
launcher.go Normal file
View File

@@ -0,0 +1,61 @@
package asynq
import (
"sync"
"time"
"github.com/go-redis/redis/v7"
)
// Launcher starts the manager and poller.
type Launcher struct {
// running indicates whether manager and poller are both running.
running bool
mu sync.Mutex
poller *poller
manager *manager
}
// NewLauncher creates and returns a new Launcher.
func NewLauncher(poolSize int, opt *RedisOpt) *Launcher {
client := redis.NewClient(&redis.Options{Addr: opt.Addr, Password: opt.Password})
rdb := newRDB(client)
poller := newPoller(rdb, 5*time.Second, []string{scheduled, retry})
manager := newManager(rdb, poolSize, nil)
return &Launcher{
poller: poller,
manager: manager,
}
}
// TaskHandler handles a given task and report any error.
type TaskHandler func(*Task) error
// Start starts the manager and poller.
func (l *Launcher) Start(handler TaskHandler) {
l.mu.Lock()
defer l.mu.Unlock()
if l.running {
return
}
l.running = true
l.manager.handler = handler
l.poller.start()
l.manager.start()
}
// Stop stops both manager and poller.
func (l *Launcher) Stop() {
l.mu.Lock()
defer l.mu.Unlock()
if !l.running {
return
}
l.running = false
l.poller.terminate()
l.manager.terminate()
}