Function IncreaseTokenUsageCount

model/token.go:391–419

IncreaseTokenUsageCount 增加Token的使用次数

(key string)

// IncreaseTokenUsageCount 增加Token的使用次数
func IncreaseTokenUsageCount(key string) error {
	if key == "" {
		return errors.New("key 不能为空")
	}

	// 获取当前日期
	currentDate := common.GetTimeString()[:10] // YYYY-MM-DD格式

	// 更新数据库
	err := DB.Model(&Token{}).Where(commonKeyCol+" = ?", key).Updates(map[string]interface{}{
		"total_usage_count": gorm.Expr("total_usage_count + 1"),
		"daily_usage_count": gorm.Expr("CASE WHEN last_usage_date = ? THEN daily_usage_count + 1 ELSE 1 END", currentDate),
		"last_usage_date":   currentDate,
		"accessed_time":     common.GetTimestamp(),
	}).Error

	// 更新缓存
	if common.RedisEnabled && err == nil {
		gopool.Go(func() {
			// 重新缓存Token信息
			token, getErr := GetTokenByKey(key, true) // 从DB获取最新数据
			if getErr == nil {
				_ = cacheSetToken(*token)
			}
		})
	}

	return err
}

// CheckRateLimit 检查令牌的访问频率限制
func CheckRateLimit(token *Token) error {