6. Redisson源码剖析-公平锁按顺序依次加锁

一、说明

  1. 笔记在一点点的更新,对于公平锁的加锁,其实Lua脚本都是同一套,我们在分析源码的时候,都是假设当前时间,其实这个也很重要的,假设的条件也是根据实际情况。
  2. 如果客户端A释放了锁,删除了锁key之后,客户端B和客户端C是如何按照顺序依次加锁的。假设客户端B在队列里面是排在后面的,假设锁被释放掉了之后,是客户端B先来尝试加锁
  3. 假设当前时间是10:00:40,锁已经被释放了,客户端B来尝试重新加锁,只会刷新客户端B在有序集合中的分数,获取锁是失败的
  4. 那么如果B获取失败了,C排队队头,那么此时,如果客户端C来尝试获取锁的情况呢?源码中的2代表客户端C获取锁,1代表客户端B获取锁

二、源码中的参数

  1. 单独取出源码的参数,方便阅读源码
KEYS = Arrays.<Object>asList(getName(), threadsQueueName, timeoutSetName)
KEYS[1] = getName() = 锁的名字,“anyLock”
KEYS[2] = threadsQueueName = redisson_lock_queue:{anyLock},基于redis的数据结构实现的一个队列
KEYS[3] = timeoutSetName = redisson_lock_timeout:{anyLock},基于redis的数据结构实现的一个Set数据集合,有序集合,可以自动按照你给每个数据指定的一个分数(score)来进行排序

ARGV =  internalLockLeaseTime, getLockName(threadId), currentTime + threadWaitTime, currentTime
ARGV[1] = 30000毫秒
ARGV[2] = UUID:threadId
ARGV[3] = 当前时间(10:00:40) + 5000毫秒 = 10:00:05
ARGV[4] = 当前时间(10:00:40)(这里的时间和之前笔记中的时间已经向后移动了)

三、源码

RedissonFairLock类中

@Override
<T> RFuture<T> tryLockInnerAsync(long leaseTime, TimeUnit unit, long threadId, RedisStrictCommand<T> command) {
    internalLockLeaseTime = unit.toMillis(leaseTime);

    //********************BEGIN-command为EVAL_LONG,所以不走这段逻辑*************************
    long currentTime = System.currentTimeMillis();
    if (command == RedisCommands.EVAL_NULL_BOOLEAN) {
        return commandExecutor.evalWriteAsync(getName(), LongCodec.INSTANCE, command,
                // remove stale threads
                "while true do "
                + "local firstThreadId2 = redis.call('lindex', KEYS[2], 0);"
                + "if firstThreadId2 == false then "
                    + "break;"
                + "end; "
                + "local timeout = tonumber(redis.call('zscore', KEYS[3], firstThreadId2));"
                + "if timeout <= tonumber(ARGV[3]) then "
                    + "redis.call('zrem', KEYS[3], firstThreadId2); "
                    + "redis.call('lpop', KEYS[2]); "
                + "else "
                    + "break;"
                + "end; "
              + "end;"
                + 
                
                "if (redis.call('exists', KEYS[1]) == 0) and ((redis.call('exists', KEYS[2]) == 0) "
                        + "or (redis.call('lindex', KEYS[2], 0) == ARGV[2])) then " +
                        "redis.call('lpop', KEYS[2]); " +
                        "redis.call('zrem', KEYS[3], ARGV[2]); " +
                        "redis.call('hset', KEYS[1], ARGV[2], 1); " +
                        "redis.call('pexpire', KEYS[1], ARGV[1]); " +
                        "return nil; " +
                    "end; " +
                    "if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then " +
                        "redis.call('hincrby', KEYS[1], ARGV[2], 1); " +
                        "redis.call('pexpire', KEYS[1], ARGV[1]); " +
                        "return nil; " +
                    "end; " +
                    "return 1;", 
                Arrays.<Object>asList(getName(), threadsQueueName, timeoutSetName), 
                internalLockLeaseTime, getLockName(threadId), currentTime);
    }
    ////********************END-command为EVAL_LONG,所以不走这段逻辑*************************
    
    if (command == RedisCommands.EVAL_LONG) {
        return commandExecutor.evalWriteAsync(getName(), LongCodec.INSTANCE, command,
                // remove stale threads
                // 1.客户端B进入到死循环逻辑
                "while true do “
                // 1.lindex redisson_lock_queue:{anyLock} 0 取出队列中的第一个元素,
                // 1. 此时队列的第一个元素是客户端C,firstThreadId2 = UUID_03:threadId_03
                + "local firstThreadId2 = redis.call('lindex', KEYS[2], 0);”
                // 1. 因为firstThreadId2不为空,所以if条件不成立
                + "if firstThreadId2 == false then "
                    + "break;"
                + "end; “
                // 1.zscore redisson_lock_timeout:{anyLock} UUID_03:threadId_03,取出队列中的第一个元素的分数
                + "local timeout = tonumber(redis.call('zscore', KEYS[3], firstThreadId2));”
                // 1. 10:01:04(客户端C的超时时间) <= 10:00:40(当前时间) ? 条件不成立
                + "if timeout <= tonumber(ARGV[4]) then "
                    + "redis.call('zrem', KEYS[3], firstThreadId2); "
                    + "redis.call('lpop', KEYS[2]); "
                + "else "
                    + "break;"
                + "end; "
              + "end;"
                // 1.exists anyLock == 0 锁不存在,这个条件是成立的,因为客户端A已经释放锁了
                // 1.exists redisson_lock_queue:{anyLock} 不存在或者lindex redisson_lock_queue:{anyLock} 第一个元素的值
                // 1. UUID_03:threadId_03 = UUID_02:threadId_02(当前是客户端B来获取锁),所以综合,if条件不成立,无法加锁

                // 2.anyLock 不存在,而且队头就是客户端C,所以if条件成立,进入正常的加锁逻辑
                  + "if (redis.call('exists', KEYS[1]) == 0) and ((redis.call('exists', KEYS[2]) == 0) "
                        + "or (redis.call('lindex', KEYS[2], 0) == ARGV[2])) then “ +
                    //2.lpop redisson_lock_queue:{anyLock} 将客户端C从队列中弹出
                    //2.完成加锁,而且客户端C从队列中出队,此时排在队头的就是客户端B了
                        "redis.call('lpop', KEYS[2]); “ +
                    //2.zerm redisson_lock_timeout:{anyLock} UUID_03:threadId_03 ,从有序集合中把客户端C的线程ID删除
                        "redis.call('zrem', KEYS[3], ARGV[2]); “ +
                    // 2.hset anyLock UUID_03:threadId_03 1 
                    // 2.数据结构为:anyLock:{UUID_03:threadiD_03 : 1}其实就是对客户端C进行加锁
                        "redis.call('hset', KEYS[1], ARGV[2], 1); “ +
                    // 2. pexpire anyLock 30000 刷新锁的生存时间为30000毫秒,即:30秒
                        "redis.call('pexpire', KEYS[1], ARGV[1]); “ +
                    // 2.返回nil,java代码拿到null,就会启动watchdog看门狗,后台定时调度定时进行刷新
                        "return nil; " +
                    "end; “ +
                    // 1.hexists anyLock UUID_03:threadId_03 == 1,其实仔细想一下就知道了,这块的anyLock已经不存在了,
                    // 所以这里的if条件也是不成立的,其实稍微想一下,就知道,客户端A释放锁的时候就已经把anyLock给删除了,
                    // 1.这里的条件是怎么都不会成立的
                    "if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then " +
                        "redis.call('hincrby', KEYS[1], ARGV[2], 1); " +
                        "redis.call('pexpire', KEYS[1], ARGV[1]); " +
                        "return nil; " +
                    "end; " +
                    //1.lindex redisson_lock_queue:{anyLock} 0 取出队列的第一个元素,UUID_03:threadId_03,为客户端C    
                    "local firstThreadId = redis.call('lindex', KEYS[2], 0); " +
                    "local ttl; “ + 
                    // 1.if条件成立,首先是firstThreadId不为空且firstThreadId不等于UUID_03:threadId_03
                    "if firstThreadId ~= false and firstThreadId ~= ARGV[2] then “ + 
                    // 1.重新计算ttl,zscore redisson_lock_timeout:{anyLock} UUID_03:threadId_03 
                    // 1. 10:01:04(客户端C的超时时间) - 10:00:40(当前时间) = 24000毫秒 
                        "ttl = tonumber(redis.call('zscore', KEYS[3], firstThreadId)) - tonumber(ARGV[4]);" + 
                    "else “
                      + "ttl = redis.call('pttl', KEYS[1]);" + 
                    "end; " + 
                     
                    // 1. timeout = 24000毫秒 + 10:00:40 + 5000 = 10:01:09(在Redis里面的数据表现形式是long类型的时间戳)   
                    "local timeout = ttl + tonumber(ARGV[3]);” + 
                    // 1.zadd redisson_lock_timeout:{anyLock} 10:01:09 UUID_02:threadId_02 
                    // 1. 刷新一下客户端B在有序集合中的timeout分数,10:01:09
                    // 1. 所以,根据我们之前的假设,即使客户端A释放了锁,但是现在排在队头的是客户端C,
                    // 1.客户端B现在来获取锁也是获取不到的,只是把客户端B在有序集合中的分数刷新一下而已
                    "if redis.call('zadd', KEYS[3], timeout, ARGV[2]) == 1 then “ +
                    // 1. rpush redisson_lock_queue:{anyLock} UUID_02:threadId_02
                        "redis.call('rpush', KEYS[2], ARGV[2]);" +
                    "end; " +
                    "return ttl;", 
                    Arrays.<Object>asList(getName(), threadsQueueName, timeoutSetName), 
                                internalLockLeaseTime, getLockName(threadId), currentTime + threadWaitTime, currentTime);
    }
    
    throw new IllegalArgumentException();
}

四、总结

  1. 获取锁超时,其他客户端获取不到锁,一定会在java代码里进入一个while true死循环,一定时间内没有获取到锁,就返回false标识获取锁失败,过了一段时间,只要没有刷新有序集合中的timeout分数,就会自然被lua脚本里的while true逻辑给清理掉
  2. 超时自动释放锁,不会开启lock watchdog后台定时调度的任务
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,837评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,551评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,417评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,448评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,524评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,554评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,569评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,316评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,766评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,077评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,240评论 1 343
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,912评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,560评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,176评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,425评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,114评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,114评论 2 352