138 lines
2.9 KiB
Go
138 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"code.tczkiot.com/wlw/taskq"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
// 定义任务数据结构
|
|
type EmailTask struct {
|
|
UserID int `json:"user_id"`
|
|
TemplateID string `json:"template_id"`
|
|
}
|
|
|
|
type ImageResizeTask struct {
|
|
SourceURL string `json:"source_url"`
|
|
}
|
|
|
|
// 定义任务处理器
|
|
func handleEmailTask(ctx context.Context, t EmailTask) error {
|
|
log.Printf("处理邮件任务: 用户ID=%d, 模板ID=%s", t.UserID, t.TemplateID)
|
|
// 模拟邮件发送逻辑
|
|
return nil
|
|
}
|
|
|
|
func handleImageResizeTask(ctx context.Context, t ImageResizeTask) error {
|
|
log.Printf("处理图片调整任务: 源URL=%s", t.SourceURL)
|
|
// 模拟图片调整逻辑
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
// 创建 Redis 客户端
|
|
rdb := redis.NewClient(&redis.Options{
|
|
Addr: "127.0.0.1:6379",
|
|
DB: 1,
|
|
})
|
|
defer rdb.Close()
|
|
|
|
// 初始化 taskq
|
|
taskq.SetRedis(rdb)
|
|
taskq.Init()
|
|
|
|
// 创建邮件任务
|
|
emailTask := &taskq.Task[EmailTask]{
|
|
Queue: "email",
|
|
Name: "email:deliver",
|
|
MaxRetries: 3,
|
|
Priority: 5,
|
|
TTR: 0,
|
|
Handler: handleEmailTask,
|
|
}
|
|
|
|
// 创建图片调整任务
|
|
imageTask := &taskq.Task[ImageResizeTask]{
|
|
Queue: "image",
|
|
Name: "image:resize",
|
|
MaxRetries: 3,
|
|
Priority: 3,
|
|
TTR: 0,
|
|
Handler: handleImageResizeTask,
|
|
}
|
|
|
|
// 注册任务
|
|
if err := taskq.Register(emailTask); err != nil {
|
|
log.Fatal("注册邮件任务失败:", err)
|
|
}
|
|
if err := taskq.Register(imageTask); err != nil {
|
|
log.Fatal("注册图片任务失败:", err)
|
|
}
|
|
|
|
// 创建监控处理器
|
|
handler, err := taskq.NewInspectHandler(taskq.InspectOptions{
|
|
RootPath: "/monitor",
|
|
ReadOnly: false,
|
|
})
|
|
if err != nil {
|
|
log.Fatal("创建监控处理器失败:", err)
|
|
}
|
|
|
|
// 启动 taskq 服务器
|
|
ctx := context.Background()
|
|
go func() {
|
|
err := taskq.Start(ctx)
|
|
if err != nil {
|
|
log.Fatal("启动 taskq 服务器失败:", err)
|
|
}
|
|
}()
|
|
|
|
// 定时发布任务
|
|
go func() {
|
|
ticker := time.NewTicker(5 * time.Second) // 每5秒发布一次任务
|
|
defer ticker.Stop()
|
|
|
|
taskCounter := 0
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
taskCounter++
|
|
|
|
// 发布邮件任务
|
|
err := emailTask.Publish(ctx, EmailTask{
|
|
UserID: taskCounter,
|
|
TemplateID: "welcome",
|
|
})
|
|
if err != nil {
|
|
log.Printf("发布邮件任务失败: %v", err)
|
|
} else {
|
|
log.Printf("发布邮件任务成功: 用户ID=%d", taskCounter)
|
|
}
|
|
|
|
// 发布图片调整任务
|
|
err = imageTask.Publish(ctx, ImageResizeTask{
|
|
SourceURL: fmt.Sprintf("https://example.com/image%d.jpg", taskCounter),
|
|
})
|
|
if err != nil {
|
|
log.Printf("发布图片任务失败: %v", err)
|
|
} else {
|
|
log.Printf("发布图片任务成功: 任务ID=%d", taskCounter)
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
// 启动 HTTP 服务器提供监控界面
|
|
log.Printf("启动监控服务器在 http://localhost:8080")
|
|
log.Fatal(http.ListenAndServe(":8080", handler))
|
|
}
|