package com.hx.repeat.check;
|
|
import com.hx.common.annotations.repeat.RequestRepeat;
|
import com.hx.exception.TipsException;
|
import com.hx.redis.RedisUtil;
|
import com.hx.util.DateUtil;
|
import com.hx.util.IPUtils;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.lang.reflect.Method;
|
import java.util.Date;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* URL防重复提交
|
* @author CJH
|
*/
|
public class RequestRepeatUtil {
|
|
/**请求重复校验
|
* 目前区分重复是IP地址和请求方法，如果想要更精准，传code
|
* @param code 前缀标识，用于区分不同请求来源，长度最好12位之内，可空
|
* @param request HttpServletRequest
|
* @param redisUtil redis
|
* @param millisecond 间隔时间，毫秒（默认500毫秒）
|
* @param annotateCheck 是否注解校验，如果是，将以注解来做校验，否则就全部做校验
|
* @param method 请求方法，方法或者类上面有校验注解，注解跳过或者拦截
|
*/
|
public static void checkRequest(String code,HttpServletRequest request, RedisUtil redisUtil, Long millisecond
|
, boolean annotateCheck, Method method){
|
//优先校验注解，优先方法再到类
|
RequestRepeat requestRepeat = method.getAnnotation(RequestRepeat.class);
|
if(requestRepeat != null){
|
if(requestRepeat.isRepeat()){
|
check(code,request,redisUtil,requestRepeat.millisecond());
|
}else{
|
return;
|
}
|
}else{
|
requestRepeat = method.getClass().getAnnotation(RequestRepeat.class);
|
if(requestRepeat != null){
|
if(requestRepeat.isRepeat()){
|
check(code,request,redisUtil,requestRepeat.millisecond());
|
}else{
|
return;
|
}
|
}
|
}
|
//通过注解校验，不需要全部
|
if(!annotateCheck){
|
check(code,request,redisUtil,millisecond);
|
}
|
}
|
|
public static void check(String code,HttpServletRequest request,RedisUtil redisUtil,Long millisecond){
|
//没有设置，默认0.5秒
|
if(millisecond == null){
|
millisecond = 500L;
|
}
|
//获取请求的IP地址
|
String ip = IPUtils.getIpAddr(request);
|
//获取URL
|
String method = request.getServletPath();
|
if(code != null){
|
code = code+"-"+ip+"-"+method;
|
}else{
|
code = ip+"-"+method;
|
}
|
if(!redisUtil.setIfAbsent(code, DateUtil.formatDate(new Date(),"yyyy-MM-dd HH:mm:ss"),millisecond, TimeUnit.MILLISECONDS)){
|
throw new TipsException("请勿频繁操作！");
|
}
|
|
}
|
|
|
}
|
