限制进入系统的请求在1w之内，限流的设计方案

最新推荐文章于 2025-12-05 17:02:52 发布

原创

最新推荐文章于 2025-12-05 17:02:52 发布 · 262 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

本文介绍了使用Java实现请求限流的方案，主要利用令牌桶算法，确保系统每秒只处理1w个请求。通过创建一个令牌桶数据结构，设置固定生成速率，当请求到来时检查令牌数量决定是否允许通过。此外，还探讨了如何在Spring Boot框架中实现限流，并提到了固定窗口、滑动窗口和漏斗算法作为额外的限流策略。同时，文章提及基于IP地址的限流策略，考虑了IP伪造和代理的场景。

1、创建一个令牌桶数据结构用于存储令牌。

设定令牌桶的容量为1万个令牌，并设置一个固定的生成速率，例如每秒生成1万个令牌。
在系统启动时，启动一个定时任务，每秒向令牌桶中添加1万个令牌，保持令牌桶的容量不超过1万个令牌。
当有请求进入系统时，首先检查令牌桶中是否有足够的令牌。
如果令牌桶中有足够的令牌，则取出一个令牌，表示允许请求通过，并处理该请求。
如果令牌桶中没有足够的令牌，则拒绝请求。
可以设置一个超时机制，当请求被拒绝时，可以等待一段时间后重新尝试

部分模拟代码示例：

import java.time.Instant;
import java.util.concurrent.TimeUnit;

public class TokenBucket {
    private final long capacity; // 令牌桶容量
    private final double refillRate; // 令牌生成速率（每秒生成的令牌数）
    private double tokens; // 当前令牌数量
    private Instant lastRefillTime; // 上次令牌生成时间

    public TokenBucket(long capacity, double refillRate) {
        this.capacity = capacity;
        this.refillRate = refillRate;
        this.tokens = capacity;
        this.lastRefillTime = Instant.now();
    }

    public synchronized boolean allowRequest(int tokensRequested) {
        refillTokens();

        if (tokens >= tokensRequested) {
            tokens -= tokensRequested;
            return true;
        }
        return false;
    }

    private void refillTokens() {
        Instant now = Instant.now();
        long timeElapsed = now.toEpochMilli() - lastRefillTime.toEpochMilli();
        double tokensToAdd = timeElapsed * refillRate / 1000.0; // 转换为秒

        if (tokensToAdd > 0) {
            tokens = Math.min(capacity, tokens + tokensToAdd);