java 重复请求接口过滤

1. 自定义注解

@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface UrlRequest {

    //是否开启
    boolean opened() default false;

    //限制时间
    long time() default 200L;
}

2. 创建 UrlDataInterceptor  类 继承 HandlerInterceptorAdapter类

/**
     * 是否阻止提交
     *
     * @return
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
// 校验是否存在token 不存在则直接放行 Authorization => 获取请求头是否有token 
        String tokenStr = request.getHeader("Authorization");
        if (StringUtils.isBlank(tokenStr)) {
            return true;
        }
//截取token Bearer=> 
        tokenStr = tokenStr.replaceAll(SystemConstant.JWT_TOKEN_PREFIX, "");
        if (StringUtils.isNoneBlank(tokenStr)) {
            if (handler instanceof HandlerMethod) {
                HandlerMethod handlerMethod = (HandlerMethod) handler;
                Method method = handlerMethod.getMethod();
                RepeatRequest su = method.getAnnotation(RepeatRequest.class);
                if (su != null && su.opened()) {
                    if (validateRepeatRequest(request, su.time(), tokenStr)) {
                        JSONObject result = new JSONObject();
                        result.put("statusCode", "500");
                        result.put("message", "请勿重复发起请求");
                        response.setCharacterEncoding("UTF-8");
                        response.setContentType("application/json; charset=utf-8");
                        response.getWriter().write(result.toString());
                        response.getWriter().close();
                        return false;
                    } else {
                        return true;
                    }
                }
            }
        }
        return true;
    }

    public boolean validateRepeatRequest(HttpServletRequest request, long time, String token) {
// 具体的业务逻辑判断 这
        String remoteAddr = StringUtils.getRemoteAddr(request);
        String url = request.getRequestURI();
        Map<String, String> map = new HashMap<>(1);
        map.put(url, remoteAddr);
        String nowUrlParams = map.toString();
        String requestRepeatKey = remoteAddr.concat(url).concat("-").concat(token);
// 将请求数据缓存到redis中 , SystemConstant.REPEAT_REQUEST => 配置常量类 可自定义
        RMapCache<String, String> requestMap = redissonService.getRMapCache(SystemConstant.REPEAT_REQUEST);
        String msg = requestMap.get(nowUrlParams);
        if (ObjectUtils.isEmpty(msg)) {
            // 当前key 不存在则允许请求
            time = ObjectUtils.isEmpty(time) ? 200 : time;
            //默认200毫秒
            requestMap.put(nowUrlParams, requestRepeatKey, time, TimeUnit.MILLISECONDS);
        } else {
            // 当前key存在 判断值是否一致
            if (msg.equals(requestRepeatKey)) {
                return true;
            }
        }
        return false;
    }

3. 创建WebMvcConfigExt 拦截器

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

@Configuration
public class WebcConfigExt implements WebMvcConfigurer {

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(urlDataInterceptor()).addPathPatterns("/**");
    }

    @Bean
    public UrlDataInterceptor urlDataInterceptor() {
        return new UrlDataInterceptor ();
    }

}

4. 测试请求

@PostMapping("/test")
@ApiOperation(value = "测试")
@UrlRequest(opened = true, time = 2000L) // opened = ture 开启,flash 关闭 ; time 时间 单位 毫秒
public void test() {
    System.out.println("test");
}