RedisTemplate用SETNX命令实现分布式锁

使用SETNX命令获取分布式锁的步骤:

  • C1和C2线程同时检查时间戳获取锁,执行SETNX命令并都返回0,此时锁仍被C3持有,并且C3已经崩溃
  • C1 DEL
  • C1 使用SETNX命令获取锁,并且成功
  • C2 DEL
  • C2 使用SETNX命令获取锁,并且成功
  • ERROR : 由于竞态条件,C1和C2都获取到了锁

幸运的是,以下面的步骤完全可以避免这种情况发生,看看C4线程如何操作

  • C4使用SETNX命令获取锁
  • C3已经崩溃但是仍然持有锁,所以Redis返回0给C4
  • C4使用GET命令获取锁并检查锁是否已经过期,如果没有过期,则继续等待一段时间并重新重试
  • 如果锁已经过期,C4尝试 GETSET lock.foo
  • 利用GETSET语法,C4可以检查旧时间是否仍然是过期时间,如果是,则获取锁
  • 如果另一个客户端C5率先获取到锁,C4执行GETSET命令后将返回非过期时间,然后C4继续从头开始重新尝试获取锁。此操作C4将延长一点C5获取到的锁的过期时间,不过这不是什么大问题。
private static String LOCK_PREFIX = "prefix";

public boolean lock(String key) {

        String lock = LOCK_PREFIX + key;

        return (Boolean) redisTemplate.execute((RedisCallback) connection -> {

            long expireAt = System.currentTimeMillis() + LOCK_EXPIRE + 1;
            Boolean acquire = connection.setNX(lock.getBytes(), String.valueOf(expireAt).getBytes());


            if (acquire) {
                return true;
            } else {

                byte[] value = connection.get(lock.getBytes());

                if (Objects.nonNull(value) && value.length > 0) {

                    long expireTime = Long.parseLong(new String(value));

                    if (expireTime < System.currentTimeMillis()) {
                        byte[] oldValue = connection.getSet(lock.getBytes(), String.valueOf(System.currentTimeMillis() + LOCK_EXPIRE + 1).getBytes());

                        return Long.parseLong(new String(oldValue)) < System.currentTimeMillis();
                    }
                }
            }
            return false;
        });
    }

链接:更强大的Redis分布式客户端 - Redisson

你可能感兴趣的:(RedisTemplate用SETNX命令实现分布式锁)