add article

internal/infrastructure/task/asynq_client.go

@@ -0,0 +1,75 @@
+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
+}