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

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

部分模拟代码示例:

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);
            lastRe
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值