feat(架构): 完善基础架构设计

internal/domains/user/repositories/sms_code_repository.go

@@ -0,0 +1,120 @@
+package repositories
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+
+	"tyapi-server/internal/domains/user/entities"
+	"tyapi-server/internal/shared/interfaces"
+)
+
+// SMSCodeRepository 短信验证码仓储
+type SMSCodeRepository struct {
+	db     *gorm.DB
+	cache  interfaces.CacheService
+	logger *zap.Logger
+}
+
+// NewSMSCodeRepository 创建短信验证码仓储
+func NewSMSCodeRepository(db *gorm.DB, cache interfaces.CacheService, logger *zap.Logger) *SMSCodeRepository {
+	return &SMSCodeRepository{
+		db:     db,
+		cache:  cache,
+		logger: logger,
+	}
+}
+
+// Create 创建短信验证码记录
+func (r *SMSCodeRepository) Create(ctx context.Context, smsCode *entities.SMSCode) error {
+	if err := r.db.WithContext(ctx).Create(smsCode).Error; err != nil {
+		r.logger.Error("创建短信验证码失败", zap.Error(err))
+		return err
+	}
+
+	// 缓存验证码
+	cacheKey := r.buildCacheKey(smsCode.Phone, smsCode.Scene)
+	r.cache.Set(ctx, cacheKey, smsCode, 5*time.Minute)
+
+	return nil
+}
+
+// GetValidCode 获取有效的验证码
+func (r *SMSCodeRepository) GetValidCode(ctx context.Context, phone string, scene entities.SMSScene) (*entities.SMSCode, error) {
+	// 先从缓存查找
+	cacheKey := r.buildCacheKey(phone, scene)
+	var smsCode entities.SMSCode
+	if err := r.cache.Get(ctx, cacheKey, &smsCode); err == nil {
+		return &smsCode, nil
+	}
+
+	// 从数据库查找最新的有效验证码
+	if err := r.db.WithContext(ctx).
+		Where("phone = ? AND scene = ? AND expires_at > ? AND used_at IS NULL",
+			phone, scene, time.Now()).
+		Order("created_at DESC").
+		First(&smsCode).Error; err != nil {
+		return nil, err
+	}
+
+	// 缓存结果
+	r.cache.Set(ctx, cacheKey, &smsCode, 5*time.Minute)
+
+	return &smsCode, nil
+}
+
+// MarkAsUsed 标记验证码为已使用
+func (r *SMSCodeRepository) MarkAsUsed(ctx context.Context, id string) error {
+	now := time.Now()
+	if err := r.db.WithContext(ctx).
+		Model(&entities.SMSCode{}).
+		Where("id = ?", id).
+		Update("used_at", now).Error; err != nil {
+		r.logger.Error("标记验证码为已使用失败", zap.Error(err))
+		return err
+	}
+
+	r.logger.Info("验证码已标记为使用", zap.String("code_id", id))
+	return nil
+}
+
+// CleanupExpired 清理过期的验证码
+func (r *SMSCodeRepository) CleanupExpired(ctx context.Context) error {
+	result := r.db.WithContext(ctx).
+		Where("expires_at < ?", time.Now()).
+		Delete(&entities.SMSCode{})
+
+	if result.Error != nil {
+		r.logger.Error("清理过期验证码失败", zap.Error(result.Error))
+		return result.Error
+	}
+
+	if result.RowsAffected > 0 {
+		r.logger.Info("清理过期验证码完成", zap.Int64("count", result.RowsAffected))
+	}
+
+	return nil
+}
+
+// CountRecentCodes 统计最近发送的验证码数量
+func (r *SMSCodeRepository) CountRecentCodes(ctx context.Context, phone string, scene entities.SMSScene, duration time.Duration) (int64, error) {
+	var count int64
+	if err := r.db.WithContext(ctx).
+		Model(&entities.SMSCode{}).
+		Where("phone = ? AND scene = ? AND created_at > ?",
+			phone, scene, time.Now().Add(-duration)).
+		Count(&count).Error; err != nil {
+		r.logger.Error("统计最近验证码数量失败", zap.Error(err))
+		return 0, err
+	}
+
+	return count, nil
+}
+
+// buildCacheKey 构建缓存键
+func (r *SMSCodeRepository) buildCacheKey(phone string, scene entities.SMSScene) string {
+	return fmt.Sprintf("sms_code:%s:%s", phone, string(scene))
+}