快捷導(dǎo)航

Redisson可重入鎖解鎖邏輯詳細(xì)講解

更新時(shí)間：2023年02月11日 13:57:49 作者：每天都要進(jìn)步一點(diǎn)點(diǎn)

Redisson開(kāi)源框架是一個(gè)Redis的分布式鎖的現(xiàn)成實(shí)現(xiàn)方案，是Redis的java實(shí)現(xiàn)的客戶端。通過(guò)Netty支持非阻塞I/O。Redisson實(shí)現(xiàn)了分布式鎖的自動(dòng)續(xù)期機(jī)制、鎖的互斥自等待機(jī)制、鎖的可重入加鎖與釋放鎖的機(jī)制

主動(dòng)釋放

我們查看org.redisson.RedissonLock#unlock()方法：

public void unlock() {
        try {
            get(unlockAsync(Thread.currentThread().getId()));
        } catch (RedisException e) {
            if (e.getCause() instanceof IllegalMonitorStateException) {
                throw (IllegalMonitorStateException) e.getCause();
            } else {
                throw e;
            }
        }
//        Future<Void> future = unlockAsync();
//        future.awaitUninterruptibly();
//        if (future.isSuccess()) {
//            return;
//        }
//        if (future.cause() instanceof IllegalMonitorStateException) {
//            throw (IllegalMonitorStateException)future.cause();
//        }
//        throw commandExecutor.convertException(future);
    }

同樣采用異步的方式釋放鎖，unlockAsync()：

public RFuture<Void> unlockAsync(long threadId) {
    // 異步方式釋放鎖
    RFuture<Boolean> future = unlockInnerAsync(threadId);
    CompletionStage<Void> f = future.handle((opStatus, e) -> {
        // 取消看門(mén)狗定時(shí)任務(wù)
        cancelExpirationRenewal(threadId);
        if (e != null) {
            throw new CompletionException(e);
        }
        if (opStatus == null) {
            IllegalMonitorStateException cause = new IllegalMonitorStateException("attempt to unlock lock, not locked by current thread by node id: "
                    + id + " thread-id: " + threadId);
            throw new CompletionException(cause);
        }
        return null;
    });
    return new CompletableFutureWrapper<>(f);
}

可以看到，首先調(diào)用unlockInnerAsync()方法釋放鎖，在釋放鎖之后，執(zhí)行cancelExpirationRenewal(threadId)取消看門(mén)狗自動(dòng)續(xù)期的定時(shí)任務(wù)。

釋放鎖核心邏輯：

protected RFuture<Boolean> unlockInnerAsync(long threadId) {
    /**
     * Redisson解鎖：通過(guò)LUA腳本釋放鎖，保證多個(gè)命令之間的原子性
     */
    return evalWriteAsync(getRawName(), LongCodec.INSTANCE, RedisCommands.EVAL_BOOLEAN,
            "if (redis.call('hexists', KEYS[1], ARGV[3]) == 0) then " +
                    "return nil;" +
                    "end; " +
                    "local counter = redis.call('hincrby', KEYS[1], ARGV[3], -1); " +
                    "if (counter > 0) then " +
                    "redis.call('pexpire', KEYS[1], ARGV[2]); " +
                    "return 0; " +
                    "else " +
                    "redis.call('del', KEYS[1]); " +
                    "redis.call('publish', KEYS[2], ARGV[1]); " +
                    "return 1; " +
                    "end; " +
                    "return nil;",
            Arrays.asList(getRawName(), getChannelName()), LockPubSub.UNLOCK_MESSAGE, internalLockLeaseTime, getLockName(threadId));
}

Redisson可重入鎖的釋放還是通過(guò)LUA腳本實(shí)現(xiàn)，保證多個(gè)命令之間的原子性。

基本流程：

1、通過(guò)hexists指令判斷鎖是不是自己的，如果不是自己的鎖，說(shuō)明非法釋放鎖，返回nil，

2、如果是自己的鎖，通過(guò)hincrby將鎖的重入次數(shù)減1；

3、判斷減1后的數(shù)是否大于0，如果減1后的數(shù)大于0，說(shuō)明還沒(méi)有完全釋放鎖，則重置鎖的過(guò)期時(shí)間，并返回0；

4、如果減1后的數(shù)已經(jīng)等于0，說(shuō)明已經(jīng)完全釋放鎖，則通過(guò)del指令釋放鎖，并通過(guò)publish發(fā)布一條消息，告訴其它訂閱了這把鎖的線程，我已經(jīng)釋放鎖了，你們可以過(guò)來(lái)獲取了；釋放鎖成功，返回1

5、其它情況，返回nil；

主動(dòng)釋放鎖這塊考慮的不僅僅是對(duì) key 進(jìn)行處理，因?yàn)榭赡艽嬖谥厝腈i，所以會(huì)先對(duì) redis key 對(duì)應(yīng)的 hash value 進(jìn)行遞減，相當(dāng)于減去重入次數(shù)。