三種分佈式鎖的區別

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章