背景:在秒杀中,我们需要实现限制客户端的流量,常用的实现限流的方式有:hystrix、nginx、guava等方式。下面我们介绍下guava如何实现限流,限流算法:令牌桶算法。
原理:以规定的速率往令牌桶中存入token,用户请求必须获取到令牌中的token才可以处理请求,如果没有从令牌桶中获取到令牌则丢失该请求。
1.maven引入
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
<version>18.0</version>
</dependency>
2.测试类
package com.huaj.controller;
import com.google.common.util.concurrent.RateLimiter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.servlet.http.HttpServletRequest;
import java.util.concurrent.TimeUnit;
/**
* @author huaj
* @date 2020/2/28
* @description
*/
@RestController
public class TestController{
private static final Logger logger = LoggerFactory.getLogger(TestController.class);
//创建一个限流器,参数代表每秒生成的令牌数(用户限流频率设置 每秒中限制1个请求,高并发时QPS为1)
private RateLimiter rateLimiter = RateLimiter.create(1);
@GetMapping("/test")
public String test(HttpServletRequest request){
//设置等待超时时间的方式获取令牌,如果超timeout为0,则代表非阻塞,获取不到立即返回
boolean tryAcquire = rateLimiter.tryAcquire(0, TimeUnit.SECONDS);
if (!tryAcquire) {
return "现在抢购的人数过多,请稍等一下哦!";
}
return "抢购排队中...";
}
}