深入浅出：限流的三个核心思路与实践

<?php
function isRateLimited($userId, $action, $limit, $interval) {
    $redis = new Redis();
    $redis->connect('127.0.0.1', 6379);
 
    // 构造唯一的键，例如：rate_limit:user_123:login
    $key = "rate_limit:user_{$userId}:{$action}";
    
    // 获取当前计数
    $currentCount = $redis->get($key);
    if ($currentCount === false) {
        // 键不存在，说明是窗口内的第一个请求，设置键并过期时间
        $redis->setex($key, $interval, 1);
        return false; // 不限流
    }
 
    if ($currentCount < $limit) {
        // 计数器加1
        $redis->incr($key);
        return false; // 不限流
    } else {
        return true; // 限流
    }
}
 
// 使用示例：检查用户123的登录操作是否在60秒内超过5次
if (isRateLimited(123, 'login', 5, 60)) {
    http_response_code(429);
    echo "请求过于频繁，请稍后再试。";
    exit;
}
// 正常处理请求

from time import time
 
class SlidingWindow:
    def __init__(self, limit, window_size_sec):
        self.limit = limit
        self.window_size_sec = window_size_sec
        # 使用一个列表来存储每次请求的时间戳
        self.requests = []
 
    def allow_request(self):
        now = time()
        # 1. 移除过期的时间戳（超出当前窗口范围的请求）
        cutoff_time = now - self.window_size_sec
        # 保留时间戳大于 cutoff_time 的请求
        self.requests = [req_time for req_time in self.requests if req_time > cutoff_time]
 
        # 2. 检查当前窗口内的请求数是否超过阈值
        if len(self.requests) < self.limit:
            self.requests.append(now)
            return True  # 允许请求
        else:
            return False # 拒绝请求
 
# 使用示例：限制每秒10个请求
limiter = SlidingWindow(limit=10, window_size_sec=1)
for i in range(15):
    if limiter.allow_request():
        print(f"请求 {i} 被允许")
    else:
        print(f"请求 {i} 被限流")

import com.google.common.util.concurrent.RateLimiter;
 
public class TokenBucketDemo {
    public static void main(String[] args) {
        // 创建一个每秒产生2个令牌的限流器
        RateLimiter rateLimiter = RateLimiter.create(2.0); // 2 permits per second
 
        for (int i = 0; i < 10; i++) {
            // 尝试获取1个令牌，如果没有则等待，直到获取成功
            double waitTime = rateLimiter.acquire();
            System.out.println("处理请求 " + i + ", 等待了 " + waitTime + " 秒");
 
            // 非阻塞尝试
            /*
            if (rateLimiter.tryAcquire()) {
                // 成功获取令牌，处理请求
                System.out.println("处理请求 " + i);
            } else {
                // 获取失败，立即返回或执行降级策略
                System.out.println("请求被限流 " + i);
            }
            */
        }
    }
}