Redis实现接口限流

准备

1. 自定一个redis接口限流注解

   @Target(ElementType.METHOD)
   @Retention(RetentionPolicy.RUNTIME)
   @Documented
   public @interface AccessLimit {
   
       /**
        * 单位时间（秒）
        *
        * @return int
        */
       int seconds();
   
       /**
        * 单位时间最大请求次数
        *
        * @return int
        */
       int maxCount();
   }
   
   java

2. 标记在你想要限流的接口上

   @AccessLimit(seconds = 6, maxCount = 1)
   @ApiOperation(value = "发送邮箱验证码")
   @ApiImplicitParam(name = "username", value = "用户名", required = true, dataType = "String")
   @GetMapping("/users/code")
   public Result<?> sendCode(String username) {
       userAuthService.sendCode(username);
       return Result.ok();
   }
   java

拦截

1. 自定义限流拦截器AccessLimitHandler

   /**
    * @author: ladidol
    * @date: 2022/11/28 12:25
    * @description: 限流拦截器
    */
   @Log4j2
   public class AccessLimitHandler implements HandlerInterceptor {
       @Autowired
       private RedisService redisService;
   
       @Override
       public boolean preHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object handler) throws Exception {
           // 如果请求输入方法
           if (handler instanceof HandlerMethod) {
               HandlerMethod hm = (HandlerMethod) handler;
               // 获取方法中的注解,看是否有该注解
               AccessLimit accessLimit = hm.getMethodAnnotation(AccessLimit.class);
               if (accessLimit != null) {
                   long seconds = accessLimit.seconds();
                   int maxCount = accessLimit.maxCount();
                   // 关于key的生成规则可以自己定义 本项目需求是对每个方法都加上限流功能，如果你只是针对ip地址限流，那么key只需要只用ip就好
                   String key = IpUtils.getIpAddress(httpServletRequest) + hm.getMethod().getName();
                   // 从redis中获取用户访问的次数
                   try {
                       // 此操作代表获取该key对应的值自增1后的结果
                       long q = redisService.incrExpire(key, seconds);
                       if (q > maxCount) {
                           render(httpServletResponse, Result.fail("请求过于频繁，请稍候再试"));
                           log.warn(key + "请求次数超过每" + seconds + "秒" + maxCount + "次");
                           return false;
                       }
                       return true;
                   } catch (RedisConnectionFailureException e) {
                       log.warn("redis错误: " + e.getMessage());
                       return false;
                   }
               }
           }
           return true;
       }
   
       /**
        * 作者：Ladidol
        * 描述：将不通过的结果返回给前端
        */
       private void render(HttpServletResponse response, Result<?> result) throws Exception {
           response.setContentType(APPLICATION_JSON);
           OutputStream out = response.getOutputStream();
           String str = JSON.toJSONString(result);
           out.write(str.getBytes(StandardCharsets.UTF_8));
           out.flush();
           out.close();
       }
   
   }
   java

2. 在WebMvcConfigurer的实现类中注册拦截器

   /** <p>
       * 作者：Ladidol
       * 描述：WebMvc注册拦截器
       */
       @Override
       public void addInterceptors(InterceptorRegistry registry) {
   
       //注册限流拦截器
       registry.addInterceptor(getAccessLimitHandler());
   }
   java

另一种实现方式：[我司用了 6 年的 Redis 分布式限流器，可以说是非常厉害了！ (qq.com)](