我使出這“三板斧”(分段鎖、雜湊鎖、弱引用鎖)滅霸跑了......
阿新 • • 發佈:2020-08-27
有同學說,學了Java那麼多鎖,還是沒能鎖住滅霸,本文教你“三板斧”,鎖滅霸足矣。
據說,沒幾個人能真正參透這“三板斧”的精髓,你是不是那個有緣人呢?
最近,在工作上碰見了一些高併發的場景需要加鎖來保證業務邏輯的正確性,並且要求加鎖後效能不能受到太大的影響。初步的想法是通過資料的時間戳、id等關鍵字來加鎖,從而保證不同型別資料處理的併發性。而Java自身api提供的鎖粒度太大,很難同時滿足這些需求,於是自己動手寫了幾個簡單的擴充套件。。。
分段鎖
借鑑ConcurrentHashMap的分段思想,先生成一定數量的鎖,具體使用的時候再根據key來返回對應的lock。這是幾個實現裡最簡單,效能最高,也是最終被採用的鎖策略,程式碼如下:
- 分段鎖,系統提供一定數量的原始鎖,根據傳入物件的雜湊值獲取對應的鎖並加鎖
- 注意:要鎖的物件的雜湊值如果發生改變,有可能導致鎖無法成功釋放!!!
public class SegmentLock<T> {
private Integer segments = 16;//預設分段數量
private final HashMap<Integer, ReentrantLock> lockMap = new HashMap<>();
public SegmentLock() {
init(null, false);
}
public SegmentLock(Integer counts, boolean fair) {
init(counts, fair);
}
private void init(Integer counts, boolean fair) {
if (counts != null) {
segments = counts;
}
for (int i = 0; i < segments; i++) {
lockMap.put(i, new ReentrantLock(fair));
}
}
public void lock(T key) {
ReentrantLock lock = lockMap.get((key.hashCode()>>>1) % segments);
lock.lock();
}
public void unlock(T key) {
ReentrantLock lock = lockMap.get((key.hashCode()>>>1) % segments);
lock.unlock();
}
}
雜湊鎖
上述分段鎖的基礎上發展起來的第二種鎖策略,目的是實現真正意義上的細粒度鎖。每個雜湊值不同的物件都能獲得自己獨立的鎖。在測試中,在被鎖住的程式碼執行速度飛快的情況下,效率比分段鎖慢 30% 左右。如果有長耗時操作,感覺表現應該會更好。程式碼如下:
public class HashLock<T> {
private boolean isFair = false;
private final SegmentLock<T> segmentLock = new SegmentLock<>();//分段鎖
private final ConcurrentHashMap<T, LockInfo> lockMap = new ConcurrentHashMap<>();
public HashLock() {
}
public HashLock(boolean fair) {
isFair = fair;
}
public void lock(T key) {
LockInfo lockInfo;
segmentLock.lock(key);
try {
lockInfo = lockMap.get(key);
if (lockInfo == null) {
lockInfo = new LockInfo(isFair);
lockMap.put(key, lockInfo);
} else {
lockInfo.count.incrementAndGet();
}
} finally {
segmentLock.unlock(key);
}
lockInfo.lock.lock();
}
public void unlock(T key) {
LockInfo lockInfo = lockMap.get(key);
if (lockInfo.count.get() == 1) {
segmentLock.lock(key);
try {
if (lockInfo.count.get() == 1) {
lockMap.remove(key);
}
} finally {
segmentLock.unlock(key);
}
}
lockInfo.count.decrementAndGet();
lockInfo.unlock();
}
private static class LockInfo {
public ReentrantLock lock;
public AtomicInteger count = new AtomicInteger(1);
private LockInfo(boolean fair) {
this.lock = new ReentrantLock(fair);
}
public void lock() {
this.lock.lock();