Redisson可重入鎖解鎖邏輯詳細講解

更新時間：2023年02月11日 13:57:49 作者：每天都要進步一點點

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

主動釋放

我們查看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) -> {
        // 取消看門狗定時任務
        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)取消看門狗自動續(xù)期的定時任務。

釋放鎖核心邏輯：

protected RFuture<Boolean> unlockInnerAsync(long threadId) {
    /**
     * Redisson解鎖：通過LUA腳本釋放鎖，保證多個命令之間的原子性
     */
    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可重入鎖的釋放還是通過LUA腳本實現(xiàn)，保證多個命令之間的原子性。

基本流程：

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

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

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

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

5、其它情況，返回nil；

主動釋放鎖這塊考慮的不僅僅是對 key 進行處理，因為可能存在重入鎖，所以會先對 redis key 對應的 hash value 進行遞減，相當于減去重入次數(shù)。