使用Guava实现限流器

response.setStatus(HttpServletResponse.SC_OK);

response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);

PrintWriter printWriter = null;

try {

printWriter = response.getWriter();

printWriter.write(JsonUtils.toJson(responseDto));

} catch (Exception e) {

logger.error(“handlerResponse catch a exception:” + e.getMessage());

} finally {

if (printWriter != null) {

printWriter.close();

}

}

}

}

b）定义流量控制拦截器，流量控制拦截器继承自上面的拦截器抽象类，在preFilter方法中进行流量控制。

/**

• @author linzhiqiang

• @date 2019/4/17

*/

@Component(“rateLimitInterceptor”)

public class RateLimitInterceptor extends AbstractInterceptor {

private Logger logger = LoggerFactory.getLogger(RateLimitInterceptor.class);

/**

• 单机全局限流器(限制QPS为250)

*/

private static final RateLimiter rateLimiter = RateLimiter.create(300);

public static void setRate(double limiterQPS){

rateLimiter.setRate(limiterQPS);

}

@Override

protected ResponseEnum preFilter(HttpServletRequest request) {

if (!rateLimiter.tryAcquire()) {

logger.warn(“限流中…”);

return ResponseEnum.RATE_LIMIT;

}

return ResponseEnum.SUCCESS;

}

}

使用Guava提供的RateLimiter类来实现流量控制，过程很简单：定义了一个QPS为1的全局限流器（便于测试），使用tryAcquire()方法来尝试获取令牌，如果成功则返回ResponseEnum.OK，否则返回ResponseEnum.RATE_LIMIT。

3.继承WebMvcConfigurerAdapter来添加自定义拦截器

@Configuration

public class MyWebAppConfigurer extends WebMvcConfigurationSupport {

@Override

public void addInterceptors(InterceptorRegistry registry) {

// 多个拦截器组成一个拦截器链

// addPathPatterns 用于添加拦截规则

// excludePathPatterns 用户排除拦截

registry.addInterceptor(new RateLimitInterceptor()).addPathPatterns(“/**”);

super.addInterceptors(registry);

}

}

4.写一个Controller来提供一个简单的访问接口

/**

• request测试专用

• @return

*/

@RequestMapping(value = “getUserList”, method = RequestMethod.GET)

public String getUserList() {

String result = null;

try {

result = “请求成功”;

}catch (Exception e){

logger.error(“请求失败”, e);

return JsonUtils.toJson(ResponseUtils.failInServer(result));

}

return JsonUtils.toJson(ResponseUtils.success(result));

}

上文使用到的ResponseEnum是一个返回Code的枚举：

/**

• @description:

• @Date : 2019/3/19 19:04

• @Author : 樊康康-(kangkang.fan@mljr.com)

*/

public enum ResponseEnum {

SUCCESS(“200”,“000000”,“请求成功”),

FAIL(“200”,“100000”,“请求失败”),

FAIL_BY_PARAMS(“200”,“200000”,“请求参数异常”),

FAIL_IN_SERVER(“200”,“300000”,“服务器内部异常”),

RATE_LIMIT(“200”,“400000”,“限流中”);

public String status;

public String code;

public String message;

ResponseEnum(String s, String s1, String s2) {

this.status = s;

this.code = s1;

this.message = s2;

}

}

5.使用Postman来测试接口

快速并且反复的调用接口，可以很容易的看到两种结果。

成功通过限流器的结果：

没有成功通过限流器的返回结果：

反复调用时，Console输出如下：

至此，简单的限流器实现完成。
文章转载于：https://wolzq.com

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

最后

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
alt=“img” style=“zoom: 33%;” />

最后

[外链图片转存中…(img-VUW1nhC1-1712760333188)]

[外链图片转存中…(img-tJV67RX2-1712760333188)]

[外链图片转存中…(img-IT2C9AxO-1712760333188)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！