Springboot实现令牌桶限流

在对一个秒杀系统进行接口访问限流,感觉计数法太过于简单粗暴,便实现一个简单的令牌桶限流

考虑到并发问题,将用AtomicInteger变量存储size,声明为volatile

令牌桶

大小固定的令牌桶可自行以恒定的速率源源不断地产生令牌。如果令牌不被消耗,或者被消耗的速度小于产生的速度,令牌就会不断地增多,直到把桶填满。后面再产生的令牌就会从桶中溢出。最后桶中可以保存的最大令牌数永远不会超过桶的大小。


设计思路

  • 实现令牌桶类
  • HashMap<>存储多个桶,为各个接口提供服务,静态便于访问
  • 记得要重写hash()equal()方法
  • 配置Spring Boot计时器,实现按一定速率生产令牌
  • 每次访问接口时,取走一个令牌

令牌桶实现类


import java.util.HashMap;
import java.util.Objects;

public class Bucket {
   

    //默认容量10 
    static final int DEFAULT_MAX_COUNT = 10;
    // 默认增长速率为1
    static final int DEFAULT_CREATE_RATE = 1;
  • 3
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值