快捷導(dǎo)航

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

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

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

拋出一個問題

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

正常的代碼邏輯

1、注冊controller

/**
 * 用戶注冊請求
 * @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("注冊信息： {} ",userDto);
        //注冊之前，我們先判斷是否已經(jīng)注冊了。（正常邏輯）
        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中判斷用戶是否已經(jīng)注冊，如果沒有注冊，則調(diào)用注冊邏輯。

2、注冊service

/**
 * 用戶注冊——最普通的操作，沒有任何加鎖，沒有任何防止重復(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;
}

簡單的增加一個用戶信息。

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

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

3、加入分布式鎖，問題依舊

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

/**
 * 用戶注冊，基于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;
}

加入分布式鎖之后，再進行測試。

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

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

解決方法

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

分布式鎖+防重操作

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

/**
 * 用戶注冊，redisson分布式鎖，redis防止重復(fù)提交
 *
 * @param userDto
 * @return
 */
public int registerLockAvoidDupPost(UserDto userDto) {
    int result = 0;
    RLock rLock = redissonLockComponent.acquireLock(userDto.getUserName());
    try {
        //redis中根據(jù)用戶名存儲作為key值
        String key = lockKeyPrefix+userDto.getUserName();
        if (!stringRedisTemplate.hasKey(key)) {//如果不存在key則進入注冊階段
            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("{},注冊成功",userDto.getUserName());
        }else{//如果存在，則提示不可重復(fù)提交
            log.error("10秒內(nèi)，請勿重復(fù)提交注冊信息");
        }
    } catch (Exception e) {
        log.error("獲取redisson分布式鎖異常");
    } finally {
        if (rLock != null) {
            redissonLockComponent.releaseLock(rLock);
        }
    }
    return result;
}

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

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

/**
 * 用戶注冊，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("{},注冊成功",userDto.getUserName());
            }else{
                log.error("10秒內(nèi)，請勿重復(fù)提交注冊信息");
            }

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

測試結(jié)果：