springboot 防止重复请求防止重复点击的操作

时间:2023-02-03 低端程序狗人气:0

利用 springboot + redis 实现过滤重复提交的请求，业务流程如下所示，首先定义一个拦截器，拦截需要进行过滤的URL，然后用 session + URL 作为唯一 key，利用 redis setnx 命令，来判断请求是否重复，如果 key set 成功，说明非重复请求，失败则说明重复请求；

在这里插入图片描述

URL 拦截器可以使用 spring 拦截器，但使用 spring，每个需要过滤的新 URL 都需要添加配置，因此这里使用 AOP 注解的形式来实现，这样更直观一点；
首先，定义注解：

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.TYPE})
public @interface AvoidRepeatRequest {
    /** 请求间隔时间，单位秒，该时间范围内的请求为重复请求 */
    int intervalTime() default 5;
    /** 返回的提示信息 */
    String msg() default "请不要频繁重复请求！";
}

然后定义 AOP，实现重复请求过滤功能：

@Component
@Aspect
@Order(100)
public class FilterRepeatRequest {
    private static final String SUFFIX = "REQUEST_";
    @Autowired
    private RedisTemplate redisTemplate;

    // 定义 注解 类型的切点
    @Pointcut("@annotation(com.common.ann.AvoidRepeatRequest)")
    public void arrPointcut() {}

    // 实现过滤重复请求功能
    @Around("arrPointcut()")
    public Object arrBusiness(ProceedingJoinPoint joinPoint) {
        // 获取 redis key，由 session ID 和 请求URI 构成
        ServletRequestAttributes sra = (ServletRequestAttributes) 
        	RequestContextHolder.currentRequestAttributes();
        HttpServletRequest request = sra.getRequest();
        String key = SUFFIX + request.getSession().getId() + "_" + request.getRequestURI();

        // 获取方法的 AvoidRepeatRequest 注解
        Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
        AvoidRepeatRequest arr = method.getAnnotation(AvoidRepeatRequest.class);

        // 判断是否是重复的请求
        if (!redisTemplate.opsForValue().setIfAbsent(key, 1, arr.intervalTime(), TimeUnit.SECONDS)) {
            // 已发起过请求
            System.out.println("重复请求");
            return arr.msg();
        }

        try {
            // 非重复请求，执行业务代码
            return joinPoint.proceed();
        } catch (Throwable throwable) {
            throwable.printStackTrace();
            return "error";
        }
    }
}

测试使用：

@RestController
public class TestAop {

    // 使用 AvoidRepeatRequest 注解的请求，表明需要进行重复请求判断
    @AvoidRepeatRequest(intervalTime = 6, msg = "慢点，兄弟")
    @GetMapping("test/aop")
    public String test() {
        return "test aop";
    }
}

加载全部内容