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

Update client dependency to base.Broker

Browse Source
This commit is contained in:
Ken Hibino 2022-03-08 05:59:58 -08:00
parent eb27b0fe1e
commit ab28234767
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
client.go
View File

@ -24,7 +24,7 @@ import (
//
// Clients are safe for concurrent use by multiple goroutines.
type Client struct {
rdb *rdb.RDB
broker base.Broker
}
// NewClient returns a new Client instance given a redis connection option.
@ -33,7 +33,7 @@ func NewClient(r RedisConnOpt) *Client {
if !ok {
panic(fmt.Sprintf("asynq: unsupported RedisConnOpt type %T", r))
}
return &Client{rdb: rdb.NewRDB(c)}
return &Client{broker: rdb.NewRDB(c)}
}
type OptionType int
@ -309,7 +309,7 @@ var (
// Close closes the connection with redis.
func (c *Client) Close() error {
return c.rdb.Close()
return c.broker.Close()
}
// Enqueue enqueues the given task to a queue.
@ -406,22 +406,22 @@ func (c *Client) EnqueueContext(ctx context.Context, task *Task, opts ...Option)
func (c *Client) enqueue(ctx context.Context, msg *base.TaskMessage, uniqueTTL time.Duration) error {
if uniqueTTL > 0 {
return c.rdb.EnqueueUnique(ctx, msg, uniqueTTL)
return c.broker.EnqueueUnique(ctx, msg, uniqueTTL)
}
return c.rdb.Enqueue(ctx, msg)
return c.broker.Enqueue(ctx, msg)
}
func (c *Client) schedule(ctx context.Context, msg *base.TaskMessage, t time.Time, uniqueTTL time.Duration) error {
if uniqueTTL > 0 {
ttl := t.Add(uniqueTTL).Sub(time.Now())
return c.rdb.ScheduleUnique(ctx, msg, t, ttl)
return c.broker.ScheduleUnique(ctx, msg, t, ttl)
}
return c.rdb.Schedule(ctx, msg, t)
return c.broker.Schedule(ctx, msg, t)
}
func (c *Client) addToGroup(ctx context.Context, msg *base.TaskMessage, groupKey string, uniqueTTL time.Duration) error {
if uniqueTTL > 0 {
return c.rdb.AddToGroupUnique(ctx, msg, groupKey, uniqueTTL)
return c.broker.AddToGroupUnique(ctx, msg, groupKey, uniqueTTL)
}
return c.rdb.AddToGroup(ctx, msg, groupKey)
return c.broker.AddToGroup(ctx, msg, groupKey)
}