java高级进阶|单机版限流之Ratelimiter

0x01，感悟

0x02，文章整理

这篇文章的内容是自己在18年的时候写了一下示例程序，但是没有将它整理成文章，所以这里就简单整理了一下，顺便用了一下测试工具jmeter，很简单，毕竟不是专业的测试工具，若后面需要再继续看下jmeter的用法吧，这里就先有个概念。

0x03，拦截器和限流组件的整合

本文基于拦截器和guava提供的Ratelimiter进行请求的限流的示例程序的模拟的，文章整合起来很简单，限流主要为了防止过量的请求对服务器造成的压力而拒绝服务了，所以这里就简单模拟了一下，具体的示例程序如下。

0x04，需要的jar包依赖信息

org.springframework.bootspring-boot-starter-webcom.google.guavaguava27.1-jre


0x05，示例程序

其次就是编写拦截器和注册拦截器的示例程序了。

package com.wpw.springbootratelimiter;import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;public abstract class AbstractIntercepter extends HandlerInterceptorAdapter {@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {return preHandle(request);}protected abstract boolean preHandle(HttpServletRequest request);
}
package com.wpw.springbootratelimiter;import com.google.common.util.concurrent.RateLimiter;
import org.springframework.stereotype.Component;import javax.servlet.http.HttpServletRequest;@Component
public class RatelimiterInterceptor extends AbstractIntercepter {private static final RateLimiter rateLimiter = RateLimiter.create(1);@Overrideprotected boolean preHandle(HttpServletRequest request) {if (!rateLimiter.tryAcquire()) {System.out.println("稍后重试");return false;}System.out.println("获取许可成功");return true;}
}
package com.wpw.springbootratelimiter;import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;@Component
public class WebMvcConfigurationSupport implements WebMvcConfigurer {@Autowiredprivate RatelimiterInterceptor ratelimiterInterceptor;@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(ratelimiterInterceptor);}
}
package com.wpw.springbootratelimiter;import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;/*** @author wpw*/
@RestController
public class UserController {@GetMapping(value = "/hello")public String hello() {return "access success";}
}

0x06，测试工具的使用

测试程序可以基于postman测试工具测试，不过为了模拟并发的请求，这里采用了jemeter进行了测试，截个图，不会jemeter的可以自己看下教程，这里就不做演示了。

看到后台的日志信息打印，发现了下面的重试信息的输出，说明我们的限流作用起作用了

0x07，总结

理解限流可以去理解火车站排队的场景或者去景点玩的场景，这里就不过多介绍了，这里自己就是为了总结一下和回顾一下自己的知识点，梳理一下。