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

     */

    @Override

    public 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();

    }

}

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 {

    @Autowired

    private xxx.RedisUtils redisUtils;

    @Override

    public void addInterceptors(InterceptorRegistry registry) {

        registry.addInterceptor(validateInterceptorHandler()).addPathPatterns("/**");

    }

    @Bean

    public com.bqmart.interceptor.ResubmitInterceptorHandler resubmitInterceptorHandler() {

        return new ResubmitInterceptorHandler(redisUtils);

    }

}