快捷導(dǎo)航

Redis對批量數(shù)據(jù)實(shí)現(xiàn)分布式鎖的實(shí)現(xiàn)代碼

更新時(shí)間：2022年03月24日 09:46:10 作者：HCAN

為了防止多人多電腦同時(shí)操作一條數(shù)據(jù)，我們自己開發(fā)了一個(gè)簡單的基于Redis實(shí)現(xiàn)的分布式鎖，Redis對批量數(shù)據(jù)實(shí)現(xiàn)分布式鎖相關(guān)知識感興趣的朋友一起看看吧

需求背景

在開發(fā)的收入結(jié)轉(zhuǎn)平臺界面上有一個(gè)歸集按鈕，可以實(shí)現(xiàn)抓取結(jié)轉(zhuǎn)表里面的多條數(shù)據(jù)進(jìn)行歸集操作。為了防止多人多電腦同時(shí)操作一條數(shù)據(jù)，我們自己開發(fā)了一個(gè)簡單的基于Redis實(shí)現(xiàn)的分布式鎖。

代碼實(shí)現(xiàn)

邏輯代碼中的使用案例

參數(shù)說明：

scIds：結(jié)轉(zhuǎn)數(shù)據(jù)的ID主鍵集合。

timeOutToDeleteRedisKey：最大鎖超時(shí)時(shí)間（用于自動解鎖）

organizationId：租戶ID（這個(gè)參數(shù)根據(jù)情況選擇是否需要）

ReturnLock returnLock = RedisLock.applyByIds(scIds, redisLockTime, organizationId, () -> {
    // dothing 具體的代碼業(yè)務(wù)邏輯
    return 123;
});
 
if (!returnLock.getFlag()) {
    if(returnLock.getLock()){
        throw new CommonException("歸集數(shù)據(jù)有程序正在運(yùn)行，請稍候刷新頁面重試");
    }else {
        throw new CommonException(returnLock.getErrorMsg());
    }
}
// 返回對象
System.out.println(returnLock.getResObj()) // 123

Redis加鎖方法封裝

public static ReturnLock applyByIds(List<Long> scIds, Long timeOutToDeleteRedisKey, Long tenantId,
                                    Supplier<Object> supplier) {
    // 獲得鎖
    Map<String, String> keyMap = getLockByIds(scIds, timeOutToDeleteRedisKey, tenantId);
    ReturnLock returnLock = new ReturnLock();
    returnLock.setFlag(true);
    returnLock.setLock(false);
    try {
        // 判斷主鍵ID數(shù)量和加鎖的數(shù)量是否一致，不一致說明有加鎖失敗的數(shù)據(jù)，返回失敗鎖信息
        if(scIds.size() > keyMap.size()){
            returnLock.setFlag(false);
            returnLock.setLock(true);
            returnLock.setKeyMap(keyMap);
            return returnLock;
        }
        // 應(yīng)用代碼執(zhí)行后的返回結(jié)果 supplier：java8四大內(nèi)置函數(shù)的供給型接口
        // Supplier<T>（供給型接口）無參數(shù)，返回類型為T的對象：T get()
        returnLock.setResObj(supplier.get());
    }catch (Exception e) {
        returnLock.setFlag(false);
        returnLock.setErrorMsg(e.getMessage());
        e.printStackTrace();
    }finally {
        // 應(yīng)用代碼執(zhí)行報(bào)錯(cuò)，解鎖
        unLockByIds(keyMap);
    }
    return returnLock;
}

getLockByIds

批量獲取每一條數(shù)據(jù)的Redis鎖

private static Map<String, String> getLockByIds(List<Long> scIds, Long timeOutToDeleteRedisKey, Long tenantId) {
    Map<String, String> keyMap = new HashMap<>();
    // 從spring上下文中獲取Redis的操作對象，因?yàn)檫@個(gè)代碼是寫在util中，所以通過上下文方式獲取bean對象
    // RedisHelper：Redis的操作對象，我們自己公司基于redisTemplate封裝的
    RedisHelper redisHelper = SpringUtil.getBean(RedisHelper.class);
    try {
        if (CollectionUtil.isNotEmpty(scIds)) {
            for (int i = 0; i < scIds.size(); i++) {
                String item = "L_" + scIds.get(i);
                String UUID = UUIDUtils.generateTenantUUID(tenantId);
                // 判斷key值是否被鎖，如果沒有鎖則加鎖并設(shè)置過期時(shí)間, 如果有鎖, 則報(bào)錯(cuò)并立即釋放已加的鎖
                // strSetIfAbsent會返回true/false,底層是封裝了java的setIfAbsent方法和Redis的setnx方法
                // 如果設(shè)置成功返回true否則false
                if (redisHelper.strSetIfAbsent(item, UUID)) {
                    // 設(shè)置過期時(shí)間
                    redisHelper.setExpire(item, timeOutToDeleteRedisKey, TimeUnit.SECONDS);
                    // 保存設(shè)置的鎖信息
                    keyMap.put(item, UUID);
                } else {
                    // 鎖設(shè)置異常，表示有數(shù)據(jù)被鎖住了，解鎖之間加鎖的數(shù)據(jù)
                    if (MapUtil.isNotEmpty(keyMap)) {
                        if (unLockByIds(keyMap)) {
                            // 解鎖失敗再次解鎖
                            unLockByIds(keyMap);
                        }
                    }
                    // 返回鎖信息
                    return keyMap;
                }
            }
        }
        return keyMap;
    }catch (Exception e) {
        e.printStackTrace();
    }
}

解鎖

private static Boolean unLockByIds(Map<String, String> keyMap) {
    RedisHelper redisHelper = SpringUtil.getBean(RedisHelper.class);
    try {
        if (MapUtil.isEmpty(keyMap)) {
            return false;
        }
        // 判斷是否是自己的鎖
        for (String key : keyMap.keySet()) {
            String uuid = keyMap.get(key);
            if (StringUtils.equals(uuid, redisHelper.strGet(key))) {
                // 封裝了redisTemplate.delete(key);
                redisHelper.delKey(key);
            }
        return true;
    }catch (Exception e) {
        e.printStackTrace();
        return false;
    }
}