Go异步任务解决方案 Asynq_如何单独对go asynq servers不下发任务

本文链接：https://blog.csdn.net/2401_84252820/article/details/137705576

2. 安装asynq软件包

go get -u github.com/hibiken/asynq

3. 创建项目asynq_task

目录结构：

2. Redis连接项

Asynq 使用 Redis 作为消息代理。
client.go 和 main.go 都需要连接到 Redis 进行写入和读取。
我们将使用 asynq.RedisClientOpt 指定如何连接到本地 Redis 实例。

asynq.RedisClientOpt{
Addr: “127.0.0.1:6379”,
Password: “”,
DB: 2,
}

4. Task任务

*asynq.Task

type Task struct {
// 一个简单的字符串值，表示要执行的任务的类型.
typename string

// 有效载荷保存执行任务所需的数据，有效负载值必须是可序列化的.
payload []byte

// 保存任务的选项.
opts []Option

// 任务的结果编写器.
w *ResultWriter
}

5. 编写程序

1）test_delivery.go 一个封装任务创建和任务处理的包

package test_delivery

import (
“context”
“encoding/json”
“fmt”
“github.com/hibiken/asynq”
“log”
)

const (
TypeEmailDelivery = “email:deliver”
)

// EmailDeliveryPayload 异步任务需要传递的数据结构
type EmailDeliveryPayload struct {
UserID int
TemplateID string
DataStr string
}

// NewEmailDeliveryTask 异步任务需要传递的数据
func NewEmailDeliveryTask(userID int, tmplID, dataStr string) (*asynq.Task, error) {
payload, err := json.Marshal(EmailDeliveryPayload{UserID: userID, TemplateID: tmplID, DataStr: dataStr})
if err != nil {
fmt.Println(err)
return nil, err
}
return asynq.NewTask(TypeEmailDelivery, payload), nil
}

// HandleEmailDeliveryTask 发送email处理逻辑
func HandleEmailDeliveryTask(ctx context.Context, t *asynq.Task) error {
//接收任务数据
var p EmailDeliveryPayload
if err := json.Unmarshal(t.Payload(), &p); err != nil {
return fmt.Errorf(“json.Unmarshal failed: %v: %w”, err, asynq.SkipRetry)
}
//逻辑处理start…
log.Printf(“Sending Email to User: user_id=%d, template_id=%s data_str:%s”, p.UserID, p.TemplateID, p.DataStr)
return nil
}

client.go

在应用程序代码中，导入上述包并用于Client将任务放入队列中。

package client

import (
“asynq_task/test_delivery”
“fmt”
“github.com/hibiken/asynq”
“log”
“time”
)

func EmailDeliveryTaskAdd(i int) {
client := asynq.NewClient(asynq.RedisClientOpt{
Addr: “192.168.0.120:6379”,
Password: “123456”,
DB: 2,
})
defer client.Close()

// 初使货需要传递的数据
task, err := test_delivery.NewEmailDeliveryTask(42, fmt.Sprintf(“some:template🆔%d”, i), {"name":"lisi"})
if err != nil {
log.Fatalf(“could not create task: %v”, err)
}
// 任务入队
//info, err := client.Enqueue(task)

//info, err := client.Enqueue(task, time.Now())
// 延迟执行
info, err := client.Enqueue(task, asynq.ProcessIn(3*time.Second))
// MaxRetry 重度次数 Timeout超时时间
//info, err = client.Enqueue(task, asynq.MaxRetry(10), asynq.Timeout(3*time.Second))
if err != nil {
log.Fatalf(“could not enqueue task: %v”, err)
}
log.Printf(“enqueued task: id=%s queue=%s”, info.ID, info.Queue)
}

main.go 异步任务服务入口文件

接下来，启动一个工作服务器以在后台处理这些任务。要启动后台工作人员，使用Server并提供您Handler来处理任务。可以选择使用ServeMux来创建处理程序，就像使用net/httpHandler 一样。

package main

import (
“asynq_task/test_delivery”
“github.com/hibiken/asynq”
“log”
)

func main() {
srv := asynq.NewServer(
asynq.RedisClientOpt{
Addr: “192.168.0.120:6379”,
Password: “123456”,
DB: 2,
},
asynq.Config{
// 每个进程并发执行的worker数量
Concurrency: 5,
// Optionally specify multiple queues with different priority.
Queues: map[string]int{
“critical”: 6,
“default”: 3,
“low”: 1,
},
// See the godoc for other configuration options
},
)

mux := asynq.NewServeMux()
mux.HandleFunc(test_delivery.TypeEmailDelivery, test_delivery.HandleEmailDeliveryTask)

if err := srv.Run(mux); err != nil {
log.Fatalf(“could not run server: %v”, err)
}
}

4）test.go 用来分发异步任务

package main

import (
“asynq_task/test_delivery/client”
“time”
)

func main() {
for i := 0; i < 3; i++ {
client.EmailDeliveryTaskAdd(i)
time.Sleep(time.Second * 3)
}
}

6. 运行查看结果

首先，我们要先把异步任务启动起来准备好接收，也就是启动cmd/main.go
启动test.go文件向异步任务服务添加任务队列

结果如下：

消息者 go run main.go

$ go run main.go
asynq: pid=12092 2023/02/02 23:18:04.161872 INFO: Starting processing
asynq: pid=12092 2023/02/02 23:18:04.161872 INFO: Send signal TERM or INT to terminate the process
2023/02/03 07:18:14 Sending Email to User: user_id=42, template_id=some:template🆔0 data_str:{“name”:“lisi”}
2023/02/03 07:18:19 Sending Email to User: user_id=42, template_id=some:template🆔1 data_str:{“name”:“lisi”}
2023/02/03 07:18:19 Sending Email to User: user_id=42, template_id=some:template🆔2 data_str:{“name”:“lisi”}

生产者 go run test.go

$ go run test.go
2023/02/03 07:18:09 enqueued task: id=5d998c6b-3978-4a25-a096-6e564e032359 queue=default
2023/02/03 07:18:12 enqueued task: id=74a5fea4-d4d4-465f-b310-31981e472f6a queue=default
2023/02/03 07:18:15 enqueued task: id=41c46b7b-ea78-4abc-878a-ea65e3859e28 queue=default