统一导出任务

pkg/queue/handler.go

@@ -6,6 +6,7 @@ import (
 	"go.uber.org/zap"
 	"gorm.io/gorm"
 
+	"github.com/break/junhong_cmp_fiber/internal/exporter"
 	"github.com/break/junhong_cmp_fiber/internal/gateway"
 	"github.com/break/junhong_cmp_fiber/internal/polling"
 	iot_card_svc "github.com/break/junhong_cmp_fiber/internal/service/iot_card"
@@ -64,6 +65,7 @@ func (h *Handler) RegisterHandlers() *asynq.ServeMux {
 
 	h.registerIotCardImportHandler()
 	h.registerDeviceImportHandler()
+	h.registerExportHandlers()
 	h.registerCommissionStatsHandlers()
 	h.registerCommissionCalculationHandler()
 	h.registerPollingHandlers()
@@ -113,6 +115,46 @@ func (h *Handler) registerDeviceImportHandler() {
 	h.logger.Info("注册设备导入任务处理器", zap.String("task_type", constants.TaskTypeDeviceImport))
 }
 
+func (h *Handler) registerExportHandlers() {
+	sceneRegistry := exporter.NewDefaultRegistry(h.db)
+
+	dispatchHandler := task.NewExportDispatchHandler(
+		h.db,
+		h.redis,
+		h.workerResult.Stores.ExportTask,
+		h.workerResult.Stores.ExportShardTask,
+		h.asynqClient,
+		sceneRegistry,
+		h.logger,
+	)
+	shardHandler := task.NewExportShardHandler(
+		h.redis,
+		h.workerResult.Stores.ExportTask,
+		h.workerResult.Stores.ExportShardTask,
+		h.asynqClient,
+		h.storage,
+		sceneRegistry,
+		h.logger,
+	)
+	finalizeHandler := task.NewExportFinalizeHandler(
+		h.redis,
+		h.workerResult.Stores.ExportTask,
+		h.workerResult.Stores.ExportShardTask,
+		h.storage,
+		sceneRegistry,
+		h.logger,
+	)
+
+	h.mux.HandleFunc(constants.TaskTypeExportDispatch, dispatchHandler.HandleExportDispatch)
+	h.logger.Info("注册导出 dispatch 任务处理器", zap.String("task_type", constants.TaskTypeExportDispatch))
+
+	h.mux.HandleFunc(constants.TaskTypeExportShard, shardHandler.HandleExportShard)
+	h.logger.Info("注册导出 shard 任务处理器", zap.String("task_type", constants.TaskTypeExportShard))
+
+	h.mux.HandleFunc(constants.TaskTypeExportFinalize, finalizeHandler.HandleExportFinalize)
+	h.logger.Info("注册导出 finalize 任务处理器", zap.String("task_type", constants.TaskTypeExportFinalize))
+}
+
 func (h *Handler) registerCommissionStatsHandlers() {
 	updateHandler := task.NewCommissionStatsUpdateHandler(
 		h.redis,

pkg/queue/types.go

@@ -22,6 +22,8 @@ type WorkerStores struct {
 	IotCardImportTask         *postgres.IotCardImportTaskStore
 	IotCard                   *postgres.IotCardStore
 	DeviceImportTask          *postgres.DeviceImportTaskStore
+	ExportTask                *postgres.ExportTaskStore
+	ExportShardTask           *postgres.ExportShardTaskStore
 	Device                    *postgres.DeviceStore
 	DeviceSimBinding          *postgres.DeviceSimBindingStore
 	ShopSeriesCommissionStats *postgres.ShopSeriesCommissionStatsStore