一个注解搞定SpringBoot接口防刷

一，技术要点：springboot的基本知识，redis基本操作，

首先是写一个注解类：

importjava.lang.annotation.Retention;
importjava.lang.annotation.Target;

importstaticjava.lang.annotation.ElementType.METHOD;
importstaticjava.lang.annotation.RetentionPolicy.RUNTIME;

/**
*@authoryhq
*@date2018/9/1015:52
*/

@Retention(RUNTIME)
@Target(METHOD)
public@interfaceAccessLimit{

intseconds();
intmaxCount();
booleanneedLogin()defaulttrue;
}

拦截器中实现：

importcom.alibaba.fastjson.JSON;
importcom.example.demo.action.AccessLimit;
importcom.example.demo.redis.RedisService;
importcom.example.demo.result.CodeMsg;
importcom.example.demo.result.Result;
importorg.springframework.beans.factory.annotation.Autowired;
importorg.springframework.stereotype.Component;
importorg.springframework.web.method.HandlerMethod;
importorg.springframework.web.servlet.handler.HandlerInterceptorAdapter;

importjavax.servlet.http.HttpServletRequest;
importjavax.servlet.http.HttpServletResponse;
importjava.io.OutputStream;

/**
*@authoryhq
*@date2018/9/1016:05
*/


@Component
publicclassFangshuaInterceptorextendsHandlerInterceptorAdapter{

@Autowired
privateRedisServiceredisService;

@Override
publicbooleanpreHandle(HttpServletRequestrequest,HttpServletResponseresponse,Objecthandler)throwsException{

//判断请求是否属于方法的请求
if(handlerinstanceofHandlerMethod){

HandlerMethodhm=(HandlerMethod)handler;

//获取方法中的注解,看是否有该注解
AccessLimitaccessLimit=hm.getMethodAnnotation(AccessLimit.class);
if(accessLimit==null){
returntrue;
}
intseconds=accessLimit.seconds();
intmaxCount=accessLimit.maxCount();
booleanlogin=accessLimit.needLogin();
Stringkey=request.getRequestURI();
//如果需要登录
if(login){
//获取登录的session进行判断
//.....
key+=""+"1";//这里假设用户是1,项目中是动态获取的userId
}

//从redis中获取用户访问的次数
AccessKeyak=AccessKey.withExpire(seconds);
Integercount=redisService.get(ak,key,Integer.class);
if(count==null){
//第一次访问
redisService.set(ak,key,1);
}elseif(count< maxCount){
                //加1
                redisService.incr(ak,key);
            }else{
                //超出访问次数
                render(response,CodeMsg.ACCESS_LIMIT_REACHED); //这里的CodeMsg是一个返回参数
                return false;
            }
        }

        return true;

    }
    private void render(HttpServletResponse response, CodeMsg cm)throws Exception {
        response.setContentType("application/json;charset=UTF-8");
        OutputStream out = response.getOutputStream();
        String str  = JSON.toJSONString(Result.error(cm));
        out.write(str.getBytes("UTF-8"));
        out.flush();
        out.close();
    }
}

注册到springboot中

importcom.example.demo.ExceptionHander.FangshuaInterceptor;
importorg.springframework.beans.factory.annotation.Autowired;
importorg.springframework.context.annotation.Configuration;
importorg.springframework.web.servlet.config.annotation.InterceptorRegistry;
importorg.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;

/**
*@authoryhq
*@date2018/9/1015:58
*/
@Configuration
publicclassWebConfigextendsWebMvcConfigurerAdapter{

@Autowired
privateFangshuaInterceptorinterceptor;


@Override
publicvoidaddInterceptors(InterceptorRegistryregistry){
registry.addInterceptor(interceptor);
}
}

在Controller中加入注解

importcom.example.demo.result.Result;
importorg.springframework.stereotype.Controller;
importorg.springframework.web.bind.annotation.RequestMapping;
importorg.springframework.web.bind.annotation.ResponseBody;

/**
*@authoryhq
*@date2018/9/1015:49
*/

@Controller
publicclassFangshuaController{

@AccessLimit(seconds=5,maxCount=5,needLogin=true)
@RequestMapping("/fangshua")
@ResponseBody
publicResultfangshua(){


returnResult.success("请求成功");

}

审核编辑：刘清