junhong_cmp_fiber/internal/service/polling/config_service.go

package polling

import (
	"context"
	"time"

	"github.com/redis/go-redis/v9"
	"go.uber.org/zap"
	"gorm.io/gorm"

	auditinfra "github.com/break/junhong_cmp_fiber/internal/infrastructure/audit"
	"github.com/break/junhong_cmp_fiber/internal/model"
	"github.com/break/junhong_cmp_fiber/internal/model/dto"
	"github.com/break/junhong_cmp_fiber/internal/store"
	"github.com/break/junhong_cmp_fiber/internal/store/postgres"
	"github.com/break/junhong_cmp_fiber/pkg/constants"
	"github.com/break/junhong_cmp_fiber/pkg/errors"
	"github.com/break/junhong_cmp_fiber/pkg/middleware"
)

// ConfigService 轮询配置服务
type ConfigService struct {
	configStore *postgres.PollingConfigStore
	db          *gorm.DB
	auditWriter *auditinfra.Writer
	redis       *redis.Client
	logger      *zap.Logger
}

// SetAudit 注入轮询配置事务与统一审计 Writer。
func (s *ConfigService) SetAudit(db *gorm.DB, writer *auditinfra.Writer) {
	s.db = db
	s.auditWriter = writer
}

// NewConfigService 创建轮询配置服务实例
func NewConfigService(configStore *postgres.PollingConfigStore, redisClient *redis.Client, logger *zap.Logger) *ConfigService {
	return &ConfigService{configStore: configStore, redis: redisClient, logger: logger}
}

func (s *ConfigService) notifyConfigChanged(ctx context.Context, event string) {
	if s.redis == nil {
		return
	}
	if err := s.redis.Publish(ctx, constants.RedisPollingConfigChangedChannel(), event).Err(); err != nil {
		s.logger.Warn("通知轮询配置变更失败", zap.String("event", event), zap.Error(err))
	}
}

// Create 创建轮询配置
func (s *ConfigService) Create(ctx context.Context, req *dto.CreatePollingConfigRequest) (*dto.PollingConfigResponse, error) {
	currentUserID := middleware.GetUserIDFromContext(ctx)
	if currentUserID == 0 {
		return nil, errors.New(errors.CodeUnauthorized, "未授权访问")
	}

	// 验证配置名称唯一性
	existing, _ := s.configStore.GetByName(ctx, req.ConfigName)
	if existing != nil {
		appErr := errors.New(errors.CodeInvalidParam, "配置名称已存在")
		recordPollingFailure(ctx, s.db, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigCreated, Summary: "拒绝创建重复轮询配置",
			ResourceType: constants.AuditResourcePollingConfig, ResourceKey: req.ConfigName,
			DisplayName: req.ConfigName, OperatorID: currentUserID, Result: constants.AuditResultDenied,
			IdentitySnapshot: map[string]any{"config_name": req.ConfigName},
		}, appErr)
		return nil, appErr
	}

	// 验证检查间隔（至少一个不为空）
	if req.RealnameCheckInterval == nil && req.CarddataCheckInterval == nil &&
		req.PackageCheckInterval == nil && req.ProtectCheckInterval == nil &&
		req.CardStatusCheckInterval == nil {
		return nil, errors.New(errors.CodeInvalidParam, "至少需要配置一种检查间隔")
	}

	config := &model.PollingConfig{
		ConfigName:              req.ConfigName,
		CardCondition:           req.CardCondition,
		CardCategory:            req.CardCategory,
		CarrierID:               req.CarrierID,
		Priority:                req.Priority,
		RealnameCheckInterval:   req.RealnameCheckInterval,
		CarddataCheckInterval:   req.CarddataCheckInterval,
		PackageCheckInterval:    req.PackageCheckInterval,
		ProtectCheckInterval:    req.ProtectCheckInterval,
		CardStatusCheckInterval: req.CardStatusCheckInterval,
		Status:                  1,
		Description:             req.Description,
		CreatedBy:               &currentUserID,
		UpdatedBy:               &currentUserID,
	}

	err := runPollingTransaction(ctx, s.db, s.auditWriter, func(tx *gorm.DB) error {
		if err := s.configStore.WithTx(tx).Create(ctx, config); err != nil {
			return err
		}
		return writePollingAudit(ctx, tx, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigCreated, Summary: "创建轮询配置",
			ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
			ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
			OperatorID: currentUserID, IdentitySnapshot: pollingConfigIdentity(config), AfterData: pollingConfigState(config),
		})
	})
	if err != nil {
		appErr := errors.Wrap(errors.CodeInternalError, err, "创建轮询配置失败")
		recordPollingFailure(ctx, s.db, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigCreated, Summary: "创建轮询配置失败",
			ResourceType: constants.AuditResourcePollingConfig,
			ResourceKey:  config.ConfigName, DisplayName: config.ConfigName, OperatorID: currentUserID,
			IdentitySnapshot: pollingConfigIdentity(config), AfterData: pollingConfigState(config),
		}, appErr)
		return nil, appErr
	}

	s.notifyConfigChanged(ctx, "created")
	return s.toResponse(config), nil
}

// Get 获取轮询配置详情
func (s *ConfigService) Get(ctx context.Context, id uint) (*dto.PollingConfigResponse, error) {
	config, err := s.configStore.GetByID(ctx, id)
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, errors.New(errors.CodePollingConfigNotFound, "轮询配置不存在")
		}
		return nil, errors.Wrap(errors.CodeInternalError, err, "获取轮询配置失败")
	}
	return s.toResponse(config), nil
}

// Update 更新轮询配置
func (s *ConfigService) Update(ctx context.Context, id uint, req *dto.UpdatePollingConfigRequest) (*dto.PollingConfigResponse, error) {
	currentUserID := middleware.GetUserIDFromContext(ctx)
	if currentUserID == 0 {
		return nil, errors.New(errors.CodeUnauthorized, "未授权访问")
	}

	config, err := s.configStore.GetByID(ctx, id)
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, errors.New(errors.CodePollingConfigNotFound, "轮询配置不存在")
		}
		return nil, errors.Wrap(errors.CodeInternalError, err, "获取轮询配置失败")
	}
	before := *config

	// 更新字段
	if req.ConfigName != nil {
		// 检查名称唯一性
		existing, _ := s.configStore.GetByName(ctx, *req.ConfigName)
		if existing != nil && existing.ID != id {
			appErr := errors.New(errors.CodeInvalidParam, "配置名称已存在")
			recordPollingFailure(ctx, s.db, s.auditWriter, auditinfra.PollingInput{
				ActionCode: constants.AuditActionPollingConfigUpdated, Summary: "拒绝更新为重复轮询配置名称",
				ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
				ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
				OperatorID: currentUserID, Result: constants.AuditResultDenied,
				IdentitySnapshot: pollingConfigIdentity(config), BeforeData: pollingConfigState(config),
			}, appErr)
			return nil, appErr
		}
		config.ConfigName = *req.ConfigName
	}
	if req.CardCondition != nil {
		config.CardCondition = *req.CardCondition
	}
	if req.CardCategory != nil {
		config.CardCategory = *req.CardCategory
	}
	if req.CarrierID != nil {
		config.CarrierID = req.CarrierID
	}
	if req.Priority != nil {
		config.Priority = *req.Priority
	}
	if req.RealnameCheckInterval != nil {
		config.RealnameCheckInterval = req.RealnameCheckInterval
	}
	if req.CarddataCheckInterval != nil {
		config.CarddataCheckInterval = req.CarddataCheckInterval
	}
	if req.PackageCheckInterval != nil {
		config.PackageCheckInterval = req.PackageCheckInterval
	}
	if req.ProtectCheckInterval != nil {
		config.ProtectCheckInterval = req.ProtectCheckInterval
	}
	if req.CardStatusCheckInterval != nil {
		config.CardStatusCheckInterval = req.CardStatusCheckInterval
	}
	if req.Description != nil {
		config.Description = *req.Description
	}
	config.UpdatedBy = &currentUserID

	err = runPollingTransaction(ctx, s.db, s.auditWriter, func(tx *gorm.DB) error {
		if err := s.configStore.WithTx(tx).Update(ctx, config); err != nil {
			return err
		}
		return writePollingAudit(ctx, tx, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigUpdated, Summary: "更新轮询配置",
			ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
			ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
			OperatorID: currentUserID, IdentitySnapshot: pollingConfigIdentity(config),
			BeforeData: pollingConfigState(&before), AfterData: pollingConfigState(config),
		})
	})
	if err != nil {
		appErr := errors.Wrap(errors.CodeInternalError, err, "更新轮询配置失败")
		recordPollingFailure(ctx, s.db, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigUpdated, Summary: "更新轮询配置失败",
			ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
			ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
			OperatorID: currentUserID, IdentitySnapshot: pollingConfigIdentity(config),
			BeforeData: pollingConfigState(&before), AfterData: pollingConfigState(config),
		}, appErr)
		return nil, appErr
	}

	s.notifyConfigChanged(ctx, "updated")
	return s.toResponse(config), nil
}

// Delete 删除轮询配置
func (s *ConfigService) Delete(ctx context.Context, id uint) error {
	currentUserID := middleware.GetUserIDFromContext(ctx)
	if currentUserID == 0 {
		return errors.New(errors.CodeUnauthorized, "未授权访问")
	}
	config, err := s.configStore.GetByID(ctx, id)
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return errors.New(errors.CodePollingConfigNotFound, "轮询配置不存在")
		}
		return errors.Wrap(errors.CodeInternalError, err, "获取轮询配置失败")
	}

	err = runPollingTransaction(ctx, s.db, s.auditWriter, func(tx *gorm.DB) error {
		if err := s.configStore.WithTx(tx).Delete(ctx, id); err != nil {
			return err
		}
		return writePollingAudit(ctx, tx, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigDeleted, Summary: "删除轮询配置",
			ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
			ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
			OperatorID: currentUserID, IdentitySnapshot: pollingConfigIdentity(config), BeforeData: pollingConfigState(config),
		})
	})
	if err != nil {
		appErr := errors.Wrap(errors.CodeInternalError, err, "删除轮询配置失败")
		recordPollingFailure(ctx, s.db, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigDeleted, Summary: "删除轮询配置失败",
			ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
			ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
			OperatorID: currentUserID, IdentitySnapshot: pollingConfigIdentity(config), BeforeData: pollingConfigState(config),
		}, appErr)
		return appErr
	}

	s.notifyConfigChanged(ctx, "deleted")
	return nil
}

// List 列表查询轮询配置
func (s *ConfigService) List(ctx context.Context, req *dto.PollingConfigListRequest) ([]*dto.PollingConfigResponse, int64, error) {
	opts := &store.QueryOptions{
		Page:     req.Page,
		PageSize: req.PageSize,
		OrderBy:  "priority ASC, id DESC",
	}
	if opts.Page == 0 {
		opts.Page = 1
	}
	if opts.PageSize == 0 {
		opts.PageSize = constants.DefaultPageSize
	}

	filters := make(map[string]interface{})
	if req.Status != nil {
		filters["status"] = *req.Status
	}
	if req.CardCondition != nil {
		filters["card_condition"] = *req.CardCondition
	}
	if req.CardCategory != nil {
		filters["card_category"] = *req.CardCategory
	}
	if req.CarrierID != nil {
		filters["carrier_id"] = *req.CarrierID
	}
	if req.ConfigName != nil {
		filters["config_name"] = *req.ConfigName
	}

	configs, total, err := s.configStore.List(ctx, opts, filters)
	if err != nil {
		return nil, 0, errors.Wrap(errors.CodeInternalError, err, "查询轮询配置列表失败")
	}

	responses := make([]*dto.PollingConfigResponse, len(configs))
	for i, c := range configs {
		responses[i] = s.toResponse(c)
	}

	return responses, total, nil
}

// UpdateStatus 更新配置状态（启用/禁用）
func (s *ConfigService) UpdateStatus(ctx context.Context, id uint, status int16) error {
	currentUserID := middleware.GetUserIDFromContext(ctx)
	if currentUserID == 0 {
		return errors.New(errors.CodeUnauthorized, "未授权访问")
	}

	config, err := s.configStore.GetByID(ctx, id)
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return errors.New(errors.CodePollingConfigNotFound, "轮询配置不存在")
		}
		return errors.Wrap(errors.CodeInternalError, err, "获取轮询配置失败")
	}

	before := pollingConfigState(config)
	err = runPollingTransaction(ctx, s.db, s.auditWriter, func(tx *gorm.DB) error {
		if err := s.configStore.WithTx(tx).UpdateStatus(ctx, id, status, currentUserID); err != nil {
			return err
		}
		config.Status = status
		return writePollingAudit(ctx, tx, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigStatusUpdated, Summary: "更新轮询配置状态",
			ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
			ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
			OperatorID: currentUserID, IdentitySnapshot: pollingConfigIdentity(config),
			BeforeData: before, AfterData: pollingConfigState(config),
		})
	})
	if err != nil {
		appErr := errors.Wrap(errors.CodeInternalError, err, "更新轮询配置状态失败")
		recordPollingFailure(ctx, s.db, s.auditWriter, auditinfra.PollingInput{
			ActionCode: constants.AuditActionPollingConfigStatusUpdated, Summary: "更新轮询配置状态失败",
			ResourceType: constants.AuditResourcePollingConfig, ResourceID: config.ID,
			ResourceKey: pollingManualTriggerKey(config.ID), DisplayName: config.ConfigName,
			OperatorID: currentUserID, IdentitySnapshot: pollingConfigIdentity(config), BeforeData: before,
		}, appErr)
		return appErr
	}

	s.notifyConfigChanged(ctx, "updated")
	return nil
}

// ListEnabled 获取所有启用的配置
func (s *ConfigService) ListEnabled(ctx context.Context) ([]*dto.PollingConfigResponse, error) {
	configs, err := s.configStore.ListEnabled(ctx)
	if err != nil {
		return nil, errors.Wrap(errors.CodeInternalError, err, "获取启用配置失败")
	}

	responses := make([]*dto.PollingConfigResponse, len(configs))
	for i, c := range configs {
		responses[i] = s.toResponse(c)
	}

	return responses, nil
}

// toResponse 转换为响应 DTO
func (s *ConfigService) toResponse(c *model.PollingConfig) *dto.PollingConfigResponse {
	return &dto.PollingConfigResponse{
		ID:                      c.ID,
		ConfigName:              c.ConfigName,
		CardCondition:           c.CardCondition,
		CardCategory:            c.CardCategory,
		CarrierID:               c.CarrierID,
		Priority:                c.Priority,
		RealnameCheckInterval:   c.RealnameCheckInterval,
		CarddataCheckInterval:   c.CarddataCheckInterval,
		PackageCheckInterval:    c.PackageCheckInterval,
		ProtectCheckInterval:    c.ProtectCheckInterval,
		CardStatusCheckInterval: c.CardStatusCheckInterval,
		Status:                  c.Status,
		Description:             c.Description,
		CreatedAt:               c.CreatedAt.Format(time.RFC3339),
		UpdatedAt:               c.UpdatedAt.Format(time.RFC3339),
	}
}