暂存

internal/task/audit_daily_archive.go

@@ -0,0 +1,61 @@
+package task
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/bytedance/sonic"
+	"github.com/hibiken/asynq"
+	"go.uber.org/zap"
+
+	"github.com/break/junhong_cmp_fiber/internal/application/auditarchive"
+	"github.com/break/junhong_cmp_fiber/pkg/constants"
+)
+
+// AuditDailyArchivePayload 是人工补档时可选的任务载荷。
+type AuditDailyArchivePayload struct {
+	ArchiveDate string `json:"archive_date"`
+}
+
+// AuditDailyArchiveHandler 处理统一审计每日冷归档任务。
+type AuditDailyArchiveHandler struct {
+	service *auditarchive.Service
+	logger  *zap.Logger
+}
+
+// NewAuditDailyArchiveHandler 创建统一审计每日冷归档任务处理器。
+func NewAuditDailyArchiveHandler(service *auditarchive.Service, logger *zap.Logger) *AuditDailyArchiveHandler {
+	return &AuditDailyArchiveHandler{service: service, logger: logger}
+}
+
+// Handle 执行前一完整自然日归档，或按任务载荷补档指定自然日。
+func (h *AuditDailyArchiveHandler) Handle(ctx context.Context, task *asynq.Task) error {
+	if h.service == nil {
+		return fmt.Errorf("统一审计归档服务未配置")
+	}
+	var err error
+	if len(task.Payload()) == 0 {
+		err = h.service.ArchivePreviousDay(ctx)
+	} else {
+		var payload AuditDailyArchivePayload
+		if unmarshalErr := sonic.Unmarshal(task.Payload(), &payload); unmarshalErr != nil {
+			return fmt.Errorf("解析统一审计归档任务载荷失败: %w", unmarshalErr)
+		}
+		location, locationErr := time.LoadLocation(constants.AuditArchiveTimezone)
+		if locationErr != nil {
+			return fmt.Errorf("加载统一审计归档时区失败: %w", locationErr)
+		}
+		archiveDate, parseErr := time.ParseInLocation(time.DateOnly, payload.ArchiveDate, location)
+		if parseErr != nil {
+			return fmt.Errorf("解析统一审计归档日期失败: %w", parseErr)
+		}
+		err = h.service.ArchiveDate(ctx, archiveDate)
+	}
+	if err != nil {
+		h.logger.Error("统一审计每日冷归档失败", zap.Error(err))
+		return err
+	}
+	h.logger.Info("统一审计每日冷归档完成")
+	return nil
+}