2
0
mirror of https://github.com/hibiken/asynq.git synced 2024-12-24 23:02:18 +08:00

Remove redis.Client type from asynq package API

This commit is contained in:
Ken Hibino 2020-01-14 21:19:06 -08:00
parent cb2ebf18ac
commit 43d7591250
5 changed files with 178 additions and 52 deletions

View File

@ -1,8 +1,10 @@
# Asynq # Asynq
[![Build Status](https://travis-ci.com/hibiken/asynq.svg?token=paqzfpSkF4p23s5Ux39b&branch=master)](https://travis-ci.com/hibiken/asynq) [![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](https://opensource.org/licenses/MIT) [![Build Status](https://travis-ci.com/hibiken/asynq.svg?token=paqzfpSkF4p23s5Ux39b&branch=master)](https://travis-ci.com/hibiken/asynq) [![License: MIT](https://img.shields.io/badge/license-MIT-green.svg)](https://opensource.org/licenses/MIT)
Simple, efficent asynchronous task processing library in Go. Simple and efficent asynchronous task processing library in Go.
**Important Note**: Current major version is zero (v0.x.x) to accomodate rapid development and fast iteration while getting early feedback from users. The public API could change without a major version update before the release of verson 1.0.0.
## Table of Contents ## Table of Contents
@ -10,6 +12,7 @@ Simple, efficent asynchronous task processing library in Go.
- [Requirements](#requirements) - [Requirements](#requirements)
- [Installation](#installation) - [Installation](#installation)
- [Getting Started](#getting-started) - [Getting Started](#getting-started)
- [Monitoring CLI](#monitoring-cli)
- [Acknowledgements](#acknowledgements) - [Acknowledgements](#acknowledgements)
- [License](#license) - [License](#license)
@ -27,44 +30,49 @@ Asynq provides:
- Ability to configure max retry count per task - Ability to configure max retry count per task
- Ability to configure max number of worker goroutines to process tasks - Ability to configure max number of worker goroutines to process tasks
- Unix signal handling to safely shutdown background processing - Unix signal handling to safely shutdown background processing
- Enhanced reliability TODO(hibiken): link to wiki page describing this.
- CLI to query and mutate queues state for mointoring and administrative purposes - CLI to query and mutate queues state for mointoring and administrative purposes
## Requirements ## Requirements
| Dependency | Version | | Dependency | Version |
| -------------------------------------------------------------- | ------- | | -------------------------- | ------- |
| [Redis](https://redis.io/) | v2.6+ | | [Redis](https://redis.io/) | v2.8+ |
| [Go](https://golang.org/) | v1.12+ | | [Go](https://golang.org/) | v1.12+ |
| [github.com/go-redis/redis](https://github.com/go-redis/redis) | v.7.0+ |
## Installation ## Installation
``` ```
go get github.com/hibiken/asynq go get -u github.com/hibiken/asynq
``` ```
## Getting Started ## Getting Started
1. Import `asynq` and `redis` in your file. 1. Import `asynq` in your file.
```go ```go
import ( import "github.com/hibiken/asynq"
"github.com/go-redis/redis/v7"
"github.com/hibiken/asynq"
)
``` ```
2. Create a `Client` instance to create tasks. 2. Use one of `RedisConnOpt` types to specify how to connect to Redis.
```go
var redis = &asynq.RedisClientOpt{
Addr: "localhost:6379",
// Omit if no password is required
Password: "mypassword",
// Use a dedicated db number for asynq.
// By default, Redis offers 16 databases (0..15)
DB: 0,
}
```
3. Create a `Client` instance to create and schedule tasks.
```go ```go
func main() { func main() {
r := redis.NewClient(&redis.Options{ client := asynq.NewClient(redis)
Addr: "localhost:6379",
}
client := asynq.NewClient(r)
// create a task with typename and payload. // Create a task with typename and payload.
t1 := asynq.NewTask( t1 := asynq.NewTask(
"send_welcome_email", "send_welcome_email",
map[string]interface{}{"user_id": 42}) map[string]interface{}{"user_id": 42})
@ -73,35 +81,34 @@ func main() {
"send_reminder_email", "send_reminder_email",
map[string]interface{}{"user_id": 42}) map[string]interface{}{"user_id": 42})
// process the task immediately. // Process the task immediately.
err := client.Schedule(t1, time.Now()) err := client.Schedule(t1, time.Now())
// process the task 24 hours later. // Process the task 24 hours later.
err = client.Schedule(t2, time.Now().Add(24 * time.Hour)) err = client.Schedule(t2, time.Now().Add(24 * time.Hour))
// specify the max number of retry (default: 25) // Specify the max number of retry (default: 25)
err = client.Schedule(t1, time.Now(), asynq.MaxRetry(1)) err = client.Schedule(t1, time.Now(), asynq.MaxRetry(1))
} }
``` ```
3. Create a `Background` instance to process tasks. 4. Create a `Background` instance to process tasks.
```go ```go
func main() { func main() {
r := redis.NewClient(&redis.Options{ bg := asynq.NewBackground(redis, &asynq.Config{
Addr: "localhost:6379", Concurrency: 10,
}
bg := asynq.NewBackground(r, &asynq.Config{
Concurrency: 20,
}) })
// Blocks until signal TERM or INT is received. // Blocks until signal TERM or INT is received.
// For graceful shutdown, send signal TSTP to stop processing more tasks // For graceful shutdown, send signal TSTP to stop processing more tasks
// before sending TERM or INT signal. // before sending TERM or INT signal to terminate the process.
bg.Run(handler) bg.Run(handler)
} }
``` ```
Note that `Client` and `Background` are intended to be used in separate executable binaries.
The argument to `(*asynq.Background).Run` is an interface `asynq.Handler` which has one method `ProcessTask`. The argument to `(*asynq.Background).Run` is an interface `asynq.Handler` which has one method `ProcessTask`.
```go ```go
@ -136,11 +143,8 @@ func handler(t *asynq.Task) error {
} }
func main() { func main() {
r := redis.NewClient(&redis.Options{ bg := asynq.NewBackground(redis, &asynq.Config{
Addr: "localhost:6379", Concurrency: 10,
}
bg := asynq.NewBackground(r, &asynq.Config{
Concurrency: 20,
}) })
// Use asynq.HandlerFunc adapter for a handler function // Use asynq.HandlerFunc adapter for a handler function
@ -148,6 +152,10 @@ func main() {
} }
``` ```
## Monitoring CLI
TODO(hibiken): Describe basic usage of `asynqmon` CLI
## Acknowledgements ## Acknowledgements
- [Sidekiq](https://github.com/mperham/sidekiq) : Many of the design ideas are taken from sidekiq and its Web UI - [Sidekiq](https://github.com/mperham/sidekiq) : Many of the design ideas are taken from sidekiq and its Web UI

116
asynq.go
View File

@ -4,10 +4,12 @@
package asynq package asynq
/* import (
TODOs: "crypto/tls"
- [P0] Go docs "fmt"
*/
"github.com/go-redis/redis/v7"
)
// Task represents a task to be performed. // Task represents a task to be performed.
type Task struct { type Task struct {
@ -28,3 +30,109 @@ func NewTask(typename string, payload map[string]interface{}) *Task {
Payload: Payload{payload}, Payload: Payload{payload},
} }
} }
// RedisConnOpt is a discriminated union of redis-client-option types.
//
// RedisConnOpt represents a sum of following types:
// RedisClientOpt | *RedisClientOpt | RedisFailoverClientOpt | *RedisFailoverClientOpt
type RedisConnOpt interface{}
// RedisClientOpt is used to specify redis client options to connect
// to a redis server running as a stand alone instance.
type RedisClientOpt struct {
// Network type to use, either tcp or unix.
// Default is tcp.
Network string
// Redis server address in the format 'host:port'.
Addr string
// Redis server password.
Password string
// Redis DB to select after connecting to the server.
// See: https://redis.io/commands/select.
DB int
// Maximum number of socket connections.
// Default is 10 connections per every CPU as reported by runtime.NumCPU.
PoolSize int
// TLS Config used to connect to the server.
// TLS will be negotiated only if this field is set.
TLSConfig *tls.Config
}
// RedisFailoverClientOpt is used to specify redis failover client.
type RedisFailoverClientOpt struct {
// Redis master name that monitored by sentinels.
MasterName string
// Addresses of sentinels in the form "host:port".
// Use at least three sentinels to avoid problems described in
// https://redis.io/topics/sentinel.
SentinelAddrs []string
// Redis sentinel password.
SentinelPassword string
// Redis server password.
Password string
// Redis DB to select after connecting to the server.
// See: https://redis.io/commands/select.
DB int
// Maximum number of socket connections.
// Default is 10 connections per every CPU as reported by runtime.NumCPU.
PoolSize int
// TLS Config used to connect to the server.
// TLS will be negotiated only if this field is set.
TLSConfig *tls.Config
}
func createRedisClient(r RedisConnOpt) *redis.Client {
switch r := r.(type) {
case *RedisClientOpt:
return redis.NewClient(&redis.Options{
Network: r.Network,
Addr: r.Addr,
Password: r.Password,
DB: r.DB,
PoolSize: r.PoolSize,
TLSConfig: r.TLSConfig,
})
case RedisClientOpt:
return redis.NewClient(&redis.Options{
Network: r.Network,
Addr: r.Addr,
Password: r.Password,
DB: r.DB,
PoolSize: r.PoolSize,
TLSConfig: r.TLSConfig,
})
case *RedisFailoverClientOpt:
return redis.NewFailoverClient(&redis.FailoverOptions{
MasterName: r.MasterName,
SentinelAddrs: r.SentinelAddrs,
SentinelPassword: r.SentinelPassword,
Password: r.Password,
DB: r.DB,
PoolSize: r.PoolSize,
TLSConfig: r.TLSConfig,
})
case RedisFailoverClientOpt:
return redis.NewFailoverClient(&redis.FailoverOptions{
MasterName: r.MasterName,
SentinelAddrs: r.SentinelAddrs,
SentinelPassword: r.SentinelPassword,
Password: r.Password,
DB: r.DB,
PoolSize: r.PoolSize,
TLSConfig: r.TLSConfig,
})
default:
panic(fmt.Sprintf("unexpected type %T for RedisConnOpt", r))
}
}

View File

@ -15,7 +15,6 @@ import (
"syscall" "syscall"
"time" "time"
"github.com/go-redis/redis/v7"
"github.com/hibiken/asynq/internal/base" "github.com/hibiken/asynq/internal/base"
"github.com/hibiken/asynq/internal/rdb" "github.com/hibiken/asynq/internal/rdb"
) )
@ -95,7 +94,7 @@ var defaultQueueConfig = map[string]uint{
// NewBackground returns a new Background instance given a redis client // NewBackground returns a new Background instance given a redis client
// and background processing configuration. // and background processing configuration.
func NewBackground(r *redis.Client, cfg *Config) *Background { func NewBackground(r RedisConnOpt, cfg *Config) *Background {
n := cfg.Concurrency n := cfg.Concurrency
if n < 1 { if n < 1 {
n = 1 n = 1
@ -110,7 +109,7 @@ func NewBackground(r *redis.Client, cfg *Config) *Background {
} }
qcfg := normalizeQueueCfg(queues) qcfg := normalizeQueueCfg(queues)
rdb := rdb.NewRDB(r) rdb := rdb.NewRDB(createRedisClient(r))
scheduler := newScheduler(rdb, 5*time.Second, qcfg) scheduler := newScheduler(rdb, 5*time.Second, qcfg)
processor := newProcessor(rdb, n, qcfg, cfg.StrictPriority, delayFunc) processor := newProcessor(rdb, n, qcfg, cfg.StrictPriority, delayFunc)
return &Background{ return &Background{

View File

@ -8,7 +8,6 @@ import (
"strings" "strings"
"time" "time"
"github.com/go-redis/redis/v7"
"github.com/hibiken/asynq/internal/base" "github.com/hibiken/asynq/internal/base"
"github.com/hibiken/asynq/internal/rdb" "github.com/hibiken/asynq/internal/rdb"
"github.com/rs/xid" "github.com/rs/xid"
@ -25,8 +24,8 @@ type Client struct {
} }
// NewClient and returns a new Client given a redis configuration. // NewClient and returns a new Client given a redis configuration.
func NewClient(r *redis.Client) *Client { func NewClient(r RedisConnOpt) *Client {
rdb := rdb.NewRDB(r) rdb := rdb.NewRDB(createRedisClient(r))
return &Client{rdb} return &Client{rdb}
} }

26
doc.go
View File

@ -5,15 +5,27 @@
/* /*
Package asynq provides a framework for background task processing. Package asynq provides a framework for background task processing.
Asynq uses Redis as a message broker. To connect to redis server,
specify the options using one of RedisConnOpt types.
redis = &asynq.RedisClientOpt{
Addr: "localhost:6379",
Password: "secretpassword",
DB: 3,
}
The Client is used to register a task to be processed at the specified time. The Client is used to register a task to be processed at the specified time.
client := asynq.NewClient(redis) Task is created with two parameters: its type and payload.
t := asynq.NewTask( client := asynq.NewClient(redis)
"send_email",
map[string]interface{}{"user_id": 42})
err := client.Schedule(t, time.Now().Add(time.Minute)) t := asynq.NewTask(
"send_email",
map[string]interface{}{"user_id": 42})
// Schedule the task t to be processed a minute from now.
err := client.Schedule(t, time.Now().Add(time.Minute))
The Background is used to run the background task processing with a given The Background is used to run the background task processing with a given
handler. handler.
@ -26,7 +38,7 @@ handler.
Handler is an interface with one method ProcessTask which Handler is an interface with one method ProcessTask which
takes a task and returns an error. Handler should return nil if takes a task and returns an error. Handler should return nil if
the processing is successful, otherwise return a non-nil error. the processing is successful, otherwise return a non-nil error.
If handler returns a non-nil error, the task will be retried in the future. If handler panics or returns a non-nil error, the task will be retried in the future.
Example of a type that implements the Handler interface. Example of a type that implements the Handler interface.
type TaskHandler struct { type TaskHandler struct {
@ -42,7 +54,7 @@ Example of a type that implements the Handler interface.
// generate thumbnail image // generate thumbnail image
//... //...
default: default:
return fmt.Errorf("unepected task type %q", task.Type) return fmt.Errorf("unexpected task type %q", task.Type)
} }
return nil return nil
} }