防重复提交(分布式解决方案)

     在做项目的时候,点击注册到跳转有1s的间隙,可以多次点击注册按钮,且都能注册成功,所以我们要优化,前后端,都要防重复提交,前端方案大体就是点击后置灰或不可用或隐藏,后端可以在数据库层加锁,因为性能等一些原因效果不理想。所以我们的项目采用注解+AOP+分布式锁实现防重复提交,下面是伪代码:

package com.xxx.xxx.resolver;

import com.xxx.xxx.common.exception.SbcRuntimeException;
import com.xxx.xxx.common.util.CommonErrorCode;
import com.xxx.xxx.redis.RedisService;
import com.xxx.xxx.util.CommonUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * 使用redis分布式锁(setnx)解决
 * hash值 对象实例相同,默认的hash值相等(配合lombok的@Data对hashcode的重写),通过hash值作为锁值校验入参(不要随便重写request hash()方法)
 * hash值超时时间(默认 5 s)
 */
@Aspect
@Component
@Slf4j
public class MultiSubmitHandler {
    /**
     * 锁的默认时间为5秒
     */
    private static final long REPEAT_LOCK_TIME = 5L;

    @Autowired
    private RedisService redisService;

    @Autowired
    private CommonUtil commonUtil;

    @Pointcut("@annotation(com.wanmi.sbc.common.annotation.MultiSubmit)")
    public void pointcut() {
    }

    /**
     * 防止重复提交的请求,请求之前的逻辑
     * 对 用户id + 请求参数的hashcode 加锁
     *
     * @param joinPoint 获取请求Request
     */
    @Before("pointcut()")
    public void before(JoinPoint joinPoint) {
        long start = System.currentTimeMillis();
        Object[] objects = joinPoint.getArgs();
        if (objects.length == 1) {
            String key = getRedisKey(objects);
            if (redisService.setNx(key, "1", REPEAT_LOCK_TIME)) {
                log.info("submitting repeat check time : " + (System.currentTimeMillis() - start) + "ms, key:" + key);
            } else {
                log.error("submitting repeat: " + joinPoint.toLongString());
                throw new SbcRuntimeException(CommonErrorCode.REPEAT_REQUEST);
            }
        }
    }

    /**
     * 防止重复提交的请求,请求之后的逻辑
     * 释放锁
     *
     * @param joinPoint 获取请求Request
     * @param res       暂无用
     */
    @AfterReturning(pointcut = "pointcut()", returning = "res")
    public void afterReturning(JoinPoint joinPoint, Object res) {
        Object[] objects = joinPoint.getArgs();
        if (objects.length == 1) {
            String key = getRedisKey(objects);
            redisService.delete(key);
            log.info("submitting repeat lock released, key:" + key);
        }
    }

    /**
     * 获取防止相同请求重复提交的redis锁的key
     *
     * @param objects 请求参数
     * @return key redis锁的key
     */
    private String getRedisKey(Object[] objects) {
        String key = String.valueOf(objects[0].hashCode());
        String uid;
        try {
            uid = commonUtil.getOperatorId();
        } catch (Exception e) {
            uid = null;
        }
        if (StringUtils.isNotBlank(uid)) {
            key = "R:" + uid + ":" + key;
        } else {
            key = "R:" + key;
        }
        return key;
    }

}
package com.xxx.xxx.common.annotation;

import java.lang.annotation.*;

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

用的时候,在controller上加MultiSubmit注释就可以。此方案也不是最优,大部分场景可以满足。欢迎在评论区补充不足之处。共同学习,共同进步。

  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值