chenjiahe
2023-12-05 b474027a929c004818e5274aee79bf88def355a4
防重复提交工具
1个文件已删除
2个文件已添加
130 ■■■■ 已修改文件
src/main/java/com/hx/common/annotations/repeat/RequestRepeat.java 17 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/main/java/com/hx/repeat/check/RequestRepeat.java 37 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/main/java/com/hx/repeat/check/RequestRepeatUtil.java 76 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/main/java/com/hx/common/annotations/repeat/RequestRepeat.java
New file
@@ -0,0 +1,17 @@
package com.hx.common.annotations.repeat;
import java.lang.annotation.*;
/**
 * 重复校验注解
 * @author CJH
 */
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RequestRepeat {
    /**是否重复校验,默认true*/
    boolean isRepeat() default true;
    /**重复间隔时间,毫秒,默认1000毫秒*/
    long millisecond() default 1000;
}
src/main/java/com/hx/repeat/check/RequestRepeat.java
File was deleted
src/main/java/com/hx/repeat/check/RequestRepeatUtil.java
New file
@@ -0,0 +1,76 @@
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("请勿频繁操作!");
        }
    }
}