redis setnx java_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;

});

}