php如何实现接口限流功能_php接口限流算法与代码实现

function isAllowedFixedWindow($userId, $limit = 100, $window = 60) {
    $redis = new Redis();
    $redis->connect('127.0.0.1', 6379);

    $key = "rate_limit:fixed:" . $userId;
    $current = $redis->incr($key);

    if ($current == 1) {
        $redis->expire($key, $window); // 设置过期时间为窗口长度
    }

    return $current <= $limit;
}

function isAllowedSlidingWindow($userId, $limit = 100, $window = 60) {
    $redis = new Redis();
    $redis->connect('127.0.0.1', 6379);

    $key = "rate_limit:sliding:" . $userId;
    $now = microtime(true);

    // 移除窗口外的旧请求
    $redis->zRemRangeByScore($key, 0, $now - $window);

    // 获取当前窗口内的请求数
    $count = $redis->zCard($key);

    if ($count < $limit) {
        $redis->zAdd($key, $now, $now); // 添加当前请求
        $redis->expire($key, $window);  // 确保 key 过期
        return true;
    }

    return false;
}

function isAllowedTokenBucket($userId, $capacity = 100, $rate = 10, $window = 60) {
    $redis = new Redis();
    $redis->connect('127.0.0.1', 6379);

    $key = "rate_limit:token:" . $userId;
    $now = microtime(true);

    $data = $redis->hGetAll($key);

    if (empty($data)) {
        // 初始状态：桶满
        $tokens = $capacity;
        $lastRefillTime = $now;
    } else {
        $tokens = (float)$data['tokens'];
        $lastRefillTime = (float)$data['last_refill'];
    }

    // 补充令牌：按时间比例补充
    $elapsed = $now - $lastRefillTime;
    $newTokens = $elapsed * ($rate / $window); // 每秒补充 rate/window 个令牌
    $tokens = min($capacity, $tokens + $newTokens); // 不超过容量

    if ($tokens >= 1) {
        $tokens -= 1;
        $redis->hMSet($key, [
            'tokens' => $tokens,
            'last_refill' => $now
        ]);
        $redis->expire($key, $window * 2); // 设置合理过期时间
        return true;
    }

    // 更新时间戳，即使没令牌也更新 last_refill？
    // 可选：只在有请求时才更新
    $redis->hSet($key, 'last_refill', $now);
    return false;
}

<p>基本上就这些。选择哪种算法取决于你的业务需求：固定窗口实现简单，滑动窗口更精确，令牌桶支持突发流量。生产环境中建议结合 Redis 使用，保证分布式环境下的一致性。注意加异常处理和连接池管理，避免因 Redis 故障影响主流程。不复杂但容易忽略细节。