market-data-service/internal/service/admin.go

package service

import (
	"context"
	"fmt"
	"time"

	"github.com/google/uuid"

	"market-data-service/api"
	"market-data-service/internal/repository"
)

// AdminServiceImpl 管理服务实现
type AdminServiceImpl struct {
	db *repository.DB
}

// NewAdminService 创建管理服务
func NewAdminService(db *repository.DB) AdminService {
	return &AdminServiceImpl{
		db: db,
	}
}

// GetDataSourceStatus 获取数据源状态
func (s *AdminServiceImpl) GetDataSourceStatus(ctx context.Context) (*api.DataSourceStatusData, error) {
	// 查询数据库中的数据源配置
	query := `
		SELECT asset_class, active_source, standby_sources, updated_at
		FROM public.data_source_config
	`
	rows, err := s.db.QueryContext(ctx, query)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	data := &api.DataSourceStatusData{}
	for rows.Next() {
		var assetClass, activeSource string
		var standbySources []string
		var updatedAt time.Time
		if err := rows.Scan(&assetClass, &activeSource, &standbySources, &updatedAt); err != nil {
			return nil, err
		}

		info := api.DataSourceInfo{
			ActiveSource:   activeSource,
			StandbySources: standbySources,
			Status:         api.DataSourceHealthy,
		}

		switch assetClass {
		case "stock":
			data.Stock = info
		case "futures":
			data.Futures = info
		}
	}

	return data, rows.Err()
}

// SwitchDataSource 切换数据源
func (s *AdminServiceImpl) SwitchDataSource(ctx context.Context, req *api.SourceSwitchRequest) error {
	// 更新数据源配置
	query := `
		UPDATE public.data_source_config
		SET active_source = $1, updated_at = NOW()
		WHERE asset_class = $2
	`

	assetClasses := []string{}
	if req.AssetClass == api.AssetAll {
		assetClasses = []string{"stock", "futures"}
	} else {
		assetClasses = []string{string(req.AssetClass)}
	}

	for _, ac := range assetClasses {
		_, err := s.db.ExecContext(ctx, query, req.Source, ac)
		if err != nil {
			return fmt.Errorf("failed to switch %s data source: %w", ac, err)
		}
	}

	// 如果需要同步补录，启动后台任务
	if req.SyncBackfill {
		// TODO: 启动异步补录任务
		go func() {
			// 异步执行补录
		}()
	}

	return nil
}

// BackfillData 历史数据补录
func (s *AdminServiceImpl) BackfillData(ctx context.Context, req *api.BackfillRequest) (string, error) {
	taskID := uuid.New().String()

	// TODO: 将补录任务存入数据库，启动后台Worker执行
	// 这里简化处理，直接返回任务ID

	return taskID, nil
}

// HealthCheck 健康检查
func (s *AdminServiceImpl) HealthCheck(ctx context.Context) (*api.HealthResponse, error) {
	// 检查数据库连接
	if err := s.db.PingContext(ctx); err != nil {
		return nil, fmt.Errorf("database connection failed: %w", err)
	}

	return &api.HealthResponse{
		Status:    "healthy",
		Timestamp: time.Now(),
	}, nil
}