统一导出任务

pkg/constants/constants.go

@@ -46,6 +46,9 @@ const (
 	TaskTypeCommission             = "commission:calculate"     // 分佣计算
 	TaskTypeIotCardImport          = "iot_card:import"          // IoT 卡批量导入
 	TaskTypeDeviceImport           = "device:import"            // 设备批量导入
+	TaskTypeExportDispatch         = "export:dispatch"          // 导出任务分发
+	TaskTypeExportShard            = "export:shard"             // 导出任务分片执行
+	TaskTypeExportFinalize         = "export:finalize"          // 导出任务收尾汇总
 	TaskTypeCommissionStatsUpdate  = "commission:stats:update"  // 佣金统计更新
 	TaskTypeCommissionStatsSync    = "commission:stats:sync"    // 佣金统计同步
 	TaskTypeCommissionStatsArchive = "commission:stats:archive" // 佣金统计归档
@@ -175,6 +178,48 @@ const (
 	DefaultConcurrency = 10
 )
 
+// 导出任务相关常量
+const (
+	ExportTaskSceneDevice  = "device"   // 导出场景：设备
+	ExportTaskSceneIotCard = "iot_card" // 导出场景：IoT 卡
+)
+
+// 导出文件格式常量
+const (
+	ExportTaskFormatXLSX = "xlsx" // 导出格式：Excel
+	ExportTaskFormatCSV  = "csv"  // 导出格式：CSV
+)
+
+// 导出主任务状态常量
+const (
+	ExportTaskStatusPending    = 1 // 待处理
+	ExportTaskStatusProcessing = 2 // 处理中
+	ExportTaskStatusCompleted  = 3 // 已完成
+	ExportTaskStatusFailed     = 4 // 已失败
+	ExportTaskStatusCancelled  = 5 // 已取消
+)
+
+// 导出分片任务状态常量
+const (
+	ExportShardStatusPending    = 1 // 待处理
+	ExportShardStatusProcessing = 2 // 处理中
+	ExportShardStatusSuccess    = 3 // 已成功
+	ExportShardStatusFailed     = 4 // 已失败
+	ExportShardStatusCancelled  = 5 // 已取消
+)
+
+// 导出任务执行配置常量
+const (
+	ExportDefaultShardSize    = 2000             // 默认分片大小
+	ExportDownloadURLExpire   = 24 * time.Hour   // 下载链接固定有效期
+	ExportDispatchTaskTimeout = 10 * time.Minute // dispatch 任务超时
+	ExportShardTaskTimeout    = 15 * time.Minute // shard 任务超时
+	ExportFinalizeTaskTimeout = 10 * time.Minute // finalize 任务超时
+	ExportFinalizeRetryMax    = 30               // finalize 最大重试次数
+	ExportShardRetryMax       = 3                // shard 最大重试次数
+	ExportDispatchRetryMax    = 5                // dispatch 最大重试次数
+)
+
 // 店铺配置常量
 const (
 	MaxShopLevel = 7 // 店铺最大层级
@@ -260,6 +305,42 @@ func GetShelfStatusName(status int) string {
 	}
 }
 
+// GetExportTaskStatusName 获取导出主任务状态名称
+func GetExportTaskStatusName(status int) string {
+	switch status {
+	case ExportTaskStatusPending:
+		return "待处理"
+	case ExportTaskStatusProcessing:
+		return "处理中"
+	case ExportTaskStatusCompleted:
+		return "已完成"
+	case ExportTaskStatusFailed:
+		return "已失败"
+	case ExportTaskStatusCancelled:
+		return "已取消"
+	default:
+		return "未知"
+	}
+}
+
+// GetExportShardStatusName 获取导出分片任务状态名称
+func GetExportShardStatusName(status int) string {
+	switch status {
+	case ExportShardStatusPending:
+		return "待处理"
+	case ExportShardStatusProcessing:
+		return "处理中"
+	case ExportShardStatusSuccess:
+		return "已成功"
+	case ExportShardStatusFailed:
+		return "已失败"
+	case ExportShardStatusCancelled:
+		return "已取消"
+	default:
+		return "未知"
+	}
+}
+
 // GetExchangeStatusName 获取换货单状态名称
 // 用于 exchange_status 字段
 func GetExchangeStatusName(status int) string {

pkg/constants/redis.go

@@ -46,6 +46,27 @@ func RedisTaskStatusKey(taskID string) string {
 	return fmt.Sprintf("task:status:%s", taskID)
 }
 
+// RedisExportDispatchLockKey 生成导出 dispatch 执行锁的 Redis 键
+// 用途：防止同一导出任务并发执行 dispatch
+// 过期时间：5 分钟
+func RedisExportDispatchLockKey(taskID uint) string {
+	return fmt.Sprintf("export:dispatch:lock:%d", taskID)
+}
+
+// RedisExportShardLockKey 生成导出分片执行锁的 Redis 键
+// 用途：防止同一分片被多个 Worker 并发处理
+// 过期时间：10 分钟
+func RedisExportShardLockKey(taskID, shardID uint) string {
+	return fmt.Sprintf("export:shard:lock:%d:%d", taskID, shardID)
+}
+
+// RedisExportFinalizeLockKey 生成导出 finalize 执行锁的 Redis 键
+// 用途：防止同一导出任务并发执行 finalize
+// 过期时间：2 分钟
+func RedisExportFinalizeLockKey(taskID uint) string {
+	return fmt.Sprintf("export:finalize:lock:%d", taskID)
+}
+
 // RedisShopSubordinatesKey 生成店铺下级 ID 列表的 Redis 键
 // 用途：缓存递归查询的下级店铺 ID 列表
 // 过期时间：30 分钟