大家好，我是石头！

在前面的文章中，我曾经给大家说过要怎样实现Redis分布式锁（见《Redis实现分布式锁的正确姿势》）。但是后来发现，我给大家讲的这个实现方案，姿势也不太对。用这篇文章中的方式来实现分布式锁，会导致分布式锁无法重入。

无法重入的原因

在《Redis实现分布式锁的正确姿势》中，我们是通过Redis的set命令，再加上其NX参数来实现分布式锁的。而设置NX参数后，如果Redis中已存在key（即锁），则无法写入成功，也就无法成功加锁。因此，如果一个线程已经获取到一个锁A以后，在未释放锁之前，即便是其本身，也无法再获取到相同的锁A，即无法重入。

怎么解决？

从上述可知，我们没办法在Redis层级去实现可重入，因此，我们需要再引入一个组件，用来记录当前线程获取到的锁及其次数，从而让我们在加锁的时候，可以校验当前线程已获取过多少次锁。而与多线程相关，且各自维护数据的，我们首先想到的就是ThreadLocal。

ThreadLocal会为每一个线程都提供一份变量的副本，从而实现同时访问而互不相干扰。

实现逻辑

我们先为分布式锁定义一个ThreadLocal，代码如下：

/**线程本地存储，存储锁的密钥和计数:结构为 锁id：加锁计数*/
private static ThreadLocal<Map<String, Integer>> threadLocal = new ThreadLocal<>();

加锁

/**
  * 尝试获取锁
  */
@Override
public boolean tryLock()
    // 当前线程持有的锁及其计数
    Map<String,Integer> value = threadLocal.get();
    if(value.containsKey(lockKey)){
        value.put(lockKey,value.get(lockKey) + 1);
        return true;
    }


    // 获取锁
    String result = jedisCluster.set(lockKey, requestId, "NX", "EX", 2);
    // 获取锁成功
    if (StringUtils.isNotBlank(result) && LOCK_SUCCESS.equals(result)) {
        value.put(lockKey,1);
        return true;
    }
    return false;
}

如上述代码，在获取锁的时候，先确认当前线程是否已持有锁，有的话，增加锁计数；没有的话，就调用Redis实现加锁。加锁成功后，将lockKey写入到ThreadLocal中，表示当前线程已持有该锁。

释放锁

/**
  * 释放锁
  */
@Override
public void unlock(String lockKey,String requestId) {
    // 当前线程持有的锁及其计数
    Map<String,Integer> value = threadLocal.get();
    if(value.containsKey(lockKey)){
        Integer num = value.get(lockKey) - 1;
        // 锁计数小于等于0，需要调用Redis进行锁释放
        if(num <= 0){
            // 若redis中存在lockKey，则删除lockKey，否则返回0
            String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
            Object result = jedisCluster.eval(script,Collections.singletonList(lockKey),Collections.singletonList(requestId));
            // 成功
            if (result != null && "OK".equals(result.toString())) {
                logger.debug("释放锁成功,lockKey:{}, requestId:{}", lockKey, requestId);
                value.remove(lockKey);
            } else {// 失败
                logger.debug("释放锁失败,lockKey:{}, requestId:{}", lockKey, requestId);
            }
        } else {
            // 减少锁计数
            value.put(lockKey,num);
        }
    } else {
        logger.debug("释放锁成功,当前线程未持有锁,lockKey:{}, requestId:{}", lockKey, requestId);
    }
}

如上述代码，在释放锁的时候，先确认当前线程是否已持有锁，有的话，减少锁计数；没有的话，直接释放成功。如果锁计数小于等于0，则需要调用Redis释放锁，释放成功后，将lockKey从ThreadLocal中删除，表示当前线程已不再持有该锁。

感谢您的认真阅读！如果喜欢本篇文章，就辛苦点个赞，关注一下我吧

长 按 二 维 码 ， 一 键 关 注

MORE | 更多精彩文章

• Spring Boot 3.2 正式发布
• 大厂又崩了~
• Redis实现分布式锁的正确姿势！