限流算法：保障系统稳定运行的利器

2023-12-22 14:12:31

限流算法：保障系统稳定的守护者

在高并发场景下，保护系统免于不堪重负的侵袭至关重要。而限流算法正是保障系统稳定运行的利刃，通过控制请求访问速率，防止系统被突发流量淹没。

常见的限流算法

业界广泛应用的限流算法主要包括以下四种：

令牌桶算法： 设想一个装有令牌的桶，当请求到来时，需要从桶中获取一个令牌。如果桶中无令牌，则请求被拒绝。
漏桶算法： 类似于令牌桶算法，但桶底部有洞。请求放入桶中，如果桶已满，则新请求将被丢弃。
滑动窗口算法： 将时间窗口划分为固定子窗口，每个子窗口仅允许一定数量的请求通过。当子窗口满时，将丢弃该子窗口，并创建一个新的子窗口。
连接数限制： 设定最大同时连接数，超出上限的新连接请求将被拒绝。

令牌桶算法的Java实现

public class TokenBucketLimiter {
    private int capacity; // 令牌桶容量
    private int tokens; // 当前令牌数
    private long lastUpdateTime; // 上次更新时间

    public TokenBucketLimiter(int capacity) {
        this.capacity = capacity;
        this.tokens = capacity;
        this.lastUpdateTime = System.currentTimeMillis();
    }

    public synchronized boolean tryAcquire() {
        // 计算新增令牌数
        int newTokens = (int) ((System.currentTimeMillis() - lastUpdateTime) / 1000 * capacity);
        tokens = Math.min(capacity, tokens + newTokens);
        lastUpdateTime = System.currentTimeMillis();
        // 判断是否有足够令牌
        if (tokens > 0) {
            tokens--;
            return true;
        }
        return false;
    }
}