导出系统

internal/exporter/registry.go

@@ -1,71 +1,54 @@
 package exporter
 
 import (
-	"context"
 	"sort"
 
 	"gorm.io/gorm"
 
-	"github.com/break/junhong_cmp_fiber/internal/model"
 	"github.com/break/junhong_cmp_fiber/pkg/constants"
 )
 
-// ShardBoundary 分片边界信息。
-type ShardBoundary struct {
-	StartID  uint64
-	EndID    uint64
-	RowCount int
-}
-
-// SceneStrategy 导出场景策略接口。
-type SceneStrategy interface {
-	Scene() string
-	Headers() []string
-	NextShardBoundary(ctx context.Context, task *model.ExportTask, afterID uint64, limit int) (*ShardBoundary, error)
-	QueryRows(ctx context.Context, task *model.ExportTask, startID, endID uint64) ([][]string, error)
-}
-
-// Registry 场景策略注册中心。
+// Registry 导出数据源注册中心。
 type Registry struct {
-	strategies map[string]SceneStrategy
+	sources map[string]DataSource
 }
 
-// NewRegistry 创建场景策略注册中心。
-func NewRegistry(strategies ...SceneStrategy) *Registry {
-	m := make(map[string]SceneStrategy, len(strategies))
-	for _, strategy := range strategies {
-		if strategy == nil {
+// NewRegistry 创建导出数据源注册中心。
+func NewRegistry(sources ...DataSource) *Registry {
+	m := make(map[string]DataSource, len(sources))
+	for _, source := range sources {
+		if source == nil {
 			continue
 		}
-		m[strategy.Scene()] = strategy
+		m[source.Scene()] = source
 	}
-	return &Registry{strategies: m}
+	return &Registry{sources: m}
 }
 
-// NewDefaultRegistry 创建默认场景策略注册中心。
+// NewDefaultRegistry 创建默认导出数据源注册中心。
 func NewDefaultRegistry(db *gorm.DB) *Registry {
 	return NewRegistry(
-		NewDeviceSceneStrategy(db),
-		NewIotCardSceneStrategy(db),
+		NewDeviceDataSource(db),
+		NewIotCardDataSource(db),
 	)
 }
 
-// Get 获取指定场景策略。
-func (r *Registry) Get(scene string) (SceneStrategy, bool) {
-	strategy, ok := r.strategies[scene]
-	return strategy, ok
+// Get 获取指定场景数据源。
+func (r *Registry) Get(scene string) (DataSource, bool) {
+	source, ok := r.sources[scene]
+	return source, ok
 }
 
 // IsSupported 判断场景是否支持。
 func (r *Registry) IsSupported(scene string) bool {
-	_, ok := r.strategies[scene]
+	_, ok := r.sources[scene]
 	return ok
 }
 
 // Scenes 返回当前支持的场景列表。
 func (r *Registry) Scenes() []string {
-	result := make([]string, 0, len(r.strategies))
-	for scene := range r.strategies {
+	result := make([]string, 0, len(r.sources))
+	for scene := range r.sources {
 		result = append(result, scene)
 	}
 	sort.Strings(result)