一、原理:
- 按固定的速率往令牌桶中放入令牌,令牌桶中最多存放N个令牌,当桶满时新添加的令牌将被丢弃
- 每次请求将从桶中移除一个令牌,当令牌桶为空时拒绝请求;
二、引入依赖
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
<version>32.0.0</version>
</dependency>
三、代码实现
// 注册令牌桶,指定每秒放入多少个令牌
private RateLimiter rateLimiter = RateLimiter.create(1000);
@PostMapping(value="/test")
public String test(@RequestBody JSONObject params){
// 如果获取到令牌则往下执行业务逻辑
if(rateLimiter.tryAcquire(2,TimeUnit.SECONDS)){
// TODO 业务逻辑
return service.exec();
} else {
return "服务器正忙,请稍后再试";
}
}