76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/hibiken/asynq"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// AsynqClient Asynq 客户端
|
|
type AsynqClient struct {
|
|
client *asynq.Client
|
|
logger *zap.Logger
|
|
}
|
|
|
|
// NewAsynqClient 创建 Asynq 客户端
|
|
func NewAsynqClient(redisAddr string, logger *zap.Logger) *AsynqClient {
|
|
client := asynq.NewClient(asynq.RedisClientOpt{Addr: redisAddr})
|
|
return &AsynqClient{
|
|
client: client,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// Close 关闭客户端
|
|
func (c *AsynqClient) Close() error {
|
|
return c.client.Close()
|
|
}
|
|
|
|
// ScheduleArticlePublish 调度文章定时发布任务
|
|
func (c *AsynqClient) ScheduleArticlePublish(ctx context.Context, articleID string, publishTime time.Time) error {
|
|
payload := map[string]interface{}{
|
|
"article_id": articleID,
|
|
}
|
|
|
|
payloadBytes, err := json.Marshal(payload)
|
|
if err != nil {
|
|
c.logger.Error("序列化任务载荷失败", zap.Error(err))
|
|
return fmt.Errorf("创建任务失败: %w", err)
|
|
}
|
|
|
|
task := asynq.NewTask(TaskTypeArticlePublish, payloadBytes)
|
|
|
|
// 计算延迟时间
|
|
delay := publishTime.Sub(time.Now())
|
|
if delay <= 0 {
|
|
return fmt.Errorf("定时发布时间不能早于当前时间")
|
|
}
|
|
|
|
// 设置任务选项
|
|
opts := []asynq.Option{
|
|
asynq.ProcessIn(delay),
|
|
asynq.MaxRetry(3),
|
|
asynq.Timeout(5 * time.Minute),
|
|
}
|
|
|
|
info, err := c.client.Enqueue(task, opts...)
|
|
if err != nil {
|
|
c.logger.Error("调度定时发布任务失败",
|
|
zap.String("article_id", articleID),
|
|
zap.Time("publish_time", publishTime),
|
|
zap.Error(err))
|
|
return fmt.Errorf("调度任务失败: %w", err)
|
|
}
|
|
|
|
c.logger.Info("定时发布任务调度成功",
|
|
zap.String("article_id", articleID),
|
|
zap.Time("publish_time", publishTime),
|
|
zap.String("task_id", info.ID))
|
|
|
|
return nil
|
|
}
|