springmvc防止重復提交攔截器

一、攔截器實現，ResubmitInterceptorHandler.java

import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Map;
import java.util.Set;/*** 重復請求阻止攔截器*/
@Component("resubmitInterceptorHandler")
public class ResubmitInterceptorHandler extends HandlerInterceptorAdapter {private RedisUtils redisUtils;//自定義public ResubmitInterceptorHandler(RedisUtils redisUtils) {this.redisUtils = redisUtils;}/*** 攔截重復提交的請求** @param request* @param response* @param handler* @return* @throws Exception*/@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {HandlerMethod method = (HandlerMethod) handler;Resubmit resubmit = method.getMethodAnnotation(Resubmit.class);if (resubmit == null) {return true;} else {Long seconds = resubmit.seconds();// 獲取重復提交的鍵值String key = getKey(request, method);String value = redisUtils.get(key, String.class);if (StringUtils.isBlank(value)) {// 如果存在就存儲到redis中
                redisUtils.set(key, seconds.toString(), seconds);return true;} else {throw new Exception("請不要在" + seconds + "秒內重復請求");}}}/*** 獲取redis存儲的鍵** @param request* @param method* @return*/private String getKey(HttpServletRequest request, HandlerMethod method) {StringBuffer sb = new StringBuffer();String requestURI = request.getRequestURI();// 拼接請求路徑
        sb.append(requestURI);Method targetMethod = method.getMethod();// 拼接目標方法名稱
        sb.append(targetMethod.getName());Map<String, String[]> parameterMap = request.getParameterMap();if (parameterMap != null) {Set<Map.Entry<String, String[]>> entries = parameterMap.entrySet();if (entries != null) {for (Map.Entry<String, String[]> entry : entries) {sb.append(entry.getKey()).append(Arrays.toString(entry.getValue()));}}}return sb.toString();}
}

二、controller上要添加的注解

import java.lang.annotation.*;@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Resubmit {long value() default 0;/*** 指定多少時間以內不能重復提交* -1 表示不進行處理** @return*/long seconds();
}

三、攔截器配置

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;@Configuration
@Import(com.bqmart.utils.RedisUtils.class)
public class InterceptorConfig extends WebMvcConfigurerAdapter {@Autowiredprivate xxx.RedisUtils redisUtils;@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(validateInterceptorHandler()).addPathPatterns("/**");}@Beanpublic com.bqmart.interceptor.ResubmitInterceptorHandler resubmitInterceptorHandler() {return new ResubmitInterceptorHandler(redisUtils);}}

?