亲宝软件园·资讯

展开

Redisson可重入锁解锁逻辑详细讲解

每天都要进步一点点 人气:0

本篇文章基于redisson-3.17.6版本源码进行分析

相比较Redisson可重入锁的加锁逻辑,释放锁的逻辑就相对简单一些。释放锁分为主动释放和自动释放两种方式。

主动释放

我们查看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);
}

可以看到,首先调用unlockInnerAsync()方法释放锁,在释放锁之后,执行cancelExpirationRenewal(threadId)取消看门狗自动续期的定时任务。

释放锁核心逻辑:

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脚本实现,保证多个命令之间的原子性。

基本流程:

1、通过hexists指令判断锁是不是自己的,如果不是自己的锁,说明非法释放锁,返回nil,

2、如果是自己的锁,通过hincrby将锁的重入次数减1;

3、判断减1后的数是否大于0,如果减1后的数大于0,说明还没有完全释放锁,则重置锁的过期时间,并返回0;

4、如果减1后的数已经等于0,说明已经完全释放锁,则通过del指令释放锁,并通过publish发布一条消息,告诉其它订阅了这把锁的线程,我已经释放锁了,你们可以过来获取了;释放锁成功,返回1

5、其它情况,返回nil;

主动释放锁这块考虑的不仅仅是对 key 进行处理,因为可能存在重入锁,所以会先对 redis key 对应的 hash value 进行递减,相当于减去重入次数。

自动释放

当服务宕机时,看门狗不再看门,那么最多 30s 之后锁被自动释放;当设置锁的超时时间时,锁到了过期时间,自动释放;

加载全部内容

相关教程
猜你喜欢
用户评论