使用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 <current Unix timestamp + lock timeout + 1>
- 利用
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;
});
}