fix

internal/infrastructure/task/article_task_handler.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"tyapi-server/internal/domains/article/repositories"
 
 	"github.com/hibiken/asynq"
 	"go.uber.org/zap"
@@ -16,18 +17,21 @@ type ArticlePublisher interface {
 
 // ArticleTaskHandler 文章任务处理器
 type ArticleTaskHandler struct {
-	publisher ArticlePublisher
-	logger    *zap.Logger
+	publisher         ArticlePublisher
+	scheduledTaskRepo repositories.ScheduledTaskRepository
+	logger            *zap.Logger
 }
 
 // NewArticleTaskHandler 创建文章任务处理器
 func NewArticleTaskHandler(
 	publisher ArticlePublisher,
+	scheduledTaskRepo repositories.ScheduledTaskRepository,
 	logger *zap.Logger,
 ) *ArticleTaskHandler {
 	return &ArticleTaskHandler{
-		publisher: publisher,
-		logger:    logger,
+		publisher:         publisher,
+		scheduledTaskRepo: scheduledTaskRepo,
+		logger:           logger,
 	}
 }
 
@@ -45,14 +49,49 @@ func (h *ArticleTaskHandler) HandleArticlePublish(ctx context.Context, t *asynq.
 		return fmt.Errorf("任务载荷中缺少文章ID")
 	}
 	
+	// 获取任务状态记录
+	task, err := h.scheduledTaskRepo.GetByTaskID(ctx, t.ResultWriter().TaskID())
+	if err != nil {
+		h.logger.Error("获取任务状态记录失败", zap.String("task_id", t.ResultWriter().TaskID()), zap.Error(err))
+		// 继续执行，不阻断任务
+	} else {
+		// 检查任务是否已取消
+		if task.IsCancelled() {
+			h.logger.Info("任务已取消，跳过执行", zap.String("task_id", t.ResultWriter().TaskID()))
+			return nil
+		}
+		
+		// 标记任务为正在执行
+		task.MarkAsRunning()
+		if err := h.scheduledTaskRepo.Update(ctx, task); err != nil {
+			h.logger.Warn("更新任务状态失败", zap.String("task_id", t.ResultWriter().TaskID()), zap.Error(err))
+		}
+	}
+	
 	// 执行文章发布
 	if err := h.publisher.PublishArticleByID(ctx, articleID); err != nil {
+		// 更新任务状态为失败
+		if task.ID != "" {
+			task.MarkAsFailed(err.Error())
+			if updateErr := h.scheduledTaskRepo.Update(ctx, task); updateErr != nil {
+				h.logger.Warn("更新任务失败状态失败", zap.String("task_id", t.ResultWriter().TaskID()), zap.Error(updateErr))
+			}
+		}
+		
 		h.logger.Error("定时发布文章失败",
 			zap.String("article_id", articleID),
 			zap.Error(err))
 		return fmt.Errorf("定时发布文章失败: %w", err)
 	}
 	
+	// 更新任务状态为已完成
+	if task.ID != "" {
+		task.MarkAsCompleted()
+		if err := h.scheduledTaskRepo.Update(ctx, task); err != nil {
+			h.logger.Warn("更新任务完成状态失败", zap.String("task_id", t.ResultWriter().TaskID()), zap.Error(err))
+		}
+	}
+	
 	h.logger.Info("定时发布文章成功", zap.String("article_id", articleID))
 	return nil
 }