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;
}