快捷導(dǎo)航

分布式鎖實(shí)例教程之防止重復(fù)提交

更新時(shí)間：2021年11月29日 10:33:15 作者：謎一樣的Coder

訂單重復(fù)問(wèn)題已經(jīng)是老生常談的問(wèn)題了,下面這篇文章主要給大家介紹了關(guān)于分布式鎖實(shí)例教程之防止重復(fù)提交的相關(guān)資料,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下

拋出一個(gè)問(wèn)題

需求：現(xiàn)在有一個(gè)常見(jiàn)的場(chǎng)景——用戶(hù)注冊(cè)，但是如果出現(xiàn)重復(fù)提交的情況，則會(huì)出現(xiàn)多條注冊(cè)數(shù)據(jù)，因此這里如何做好防止重復(fù)提交這是我們需要解決的問(wèn)題。

正常的代碼邏輯

1、注冊(cè)controller

/**
 * 用戶(hù)注冊(cè)請(qǐng)求
 * @param userDto
 * @param bindingResult
 * @return
 */
@RequestMapping(value=prefix+"/db/register",method = RequestMethod.POST,consumes = MediaType.APPLICATION_JSON_UTF8_VALUE)
public BaseResponse register(@RequestBody @Validated UserDto userDto, BindingResult bindingResult){
    BaseResponse response=new BaseResponse(StatusCode.Success);
    try {
        log.debug("注冊(cè)信息： {} ",userDto);
        //注冊(cè)之前，我們先判斷是否已經(jīng)注冊(cè)了。（正常邏輯）
        User user=userService.selectByUserName(userDto.getUserName());
        if (user!=null){
            return new BaseResponse(StatusCode.UserNameExist);
        }
        userService.register(userDto);
    }catch (Exception e){
        e.printStackTrace();
        response=new BaseResponse(StatusCode.Fail);
    }
    return response;
}

在controller中判斷用戶(hù)是否已經(jīng)注冊(cè)，如果沒(méi)有注冊(cè)，則調(diào)用注冊(cè)邏輯。

2、注冊(cè)service

/**
 * 用戶(hù)注冊(cè)——最普通的操作，沒(méi)有任何加鎖，沒(méi)有任何防止重復(fù)提交
 *
 * @param userDto
 * @return
 * @throws Exception
 */
public int register(UserDto userDto) throws Exception {
    int result = 0;
    User user = new User();
    BeanUtils.copyProperties(userDto, user);
    result = userMapper.insertSelective(user);
    return result;
}

簡(jiǎn)單的增加一個(gè)用戶(hù)信息。

問(wèn)題也很明顯，這樣畢竟會(huì)出現(xiàn)問(wèn)題，并發(fā)的問(wèn)題，也會(huì)出現(xiàn)重復(fù)注冊(cè)的情況。測(cè)試結(jié)果也很明顯

一堆重復(fù)注冊(cè)的，但是加入分布式鎖就好了么？

3、加入分布式鎖，問(wèn)題依舊

分布式鎖的實(shí)現(xiàn)方式

/**
 * 用戶(hù)注冊(cè)，基于redisson的分布式鎖
 *
 * @param userDto
 * @return
 */
public int registerLockRedisson(UserDto userDto) {
    int result = 0;
    RLock rLock = redissonLockComponent.acquireLock(userDto.getUserName());
    try {
        if (rLock != null) {
            User user = new User();
            BeanUtils.copyProperties(userDto, user);
            user.setCreateTime(new Date());
            userMapper.insertSelective(user);
        }
    } catch (Exception e) {
        log.error("獲取redisson分布式鎖異常");
    } finally {
        if (rLock != null) {
            redissonLockComponent.releaseLock(rLock);
        }
    }
    return result;
}

加入分布式鎖之后，再進(jìn)行測(cè)試。

不好意思，依舊出現(xiàn)了重復(fù)注冊(cè)的情況。何解？

問(wèn)題分析，為了遵循單一職責(zé)，這里的讀取數(shù)據(jù)（判斷是否注冊(cè)）與寫(xiě)入數(shù)據(jù)（用戶(hù)注冊(cè)）操作是分開(kāi)的，分布式鎖為了進(jìn)一步細(xì)化，只是加在了寫(xiě)入數(shù)據(jù)階段，并沒(méi)有加在整個(gè)業(yè)務(wù)階段，因此會(huì)出現(xiàn)數(shù)據(jù)重復(fù)提交的問(wèn)題，解決方法有很多，最暴力的方法無(wú)非就是給數(shù)據(jù)庫(kù)user表中的用戶(hù)名字段加入唯一約束。但是這樣隨著業(yè)務(wù)規(guī)模擴(kuò)大，數(shù)據(jù)庫(kù)壓力會(huì)越來(lái)越大。

解決方法

解決方法有幾種，前面提到的給數(shù)據(jù)庫(kù)增加唯一索引也是一種方法。但是為了減輕數(shù)據(jù)庫(kù)的壓力，這種操作可以直接在應(yīng)用層處理。

分布式鎖+防重操作

在分布式鎖的基礎(chǔ)上，加入redis存儲(chǔ)key值，作為防重提交的判斷。不想過(guò)多解釋了，直接上代碼吧。

/**
 * 用戶(hù)注冊(cè)，redisson分布式鎖，redis防止重復(fù)提交
 *
 * @param userDto
 * @return
 */
public int registerLockAvoidDupPost(UserDto userDto) {
    int result = 0;
    RLock rLock = redissonLockComponent.acquireLock(userDto.getUserName());
    try {
        //redis中根據(jù)用戶(hù)名存儲(chǔ)作為key值
        String key = lockKeyPrefix+userDto.getUserName();
        if (!stringRedisTemplate.hasKey(key)) {//如果不存在key則進(jìn)入注冊(cè)階段
            stringRedisTemplate.opsForValue().set(key,UUID.randomUUID().toString(),10L,TimeUnit.SECONDS);
            User user = new User();
            BeanUtils.copyProperties(userDto, user);
            user.setCreateTime(new Date());
            userMapper.insertSelective(user);
            log.info("{},注冊(cè)成功",userDto.getUserName());
        }else{//如果存在，則提示不可重復(fù)提交
            log.error("10秒內(nèi)，請(qǐng)勿重復(fù)提交注冊(cè)信息");
        }
    } catch (Exception e) {
        log.error("獲取redisson分布式鎖異常");
    } finally {
        if (rLock != null) {
            redissonLockComponent.releaseLock(rLock);
        }
    }
    return result;
}

分布式鎖的實(shí)現(xiàn)方式有多重，redis/redisson/zookeeper等，只需要在已經(jīng)實(shí)現(xiàn)分布式鎖的基礎(chǔ)上引入防重提交的機(jī)制即可。

因此還有其他方式的實(shí)現(xiàn)，如下所示為zookeeper分布式鎖+redis防重的方式

/**
 * 用戶(hù)注冊(cè)，redisson分布式鎖，redis防止重復(fù)提交
 *
 * @param userDto
 * @return
 */
public int registerLockAvoidDupPost(UserDto userDto) {
    int result = 0;

    InterProcessMutex mutex=new InterProcessMutex(client,zkPrefix+userDto.getUserName()+"-lock");
    try {
        if (mutex.acquire(10L, TimeUnit.SECONDS)){

            final String realKey=zkRedisKeyPrefix+userDto.getUserName();
            if (!stringRedisTemplate.hasKey(realKey)){
                stringRedisTemplate.opsForValue().set(realKey, UUID.randomUUID().toString());

                User user=new User();
                BeanUtils.copyProperties(userDto,user);
                user.setCreateTime(new Date());
                userMapper.insertSelective(user);
				log.info("{},注冊(cè)成功",userDto.getUserName());
            }else{
                log.error("10秒內(nèi)，請(qǐng)勿重復(fù)提交注冊(cè)信息");
            }

        }else{
            throw new RuntimeException("獲取zk分布式鎖失敗!");
        }
    }catch (Exception e){
        e.printStackTrace();
        throw e;
    }finally {
        mutex.release();
    }
    return result;
}

測(cè)試結(jié)果：