通过aop 注解的方式防止表单重复提交

pom.xml

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

annotation

import java.lang.annotation.*;

@Inherited
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface PreventRepeatSubmit {

    public int interval() default 40000;

    public String message() default "不允许重复提交，请稍后再试";
}

aop

import com.alibaba.fastjson.JSON;
import com.zan.rwj.common.exception.BusinessException;
import com.zan.rwj.common.result.ResponseEnum;
import com.zan.rwj.system.annotation.PreventRepeatSubmit;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.concurrent.TimeUnit;

@Component
@Aspect
public class PreventRepeatSubmitAspect {

    @Value("{token.header}")
    private String header;

    @Resource
    private RedisTemplate redisTemplate;

    @Pointcut("@annotation(com.zan.rwj.system.annotation.PreventRepeatSubmit)")
    public void preventRepeatSubmit(){

    }
    @Around("preventRepeatSubmit()")
    public Object checkPrs(ProceedingJoinPoint pjp) throws Throwable {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String requestURI = request.getRequestURI();
        Object[] args = pjp.getArgs();
        String argStr = JSON.toJSONString(args);
        argStr = argStr.replace(":","#");
//        String submitKey = request.getHeader(header).trim();
//        String cacheRepeatKey = "repeat_submit:" +requestURI+":"+argStr+":"+submitKey;
        String cacheRepeatKey = "repeat_submit:" +requestURI+":"+argStr;
        MethodSignature ms = (MethodSignature) pjp.getSignature();
        Method method = ms.getMethod();
        PreventRepeatSubmit preventRepeatSubmit = method.getAnnotation(PreventRepeatSubmit.class);
        int interval = preventRepeatSubmit.interval();
        Boolean aBoolean = redisTemplate.opsForValue().setIfAbsent(cacheRepeatKey, 1, preventRepeatSubmit.interval(), TimeUnit.SECONDS);
        if(!aBoolean){
            //JSON.toJSONString(ResponseResult.errorResult(HttpCodeEnum.SYSTEM_ERROR.getCode(),annotation.message())));
            throw new BusinessException(ResponseEnum.ALIYUN_SMS_ERROR);
        }
        return pjp.proceed();

    }
}