SpringBoot实现表单重复提交检测

code2roc

发布于 2023-07-19 14:55:44

2890

发布于 2023-07-19 14:55:44

文章被收录于专栏：IT民工的代码世界IT民工的代码世界

前言

在实际开发过程中，web应用经常会出现网络延迟，接口处理时间略长，用户习惯等原因造成的客户连续多次点击提交按钮调用接口，导致数据库会出现重复数据或这接口业务逻辑bug等问题

方案

利用redis锁实同一个用户同一个请求2秒内重复提交返回错误路由

SubmitLock

标记需要拦截的方法

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface SubmitLock {
    int expire() default 2;
}

RedisLockUtil

redis锁校验及写入

@Component
public class RedisLockUtil {
    @Autowired
    private RedisUtil redisUtil;

    private int lockDBIndex = 1;

    public boolean lock(String key,String clientID,int lockExpire){
       if(redisUtil.isValid(key,lockDBIndex)){
            return false;
       }else{
           redisUtil.redisTemplateSet(key,clientID,lockDBIndex);
           redisUtil.setExpire(key,lockExpire, TimeUnit.SECONDS,lockDBIndex);
           return true;
       }
    }
}

RepeatSubmitAspect

统一拦截切面

@Aspect
@Component
@Order(value = 100)
public class RepeatSubmitAspect {
    private static Logger logger = LoggerFactory.getLogger(RepeatSubmitAspect.class);
    @Autowired
    private RedisLockUtil redisLockUtil;

    /**
     * 切面点 指定注解
     */
    @Pointcut("@annotation(com.haopan.frame.common.annotation.SubmitLock) " +
            "|| @within(com.haopan.frame.common.annotation.SubmitLock)")
    public void repeatSubmitAspect() {

    }

    /**
     * 拦截方法指定为 repeatSubmitAspect
     */
    @Around("repeatSubmitAspect()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        SubmitLock submitLock = method.getAnnotation(SubmitLock.class);
        if (submitLock != null) {
            ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
            HttpServletRequest request = requestAttributes.getRequest();
            String token = request.getHeader("token");
            if (!StringUtil.isEmpty(token)) {
                String path = request.getServletPath();
                String key = "submitLock|" + token + "|" + path;
                String clientId = CommonUtil.getNewGuid();
                if (redisLockUtil.lock(key, clientId, submitLock.expire())) {
                    // 获取锁成功
                    return point.proceed();
                } else {
                    System.out.println("tryLock fail, key = ["+key+"]");
                    return Result.errorResult().setMsg("重复请求，请稍后再试").setCode(-980);
                }
            } else {
                return point.proceed();
            }
        } else {
            return point.proceed();
        }
    }
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021-08-29，如有侵权请联系 cloudcommunity@tencent.com 删除

表单