2
0
mirror of https://github.com/hibiken/asynq.git synced 2024-11-14 11:31:18 +08:00

Add DB field to RedisOpt to specify redis db index

This commit is contained in:
Ken Hibino 2019-11-24 18:41:55 -08:00
parent d5c2b9b995
commit f91004e6aa
3 changed files with 13 additions and 2 deletions

View File

@ -54,4 +54,7 @@ type taskMessage struct {
type RedisOpt struct { type RedisOpt struct {
Addr string Addr string
Password string Password string
// DB specifies which redis database to select.
DB int
} }

View File

@ -20,7 +20,11 @@ type Background struct {
// NewBackground returns a new Background instance. // NewBackground returns a new Background instance.
func NewBackground(numWorkers int, opt *RedisOpt) *Background { func NewBackground(numWorkers int, opt *RedisOpt) *Background {
client := redis.NewClient(&redis.Options{Addr: opt.Addr, Password: opt.Password}) client := redis.NewClient(&redis.Options{
Addr: opt.Addr,
Password: opt.Password,
DB: opt.DB,
})
rdb := newRDB(client) rdb := newRDB(client)
poller := newPoller(rdb, 5*time.Second, []string{scheduled, retry}) poller := newPoller(rdb, 5*time.Second, []string{scheduled, retry})
processor := newProcessor(rdb, numWorkers, nil) processor := newProcessor(rdb, numWorkers, nil)

View File

@ -14,7 +14,11 @@ type Client struct {
// NewClient creates and returns a new client. // NewClient creates and returns a new client.
func NewClient(opt *RedisOpt) *Client { func NewClient(opt *RedisOpt) *Client {
client := redis.NewClient(&redis.Options{Addr: opt.Addr, Password: opt.Password}) client := redis.NewClient(&redis.Options{
Addr: opt.Addr,
Password: opt.Password,
DB: opt.DB,
})
return &Client{rdb: newRDB(client)} return &Client{rdb: newRDB(client)}
} }