269 lines
7.5 KiB
Go
269 lines
7.5 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"crypto/rand"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
dbent "github.com/Wei-Shaw/sub2api/ent"
|
||
infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/errors"
|
||
"github.com/Wei-Shaw/sub2api/internal/pkg/pagination"
|
||
)
|
||
|
||
var (
|
||
ErrPromoCodeNotFound = infraerrors.NotFound("PROMO_CODE_NOT_FOUND", "promo code not found")
|
||
ErrPromoCodeExpired = infraerrors.BadRequest("PROMO_CODE_EXPIRED", "promo code has expired")
|
||
ErrPromoCodeDisabled = infraerrors.BadRequest("PROMO_CODE_DISABLED", "promo code is disabled")
|
||
ErrPromoCodeMaxUsed = infraerrors.BadRequest("PROMO_CODE_MAX_USED", "promo code has reached maximum uses")
|
||
ErrPromoCodeAlreadyUsed = infraerrors.Conflict("PROMO_CODE_ALREADY_USED", "you have already used this promo code")
|
||
ErrPromoCodeInvalid = infraerrors.BadRequest("PROMO_CODE_INVALID", "invalid promo code")
|
||
)
|
||
|
||
// PromoService 优惠码服务
|
||
type PromoService struct {
|
||
promoRepo PromoCodeRepository
|
||
userRepo UserRepository
|
||
billingCacheService *BillingCacheService
|
||
entClient *dbent.Client
|
||
authCacheInvalidator APIKeyAuthCacheInvalidator
|
||
}
|
||
|
||
// NewPromoService 创建优惠码服务实例
|
||
func NewPromoService(
|
||
promoRepo PromoCodeRepository,
|
||
userRepo UserRepository,
|
||
billingCacheService *BillingCacheService,
|
||
entClient *dbent.Client,
|
||
authCacheInvalidator APIKeyAuthCacheInvalidator,
|
||
) *PromoService {
|
||
return &PromoService{
|
||
promoRepo: promoRepo,
|
||
userRepo: userRepo,
|
||
billingCacheService: billingCacheService,
|
||
entClient: entClient,
|
||
authCacheInvalidator: authCacheInvalidator,
|
||
}
|
||
}
|
||
|
||
// ValidatePromoCode 验证优惠码（注册前调用）
|
||
// 返回 nil, nil 表示空码（不报错）
|
||
func (s *PromoService) ValidatePromoCode(ctx context.Context, code string) (*PromoCode, error) {
|
||
code = strings.TrimSpace(code)
|
||
if code == "" {
|
||
return nil, nil // 空码不报错，直接返回
|
||
}
|
||
|
||
promoCode, err := s.promoRepo.GetByCode(ctx, code)
|
||
if err != nil {
|
||
// 保留原始错误类型，不要统一映射为 NotFound
|
||
return nil, err
|
||
}
|
||
|
||
if err := s.validatePromoCodeStatus(promoCode); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return promoCode, nil
|
||
}
|
||
|
||
// validatePromoCodeStatus 验证优惠码状态
|
||
func (s *PromoService) validatePromoCodeStatus(promoCode *PromoCode) error {
|
||
if !promoCode.CanUse() {
|
||
if promoCode.IsExpired() {
|
||
return ErrPromoCodeExpired
|
||
}
|
||
if promoCode.Status == PromoCodeStatusDisabled {
|
||
return ErrPromoCodeDisabled
|
||
}
|
||
if promoCode.MaxUses > 0 && promoCode.UsedCount >= promoCode.MaxUses {
|
||
return ErrPromoCodeMaxUsed
|
||
}
|
||
return ErrPromoCodeInvalid
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ApplyPromoCode 应用优惠码（注册成功后调用）
|
||
// 使用事务和行锁确保并发安全
|
||
func (s *PromoService) ApplyPromoCode(ctx context.Context, userID int64, code string) error {
|
||
code = strings.TrimSpace(code)
|
||
if code == "" {
|
||
return nil
|
||
}
|
||
|
||
// 开启事务
|
||
tx, err := s.entClient.Tx(ctx)
|
||
if err != nil {
|
||
return fmt.Errorf("begin transaction: %w", err)
|
||
}
|
||
defer func() { _ = tx.Rollback() }()
|
||
|
||
txCtx := dbent.NewTxContext(ctx, tx)
|
||
|
||
// 在事务中获取并锁定优惠码记录（FOR UPDATE）
|
||
promoCode, err := s.promoRepo.GetByCodeForUpdate(txCtx, code)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 在事务中验证优惠码状态
|
||
if err := s.validatePromoCodeStatus(promoCode); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 在事务中检查用户是否已使用过此优惠码
|
||
existing, err := s.promoRepo.GetUsageByPromoCodeAndUser(txCtx, promoCode.ID, userID)
|
||
if err != nil {
|
||
return fmt.Errorf("check existing usage: %w", err)
|
||
}
|
||
if existing != nil {
|
||
return ErrPromoCodeAlreadyUsed
|
||
}
|
||
|
||
// 增加用户余额
|
||
if err := s.userRepo.UpdateBalance(txCtx, userID, promoCode.BonusAmount); err != nil {
|
||
return fmt.Errorf("update user balance: %w", err)
|
||
}
|
||
|
||
// 创建使用记录
|
||
usage := &PromoCodeUsage{
|
||
PromoCodeID: promoCode.ID,
|
||
UserID: userID,
|
||
BonusAmount: promoCode.BonusAmount,
|
||
UsedAt: time.Now(),
|
||
}
|
||
if err := s.promoRepo.CreateUsage(txCtx, usage); err != nil {
|
||
return fmt.Errorf("create usage record: %w", err)
|
||
}
|
||
|
||
// 增加使用次数
|
||
if err := s.promoRepo.IncrementUsedCount(txCtx, promoCode.ID); err != nil {
|
||
return fmt.Errorf("increment used count: %w", err)
|
||
}
|
||
|
||
if err := tx.Commit(); err != nil {
|
||
return fmt.Errorf("commit transaction: %w", err)
|
||
}
|
||
|
||
s.invalidatePromoCaches(ctx, userID, promoCode.BonusAmount)
|
||
|
||
// 失效余额缓存
|
||
if s.billingCacheService != nil {
|
||
go func() {
|
||
cacheCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
defer cancel()
|
||
_ = s.billingCacheService.InvalidateUserBalance(cacheCtx, userID)
|
||
}()
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func (s *PromoService) invalidatePromoCaches(ctx context.Context, userID int64, bonusAmount float64) {
|
||
if bonusAmount == 0 || s.authCacheInvalidator == nil {
|
||
return
|
||
}
|
||
s.authCacheInvalidator.InvalidateAuthCacheByUserID(ctx, userID)
|
||
}
|
||
|
||
// GenerateRandomCode 生成随机优惠码
|
||
func (s *PromoService) GenerateRandomCode() (string, error) {
|
||
bytes := make([]byte, 8)
|
||
if _, err := rand.Read(bytes); err != nil {
|
||
return "", fmt.Errorf("generate random bytes: %w", err)
|
||
}
|
||
return strings.ToUpper(hex.EncodeToString(bytes)), nil
|
||
}
|
||
|
||
// Create 创建优惠码
|
||
func (s *PromoService) Create(ctx context.Context, input *CreatePromoCodeInput) (*PromoCode, error) {
|
||
code := strings.TrimSpace(input.Code)
|
||
if code == "" {
|
||
// 自动生成
|
||
var err error
|
||
code, err = s.GenerateRandomCode()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
promoCode := &PromoCode{
|
||
Code: strings.ToUpper(code),
|
||
BonusAmount: input.BonusAmount,
|
||
MaxUses: input.MaxUses,
|
||
UsedCount: 0,
|
||
Status: PromoCodeStatusActive,
|
||
ExpiresAt: input.ExpiresAt,
|
||
Notes: input.Notes,
|
||
}
|
||
|
||
if err := s.promoRepo.Create(ctx, promoCode); err != nil {
|
||
return nil, fmt.Errorf("create promo code: %w", err)
|
||
}
|
||
|
||
return promoCode, nil
|
||
}
|
||
|
||
// GetByID 根据ID获取优惠码
|
||
func (s *PromoService) GetByID(ctx context.Context, id int64) (*PromoCode, error) {
|
||
code, err := s.promoRepo.GetByID(ctx, id)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return code, nil
|
||
}
|
||
|
||
// Update 更新优惠码
|
||
func (s *PromoService) Update(ctx context.Context, id int64, input *UpdatePromoCodeInput) (*PromoCode, error) {
|
||
promoCode, err := s.promoRepo.GetByID(ctx, id)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
if input.Code != nil {
|
||
promoCode.Code = strings.ToUpper(strings.TrimSpace(*input.Code))
|
||
}
|
||
if input.BonusAmount != nil {
|
||
promoCode.BonusAmount = *input.BonusAmount
|
||
}
|
||
if input.MaxUses != nil {
|
||
promoCode.MaxUses = *input.MaxUses
|
||
}
|
||
if input.Status != nil {
|
||
promoCode.Status = *input.Status
|
||
}
|
||
if input.ExpiresAt != nil {
|
||
promoCode.ExpiresAt = input.ExpiresAt
|
||
}
|
||
if input.Notes != nil {
|
||
promoCode.Notes = *input.Notes
|
||
}
|
||
|
||
if err := s.promoRepo.Update(ctx, promoCode); err != nil {
|
||
return nil, fmt.Errorf("update promo code: %w", err)
|
||
}
|
||
|
||
return promoCode, nil
|
||
}
|
||
|
||
// Delete 删除优惠码
|
||
func (s *PromoService) Delete(ctx context.Context, id int64) error {
|
||
if err := s.promoRepo.Delete(ctx, id); err != nil {
|
||
return fmt.Errorf("delete promo code: %w", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// List 获取优惠码列表
|
||
func (s *PromoService) List(ctx context.Context, params pagination.PaginationParams, status, search string) ([]PromoCode, *pagination.PaginationResult, error) {
|
||
return s.promoRepo.ListWithFilters(ctx, params, status, search)
|
||
}
|
||
|
||
// ListUsages 获取使用记录
|
||
func (s *PromoService) ListUsages(ctx context.Context, promoCodeID int64, params pagination.PaginationParams) ([]PromoCodeUsage, *pagination.PaginationResult, error) {
|
||
return s.promoRepo.ListUsagesByPromoCode(ctx, promoCodeID, params)
|
||
}
|