redisTemplate通过setNx实现分布式锁

  • 客户端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

猜你喜欢

转载自www.cnblogs.com/panchanggui/p/12689058.html