统一导出任务

internal/exporter/iot_card_scene.go

@@ -0,0 +1,98 @@
+package exporter
+
+import (
+	"context"
+	"strconv"
+
+	"gorm.io/gorm"
+
+	"github.com/break/junhong_cmp_fiber/internal/model"
+	"github.com/break/junhong_cmp_fiber/pkg/constants"
+)
+
+// IotCardSceneStrategy IoT 卡导出场景策略。
+type IotCardSceneStrategy struct {
+	db *gorm.DB
+}
+
+// NewIotCardSceneStrategy 创建 IoT 卡场景策略。
+func NewIotCardSceneStrategy(db *gorm.DB) *IotCardSceneStrategy {
+	return &IotCardSceneStrategy{db: db}
+}
+
+// Scene 场景编码。
+func (s *IotCardSceneStrategy) Scene() string {
+	return constants.ExportTaskSceneIotCard
+}
+
+// Headers 导出表头。
+func (s *IotCardSceneStrategy) Headers() []string {
+	return []string{"ID", "ICCID", "MSISDN", "虚拟号", "运营商名称", "状态", "店铺ID", "创建时间"}
+}
+
+// NextShardBoundary 获取下一段分片边界。
+func (s *IotCardSceneStrategy) NextShardBoundary(ctx context.Context, task *model.ExportTask, afterID uint64, limit int) (*ShardBoundary, error) {
+	if limit <= 0 {
+		limit = constants.ExportDefaultShardSize
+	}
+
+	var ids []uint64
+	query := s.db.WithContext(ctx).
+		Model(&model.IotCard{}).
+		Where("id > ?", afterID).
+		Order("id ASC").
+		Limit(limit)
+	query = applyShopScopeForTask(query, task)
+
+	if err := query.Pluck("id", &ids).Error; err != nil {
+		return nil, err
+	}
+	if len(ids) == 0 {
+		return nil, nil
+	}
+
+	return &ShardBoundary{
+		StartID:  afterID,
+		EndID:    ids[len(ids)-1],
+		RowCount: len(ids),
+	}, nil
+}
+
+// QueryRows 查询分片内数据行。
+func (s *IotCardSceneStrategy) QueryRows(ctx context.Context, task *model.ExportTask, startID, endID uint64) ([][]string, error) {
+	if endID == 0 {
+		return [][]string{}, nil
+	}
+
+	var cards []model.IotCard
+	query := s.db.WithContext(ctx).
+		Model(&model.IotCard{}).
+		Where("id > ? AND id <= ?", startID, endID).
+		Order("id ASC")
+	query = applyShopScopeForTask(query, task)
+
+	if err := query.Find(&cards).Error; err != nil {
+		return nil, err
+	}
+
+	rows := make([][]string, 0, len(cards))
+	for _, item := range cards {
+		shopID := ""
+		if item.ShopID != nil {
+			shopID = strconv.FormatUint(uint64(*item.ShopID), 10)
+		}
+
+		rows = append(rows, []string{
+			strconv.FormatUint(uint64(item.ID), 10),
+			item.ICCID,
+			item.MSISDN,
+			item.VirtualNo,
+			item.CarrierName,
+			strconv.Itoa(item.Status),
+			shopID,
+			item.CreatedAt.Format("2006-01-02 15:04:05"),
+		})
+	}
+
+	return rows, nil
+}