redisTemplate通过setNx实现分布式锁

Posted 2020-11-30

 

• 客户端C2使用SETNX命令获取锁

• 假设客户端C1已经崩溃但是仍然持有锁，所以Redis返回false给客户端C2

• 客户端C2使用GET命令获取锁并检查锁是否已经过期，如果没有过期，则继续等待一段时间并重新重试

• 如果锁已经过期，客户端C2尝试 GETSET lock.name <current Unix timestamp + lock timeout + 1>

• 利用GETSET语法，客户端C2可以检查key的旧值(锁的旧时间)是否仍然是过期时间，如果是，则获取锁

• 如果另一个客户端C3率先获取到锁，客户端C2执行GETSET命令后将返回非过期时间，然后客户端C2继续从头开始重新尝试获取锁。此操作客户端C2将会延长一点客户端C3获取到的锁的过期时间，不过这不是什么大问题。

 /**
     * 获取一个redis分布锁
     *
     * @param lockKey        锁住的key
     * @param lockExpireMils 锁住的时长。如果超时未解锁，视为加锁线程死亡，其他线程可夺取锁
     * @return
     */
    public boolean lock(String lockKey, long lockExpireMils) {
        return (Boolean) redisTemplate.execute((RedisCallback) connection -> {
            long nowTime = System.currentTimeMillis();
            Boolean acquire = connection.setNX(lockKey.getBytes(), String.valueOf(nowTime + lockExpireMils + 1).getBytes());
            if (acquire) {
                return Boolean.TRUE;
            } else {
                byte[] value = connection.get(lockKey.getBytes());
                if (Objects.nonNull(value) && value.length > 0) {
                    long oldTime = Long.parseLong(new String(value));
                    if (oldTime < nowTime) {
                        //connection.getSet：返回这个key的旧值并设置新值。
                        byte[] oldValue = connection.getSet(lockKey.getBytes(), String.valueOf(nowTime + lockExpireMils + 1).getBytes());
                        //当key不存时会返回空，表示key不存在或者已在管道中使用
                        return oldValue == null ? false : Long.parseLong(new String(oldValue)) < nowTime;
                    }
                }
            }
            return Boolean.FALSE;
        });
    }

 

 

 

参考地址

RedisTemplate用SETNX命令实现分布式锁 https://www.jianshu.com/p/6dd656e7051f