首先针对分布式锁的实现,需要注意如下几点:
- 不会发生死锁(及时客户端持有锁期间崩溃,需要保证其他客户端可以解锁)
- 容错(大部分redis节点正常运行即可)
- 互斥(只有一个客户端可以持有锁)
- 加锁和解锁必须是用一个客户端
public class RedisTool {
private static final String LOCK_SUCCESS = "OK";
private static final String SET_IF_NOT_EXIST = "NX";//如果key不存在进行set操作
private static final String SET_WITH_EXPIRE_TIME = "PX";//过期时间的设置
/**
* 尝试获取分布式锁
* @param jedis Redis客户端
* @param lockKey 锁,key来当锁
* @param requestId 请求标识,用来表示哪个锁是哪个请求加的
* @param expireTime 超期时间
* @return 是否获取成功
*/
public static boolean tryGetLock(Jedis jedis, String lockKey, String requestId, int expireTime) {
String result = jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
if (LOCK_SUCCESS.equals(result)) {
return true;
}
return false;
}
}
获取锁的代码如上。
private static final Long RELEASE_SUCCESS = 1L;
/**
* 释放分布式锁
* @param jedis Redis客户端
* @param lockKey 锁
* @param requestId 请求标识,请求标识需和加锁时的标识一致,
* @return 是否释放成功
*/
public static boolean releaseDistributedLock(Jedis jedis, String lockKey, String requestId) {
String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";//lua脚本代码,获取对应的value值,并检验requestId是否一致,如果一致则解锁
Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));//eval方法确保原子性,执行lua代码时,lua代码被当成一个命令执行,并且知道eval命令执行完成,redis才会执行其他命令
if (RELEASE_SUCCESS.equals(result)) {
return true;
}
return false;
}
解锁代码如上