在对一个秒杀系统进行接口访问限流,感觉计数法太过于简单粗暴,便实现一个简单的令牌桶限流
考虑到并发问题,将用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;