//自定义一个防止重复提交的注解
package com.mingwen.common.SubmitMore;
import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* @description 防止表单重复提交注解
*/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
@Documented
public @interface DuplicateSubmitToken {
// 保存重复提交标记 默认为需要保存
boolean save() default true;
}
//写一个防止重复提交的拦截器
package com.mingwen.common.SubmitMore;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import com.mingwen.common.exception.DuplicateSubmitException;
import cn.hutool.core.lang.UUID;
import lombok.extern.slf4j.Slf4j;
/**
* @description 防止表单重复提交拦截器
*/
@Aspect
@Component
@Slf4j
public class DuplicateSubmitAspect {
public static final String DUPLICATE_TOKEN_KEY = "duplicate_token_key";
//需要扫描的controller地址
@Pointcut("execution(public * com.mingwen.*.controller..*(..))")
public void webLog() {
}
@Before("webLog() && @annotation(token)")
public void before(final JoinPoint joinPoint, DuplicateSubmitToken token) throws DuplicateSubmitException {
if (token != null) {
Object[] args = joinPoint.getArgs();
HttpServletRequest request = null;
HttpServletResponse response = null;
for (int i = 0; i < args.length; i++) {
if (args[i] instanceof HttpServletRequest) {
request = (HttpServletRequest) args[i];
}
if (args[i] instanceof HttpServletResponse) {
response = (HttpServletResponse) args[i];
}
}
boolean isSaveSession = token.save();
if (isSaveSession) {
Object t = request.getSession().getAttribute(DUPLICATE_TOKEN_KEY);
if (null == t) {
String uuid = UUID.randomUUID().toString();
request.getSession().setAttribute(DUPLICATE_TOKEN_KEY, uuid);
// log.info("进入方法:token=" + uuid);
} else {
throw new DuplicateSubmitException("请不要重复请求!");
}
}
}
}
@AfterReturning("webLog() && @annotation(token)")
public void doAfterReturning(JoinPoint joinPoint, DuplicateSubmitToken token) {
// 处理完请求,返回内容
// log.info("出来方法:");
if (token != null) {
Object[] args = joinPoint.getArgs();
HttpServletRequest request = null;
for (int i = 0; i < args.length; i++) {
if (args[i] instanceof HttpServletRequest) {
request = (HttpServletRequest) args[i];
}
}
boolean isSaveSession = token.save();
if (isSaveSession) {
Object t = request.getSession().getAttribute(DUPLICATE_TOKEN_KEY);
if (null != t) {
// 方法执行完毕移除请求重复标记
request.getSession(false).removeAttribute(DUPLICATE_TOKEN_KEY);
// log.info("方法执行完毕移除请求重复标记!");
}
}
}
}
}
//自定义一个表单重复提交异常
/**
* @author wsj 自定义异常
*/
public class DuplicateSubmitException extends Exception {
public DuplicateSubmitException(String msg) {
super(msg);
}
}
//遇到重复提交的异常要将信息返回到前端显示 拦截器抛的异常在controller是无法捕获到的 所以这里统一异常捕获类 控制器增强
@ControllerAdvice
@ResponseBody
@Slf4j
public class ExceptionCatch {
protected Map getResultObjectData(ResultCode result, Object pd, String msg) {
Map resultData = new HashMap<>();
resultData.put("code", result.getCode());
resultData.put("message", result.getMsg());
resultData.put("data", pd);
return resultData;
}
// 捕获CustomException此类异常
@ExceptionHandler(DuplicateSubmitException.class)
public Map customException(DuplicateSubmitException bmException) {
bmException.printStackTrace();
// 记录日志
log.error("catch exception:{}", "不可重复提交");
return getResultObjectData(ResultCode.NO_SUBMIT, "", "不可重复提交");
}
}
//测试controller
/**
* 防止重复提交测试
*
* @param request
* @return
*/
@DuplicateSubmitToken
@RequestMapping("/test")
public Map test(HttpServletRequest request) {
try {
//休眠的目的是让你自己在前端能有时间重复点击一下接口调用
Thread.sleep(5000);
} catch (InterruptedException e) {
e.printStackTrace();
}
Map map = new HashMap<>();
//往request 里面存入每次请求的接口地址 这样控制器能能拿到request里面的请求地址
request.getSession().setAttribute("request Url", request.getRequestURL());
map.put("request Url", request.getRequestURL());
return map;
}
//效果
接口快速点击两次