Function CheckRateLimit

model/token.go:422–492

CheckRateLimit 检查令牌的访问频率限制

(token *Token)

// CheckRateLimit 检查令牌的访问频率限制
func CheckRateLimit(token *Token) error {
	if token == nil {
		return errors.New("token不能为空")
	}

	// 如果没有设置限制，直接返回
	if token.RateLimitPerMinute <= 0 && token.RateLimitPerDay <= 0 {
		return nil
	}

	currentTime := time.Now()
	currentTimestamp := currentTime.Unix()

	// 获取当前时间的分钟级时间戳（用于分钟级限制）
	currentMinute := currentTime.Truncate(time.Minute).Unix()

	// 获取当天开始的时间戳（用于日级限制）
	currentDay := time.Date(currentTime.Year(), currentTime.Month(), currentTime.Day(), 0, 0, 0, 0, currentTime.Location()).Unix()

	// 检查是否需要重置计数器
	needUpdate := false
	originalToken := *token // 保存原始状态

	// 检查分钟级限制
	if token.RateLimitPerMinute > 0 {
		// 如果上次重置时间不在当前分钟内，重置分钟计数器
		if token.LastRateLimitReset < currentMinute {
			token.LastRateLimitReset = currentTimestamp
			needUpdate = true
		}

		// 计算当前分钟内的使用次数
		var minuteCount int64
		err := DB.Model(&TokenUsageLog{}).Where("token_id = ? AND created_at >= ?", token.Id, currentMinute).Count(&minuteCount).Error
		if err != nil {
			common.SysError("检查分钟级使用次数失败: " + err.Error())
			return errors.New("系统错误，请稍后再试")
		}

		if int(minuteCount) >= token.RateLimitPerMinute {
			return errors.New("超出分钟限制，请稍后再试")
		}
	}

	// 检查日级限制
	if token.RateLimitPerDay > 0 {
		// 计算当天内的使用次数
		var dayCount int64
		err := DB.Model(&TokenUsageLog{}).Where("token_id = ? AND created_at >= ?", token.Id, currentDay).Count(&dayCount).Error
		if err != nil {
			common.SysError("检查日级使用次数失败: " + err.Error())
			return errors.New("系统错误，请稍后再试")
		}

		if int(dayCount) >= token.RateLimitPerDay {
			return errors.New("超出日限制，请稍后再试")
		}
	}