暂存

internal/task/audit_monthly_retention.go

@@ -0,0 +1,64 @@
+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"
+)
+
+// AuditMonthlyRetentionPayload 是人工补跑月度清理时可选的任务载荷。
+type AuditMonthlyRetentionPayload struct {
+	ArchiveMonth string `json:"archive_month"`
+}
+
+// AuditMonthlyRetentionHandler 处理归档完整性门禁与上月在线日志物理清理。
+type AuditMonthlyRetentionHandler struct {
+	service *auditarchive.Service
+	logger  *zap.Logger
+}
+
+// NewAuditMonthlyRetentionHandler 创建月度日志留存清理处理器。
+func NewAuditMonthlyRetentionHandler(service *auditarchive.Service, logger *zap.Logger) *AuditMonthlyRetentionHandler {
+	return &AuditMonthlyRetentionHandler{service: service, logger: logger}
+}
+
+// Handle 校验整月归档后按固定顺序分批物理删除 PostgreSQL 在线日志。
+func (h *AuditMonthlyRetentionHandler) Handle(ctx context.Context, task *asynq.Task) error {
+	if h.service == nil {
+		return fmt.Errorf("月度日志留存清理服务未配置")
+	}
+	startedAt := time.Now()
+	var result auditarchive.RetentionResult
+	var err error
+	if len(task.Payload()) == 0 {
+		result, err = h.service.CleanupPreviousMonth(ctx)
+	} else {
+		var payload AuditMonthlyRetentionPayload
+		if unmarshalErr := sonic.Unmarshal(task.Payload(), &payload); unmarshalErr != nil {
+			return fmt.Errorf("解析月度日志留存清理任务载荷失败: %w", unmarshalErr)
+		}
+		month, parseErr := parseArchiveMonth(payload.ArchiveMonth)
+		if parseErr != nil {
+			return parseErr
+		}
+		result, err = h.service.CleanupMonth(ctx, month)
+	}
+	fields := []zap.Field{
+		zap.String("archive_month", result.Month), zap.Int64("audit_event_count", result.EventCount),
+		zap.Int64("event_resource_count", result.ResourceCount), zap.Int64("integration_log_count", result.IntegrationCount),
+		zap.Duration("duration", time.Since(startedAt)), zap.Int("manifest_count", len(result.ManifestKeys)),
+	}
+	if err != nil {
+		fields = append(fields, zap.String("severity", "critical"), zap.Error(err))
+		h.logger.Error("月度日志留存清理失败，PostgreSQL 整月清理已阻断或等待断点续跑", fields...)
+		return err
+	}
+	h.logger.Info("月度日志留存清理完成", fields...)
+	return nil
+}