springMVC系列（六）——拦截器实现自定义注解接口访问限制

逝兮诚

发布于 2019-10-30 13:14:33

9870

发布于 2019-10-30 13:14:33

文章被收录于专栏：代码人生代码人生

本文链接：https://blog.csdn.net/luo4105/article/details/72327249

通过自定义注解形式实现接口访问评论限制

1.自定义注解，访问限制值和时间放在注解的参数中

RequestLimit.java注解

@Retention(RetentionPolicy.RUNTIME)
public @interface RequestLimit {
         /**
     *
     * 允许访问的次数，默认值MAX_VALUE
     */
   intcount() defaultInteger.MAX_VALUE;
 
   /**
     *
     * 时间段，单位为毫秒，默认值十分钟
     */
   longtime() default600000;
 
}

2.拦截器主要代码

1.当拦截器获得客户端请求，redis生成根据组合code判断是否存在，

如果存在，取得计数器

a.如果计数器大于最大访问数量，return false;限制其访问

b.如果小于，计数器加1

如果不存在，创建一个带定时的、key为组合code的计数器，数字为1

/**
 * 接口的访问频次限制
 *@param request
 *@return
 */
private booleanvalidateCode(HttpServletRequest request,int maxSize, long timeOut){
    Stringcode= request.getParameter("params");
    Stringdatakey= "RequestLimit" + "Data"+ code;
    boolean vcode = true;
    redisUtils.incr(datakey);
    long count = redisUtils.incr(datakey);
    if (count == 1) {
       redisUtils.set(code, count, timeOut);
   }
   if(count> maxSize){
        logger.info("用户[" + code+ "]超过了限定的次数[" + maxSize + "]");
        vcode = false;
   }
    return vcode;
}

3. 拦截器实现类代码

@Component
public classRequestLimitInterceptor extends HandlerInterceptorAdapter {
   
    @Autowired
    private RedisUtils redisUtils;     //redis缓存操作，可以用hash对象替代
   
    @Override
    public boolean preHandle(HttpServletRequestrequest,HttpServletResponse response, Object handler){
       RequestLimitmethodRequestLimit= ((HandlerMethod) handler).getMethodAnnotation(RequestLimit.class);              //方法注解
       RequestLimitclassRequestLimit= ((HandlerMethod) handler).getBean().getClass().getAnnotation(RequestLimit.class);      //类注解
       boolean vcode = true;
       if(methodRequestLimit!= null){
           vcode = validateCode(request, methodRequestLimit.count(),methodRequestLimit.time());
       }elseif(classRequestLimit!= null){
           vcode = validateCode(request, classRequestLimit.count(), classRequestLimit.time());
       }
       return vcode;
    }
   
    /**
     * 接口的访问频次限制
     * @param request
     * @return
     */
    private boolean validateCode(HttpServletRequestrequest,int maxSize, long timeOut) {
       Stringcode= request.getParameter("params");
       Stringdatakey= "RequestLimit" + "Data"+ code;
       boolean vcode = true;
       redisUtils.incr(datakey);
       long count = redisUtils.incr(datakey);
       if (count == 1) {
            redisUtils.set(code, count, timeOut);
        }
        if (count > maxSize){
            logger.info("用户[" + code+ "]超过了限定的次数[" + maxSize + "]");
            vcode = false;
        }
       return vcode;
    }
}

配置拦截器

在springMVC配置文件中添加
<mvc:interceptors>     
   <mvc:interceptor>     
      <mvc:mapping path="/bls/*" /><!-- 如果不配置或/*,将拦截所有的Controller-->
      <beanclass="com.yoostar.support.interceptor.RequestLimitInterceptor" />
   </mvc:interceptor>
</mvc:interceptors>

拦截接口加注解