在使用SpringBoot做接口访问如何做接口的限流，这里我们可以使用google的Guava包来实现.

1.在项目中引入Guava相关包

  1.     <dependency>
  2.         <groupId>com.google.guava</groupId>
  3.         <artifactId>guava</artifactId>
  4.         <version>23.5-jre</version>
  5.     </dependency>

2.创建拦截器

  1. public class AuthInterceptor extends HandlerInterceptorAdapter {
  3.     enum LimiterTypeEnum {
  4.         DROP,//丢弃
  5.         WAIT //等待
  6.     }
  9.     //限流器
  10.     private RateLimiter rateLimiter;
  11.     //限流类型
  12.     private LimiterTypeEnum limiterType;
  14.     /**
  15.      * @param permitsPerSecond 限流量 (每秒处理量)
  16.      * @param limiterType      限流类型
  17.      */
  18.     public AuthInterceptor(int permitsPerSecond, LimiterTypeEnum limiterType) {
  19.         this.rateLimiter = RateLimiter.create(permitsPerSecond);
  20.         this.limiterType = limiterType;
  21.     }
  23.     /**
  24.      * @param permitsPerSecond 每秒新增的令牌数(这个是平滑增长,有一个预热过程)
  25.      * @param limiterType      限流类型
  26.      */
  27.     public AuthInterceptor(double permitsPerSecond, LimiterTypeEnum limiterType) {
  28.         this.rateLimiter = RateLimiter.create(permitsPerSecond, 1, TimeUnit.SECONDS);
  29.         this.limiterType = limiterType;
  30.     }
  32.     @Override
  33.     public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
  34.         if (Objects.equals(limiterType, LimiterTypeEnum.DROP)) {
  35.             return rateLimiter.tryAcquire(); //false为被限流
  36.         } else {
  37.             rateLimiter.acquire();//一直等待,直到拿到令牌
  38.             return true;
  39.         }
  40.     }
  42. }

3.添加拦截器

  1. @Configuration
  2. public class AuthHandlerAdapter extends WebMvcConfigurerAdapter {
  4.     @Override
  5.     public void addInterceptors(InterceptorRegistry registry) {
  6.         registry.addInterceptor(new AuthInterceptor(10, AuthInterceptor.LimiterTypeEnum.DROP))
  7.                 .addPathPatterns("/*");
  8.     }
  9. }

4.测试

  • controller```java @RestController public class TestController {

    @GetMapping public String TestGetString() {

    1.   return "爱生活,爱java";

    } } ```

  • 并发请求```java public class RequestTest { public static void main(String[] args) {

    1.    HttpClient httpClient = new HttpClient("http://127.0.0.1:8080", 2000, 2000);
    2.    IntStream.range(0, 1000).forEach(value -> {
    3.        testSend(httpClient);
    4.        try {
    5.            TimeUnit.MILLISECONDS.sleep(100);
    6.        } catch (InterruptedException e) {
    7.            e.printStackTrace();
    8.        }
    10.    });

    } public static void testSend(HttpClient httpClient) {

    1.    try {
    2.        httpClient.sendGet("UTF-8");
    3.    } catch (Exception e) {
    4.        e.printStackTrace();
    5.    }

    } } ```