Asynq 是一个用 Go 编写的任务队列服务器,它提供了简单的 API 来处理异步任务。以下是一个使用 Asynq 的示例,展示了如何将任务添加到队列和处理任务。
首先,确保安装了 Asynq:
go get github.com/hibiken/asynq然后,编写一个简单的任务处理器:
package main
 
import (
    "fmt"
    "time"
    "github.com/hibiken/asynq"
)
 
// 定义一个处理器,实现 asynq.Handler 接口
type HelloWorldHandler struct{}
 
func (h *HelloWorldHandler) ProcessTask(task *asynq.Task) error {
    payload := map[string]interface{}{}
    if err := task.Payload().Unmarshal(&payload); err != nil {
        return err
    }
    name, _ := payload["name"].(string)
    fmt.Printf("Hello, %s!\n", name)
    return nil
}
 
func main() {
    // 设置服务器选项
    srv := asynq.NewServer(
        asynq.RedisClientOpt(asynq.DefaultRedisClient),
        asynq.Config{
            // 可以设置不同的选项,例如并发度
            Concurrency: 10,
            // 可以设置不同的队列及其优先级
            Queues: []string{"default"},
        },
    )
 
    // 注册处理器
    hw := &HelloWorldHandler{}
    srv.Register(hw)
 
    // 启动服务器
    if err := srv.Run(); err != nil {
        fmt.Printf("Could not start asynq server: %v", err)
        return
    }
 
    // 模拟发送任务
    client := asynq.NewClient(asynq.RedisClientOpt(asynq.DefaultRedisClient))
    task := asynq.NewTask("hello_world", map[string]interface{}{"name": "World"})
    _, err := client.Enqueue(task)
    if err != nil {
        fmt.Printf("Could not enqueue task: %v", err)
        return
    }
 
    // 让主进程等待一段时间以模拟任务处理
    time.Sleep(5 * time.Second)
 
    // 关闭服务器
    srv.Close()
}在这个示例中,我们定义了一个简单的 HelloWorldHandler 处理器,它实现了 asynq.Handler 接口。我们创建了一个 Asynq 服务器,注册了这个处理器,并启动了服务器。然后,我们使用 Asynq 客户端发送了一个名为 "hello\_world" 的任务,该任务在处理器中被定义并处理。
请注意,实际应用中你可能需要处理更复杂的任务和错误情况,并且可能需要配置 Redis 服务器的地址和其他相关选项。