1,基于数据库(性能较差,锁表的风险,非阻塞,失败需要轮询耗CPU)
核心思想:
在数据库中创建一个表,表中包含方法名等字段,并在方法名字段上创建唯一索引
想要执行某个方法,就使用这个方法名向表中插入数据,成功插入则获取锁
执行完成后删除对应的行数据释放锁。
2,基于REDIS(过期时间不好控制,非阻塞,失败需要轮询耗CPU)
setnx + expire = 非原子性
(setnx无法在插入值的同时设置超时,setnx 与 expire 是两条独立的语句,这样加锁操作就是非原子性的)
set key value [EX seconds] [PX milliseconds] [NX|XX]
在redis2.6.12版本之后,redis支持通过set在设置值得同时设置超时时间,此操作是原子操作
如:set lock 8 EX 6 NX
Redis有很高的性能;
Redis命令对此支持较好,实现起来比较方便
3,基于ZK(高可用、可重入、阻塞锁)
创建一个目录mylock;
获取锁就在mylock目录下创建临时顺序节点;
线程A获取mylock目录下所有的子节点,然后获取比自己小的兄弟节点
(如果不存在,则说明当前线程顺序号最小,获得锁)
线程B获取所有节点,判断自己不是最小节点,设置监听比自己次小的节点;
线程A处理完,删除自己的节点,线程B监听到变更事件,判断自己是不是最小的节点,如果是则获得锁。
优点:具备高可用、可重入、阻塞锁特性,可解决失效死锁问题。
缺点:因为需要频繁的创建和删除节点,性能上不如Redis方式。
public static String tryLock(Jedis jedis, int timeout) throws Exception{
if(timeout == 0){
timeout = 5000;
}
String returnId = null;
// 生成随机标识
String id = UUID.randomUUID().toString();
// 设置锁超时10秒
int lockExpireMs = 10000;
long startTime = System.currentTimeMillis();
// 超时时间内循环获取
while ((System.currentTimeMillis() - startTime) < timeout){
String result = jedis.set(lockKey, id, "NX", "PX", lockExpireMs);
if(result != null){
returnId = id;
break;
}
TimeUnit.MILLISECONDS.sleep(100);
}
if(returnId == null){
// 获取锁超时,抛出异常
throw new Exception("获取锁超时");
}
// 将set的值返回,用于后续的解锁
return returnId;
}
/**
* 释放1:利用redis的watch + del
*/
public static boolean unLock(Jedis jedis, String id){
boolean result = false;
while(true){
if(jedis.get(lockKey) == null){
return false;
}
// 配置监听
jedis.watch(lockKey);
// 这里确保是加锁者进行解锁
if(id!=null && id.equals(jedis.get(lockKey))){
Transaction transaction = jedis.multi();
transaction.del(lockKey);
List<Object> results = transaction.exec();
if(results == null){
continue;
}
result = true;
}
// 释放监听
jedis.unwatch();
break;
}
return result;
}
/**
* 释放2:利用lua
*/
public static boolean unLockByLua(Jedis jedis, String id){
String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(id));
if (Objects.equals(1, result)) {
return true;
}
return false;
}