暂存

cmd/worker/main.go

@@ -16,6 +16,7 @@ import (
 
 	agentrechargeApp "github.com/break/junhong_cmp_fiber/internal/application/agentrecharge"
 	approvalApp "github.com/break/junhong_cmp_fiber/internal/application/approval"
+	auditArchiveApp "github.com/break/junhong_cmp_fiber/internal/application/auditarchive"
 	cardObservationApp "github.com/break/junhong_cmp_fiber/internal/application/cardobservation"
 	notificationApp "github.com/break/junhong_cmp_fiber/internal/application/notification"
 	walletApp "github.com/break/junhong_cmp_fiber/internal/application/wallet"
@@ -146,6 +147,7 @@ func runWorker(cfg *config.Config) {
 	taskHandler.RegisterHandlers()
 	registerWeComApprovalTasks(taskHandler.GetMux(), runtime, cfg, appLogger)
 	registerAgentRechargeRecoveryTask(taskHandler.GetMux(), runtime, appLogger)
+	registerAuditArchiveTask(taskHandler.GetMux(), runtime, appLogger)
 	outboxHandler := outbox.NewHandler(runtime.outboxConsumers)
 	taskHandler.GetMux().HandleFunc(constants.TaskTypeOutboxDeliver, outboxHandler.Handle)
 	startOutboxRelay(ctx, runtime, cfg.Worker.InstanceName, appLogger)
@@ -685,7 +687,7 @@ func startAsynqScheduler(cfg *config.Config, redisAddr string, appLogger *zap.Lo
 		}
 	}()
 
-	appLogger.Info("Asynq Scheduler 已启动（企微审批恢复: 每 2 分钟，套餐临期提醒: 上海时区每日 03:00）")
+	appLogger.Info("Asynq Scheduler 已启动（企微审批恢复: 每 2 分钟，套餐临期提醒: 上海时区每日 03:00，Audit 归档: 每日 04:00，Integration 归档: 每日 04:30/月初 05:00，月度留存清理: 月初 06:00）")
 	return asynqScheduler
 }
 
@@ -760,9 +762,80 @@ func registerAsynqScheduleTasks(asynqScheduler *asynq.Scheduler) error {
 	); err != nil {
 		return fmt.Errorf("注册每日流量落盘定时任务失败: %w", err)
 	}
+	if _, err := asynqScheduler.Register("CRON_TZ=Asia/Shanghai 0 4 * * *", asynq.NewTask(
+		constants.TaskTypeAuditDailyArchive,
+		nil,
+		asynq.MaxRetry(10),
+		asynq.Timeout(2*time.Hour),
+		asynq.Unique(23*time.Hour),
+		asynq.Queue(constants.QueueForTaskType(constants.TaskTypeAuditDailyArchive)),
+	)); err != nil {
+		return fmt.Errorf("注册统一审计每日冷归档定时任务失败: %w", err)
+	}
+	if _, err := asynqScheduler.Register("CRON_TZ=Asia/Shanghai 30 4 * * *", asynq.NewTask(
+		constants.TaskTypeIntegrationDailyArchive,
+		nil,
+		asynq.MaxRetry(10),
+		asynq.Timeout(2*time.Hour),
+		asynq.Unique(23*time.Hour),
+		asynq.Queue(constants.QueueForTaskType(constants.TaskTypeIntegrationDailyArchive)),
+	)); err != nil {
+		return fmt.Errorf("注册 Integration Log 每日冷归档定时任务失败: %w", err)
+	}
+	if _, err := asynqScheduler.Register("CRON_TZ=Asia/Shanghai 0 5 1 * *", asynq.NewTask(
+		constants.TaskTypeIntegrationMonthlyFinalize,
+		nil,
+		asynq.MaxRetry(10),
+		asynq.Timeout(6*time.Hour),
+		asynq.Unique(27*24*time.Hour),
+		asynq.Queue(constants.QueueForTaskType(constants.TaskTypeIntegrationMonthlyFinalize)),
+	)); err != nil {
+		return fmt.Errorf("注册 Integration Log 月度最终版本复核任务失败: %w", err)
+	}
+	if _, err := asynqScheduler.Register("CRON_TZ=Asia/Shanghai 0 6 1 * *", asynq.NewTask(
+		constants.TaskTypeAuditMonthlyRetention,
+		nil,
+		asynq.MaxRetry(10),
+		asynq.Timeout(12*time.Hour),
+		asynq.Unique(27*24*time.Hour),
+		asynq.Queue(constants.QueueForTaskType(constants.TaskTypeAuditMonthlyRetention)),
+	)); err != nil {
+		return fmt.Errorf("注册月度日志留存清理任务失败: %w", err)
+	}
 	return nil
 }
 
+// registerAuditArchiveTask 注册 Audit 与 Integration 冷归档任务处理器。
+func registerAuditArchiveTask(mux *asynq.ServeMux, runtime *workerRuntime, appLogger *zap.Logger) {
+	if runtime.storageSvc == nil {
+		appLogger.Warn("对象存储未配置，审计归档任务将在执行时重试")
+		mux.HandleFunc(constants.TaskTypeAuditDailyArchive, task.NewAuditDailyArchiveHandler(nil, appLogger).Handle)
+		integrationHandler := task.NewIntegrationArchiveHandler(nil, appLogger)
+		mux.HandleFunc(constants.TaskTypeIntegrationDailyArchive, integrationHandler.HandleDaily)
+		mux.HandleFunc(constants.TaskTypeIntegrationMonthlyFinalize, integrationHandler.HandleMonthlyFinalize)
+		mux.HandleFunc(constants.TaskTypeAuditMonthlyRetention, task.NewAuditMonthlyRetentionHandler(nil, appLogger).Handle)
+		return
+	}
+	auditWriter, ok := runtime.workerResult.Services.PaymentAudit.(*auditInfra.Writer)
+	if !ok || auditWriter == nil {
+		appLogger.Fatal("初始化月度日志留存清理失败：统一审计 Writer 未配置")
+	}
+	service, err := auditArchiveApp.NewService(runtime.db, runtime.storageSvc.Provider(), constants.AuditArchiveInstanceID, auditWriter)
+	if err != nil {
+		appLogger.Fatal("初始化统一审计归档服务失败", zap.Error(err))
+	}
+	mux.HandleFunc(constants.TaskTypeAuditDailyArchive, task.NewAuditDailyArchiveHandler(service, appLogger).Handle)
+	integrationHandler := task.NewIntegrationArchiveHandler(service, appLogger)
+	mux.HandleFunc(constants.TaskTypeIntegrationDailyArchive, integrationHandler.HandleDaily)
+	mux.HandleFunc(constants.TaskTypeIntegrationMonthlyFinalize, integrationHandler.HandleMonthlyFinalize)
+	mux.HandleFunc(constants.TaskTypeAuditMonthlyRetention, task.NewAuditMonthlyRetentionHandler(service, appLogger).Handle)
+	appLogger.Info("注册审计归档任务处理器",
+		zap.String("audit_task_type", constants.TaskTypeAuditDailyArchive),
+		zap.String("integration_daily_task_type", constants.TaskTypeIntegrationDailyArchive),
+		zap.String("integration_monthly_task_type", constants.TaskTypeIntegrationMonthlyFinalize),
+		zap.String("retention_task_type", constants.TaskTypeAuditMonthlyRetention))
+}
+
 // createTaskHandler 创建并返回包含全部任务处理器的 Asynq Handler。
 func createTaskHandler(runtime *workerRuntime, appLogger *zap.Logger) *queue.Handler {
 	stopResumeSvc, _ := runtime.workerResult.Services.StopResumeService.(iot_card_svc.StopResumeServiceInterface)