自 Java 5 開始,java.util.concurrent.locks 包中包含了一些鎖的實現,因此你不用去實現自己的鎖了。但是你仍然需要去了解怎樣使用這些鎖。
讓我們從 java 中的一個同步塊開始:
public class Counter{ private int count = 0; public int inc(){ synchronized(this){ return ++count; } } }
可以看到在 inc()方法中有一個 synchronized(this)代碼塊。該代碼塊可以保證在同一時間只有一個線程可以執行 return ++count。雖然在 synchronized 的同步塊中的代碼可以更加復雜,但是++count 這種簡單的操作已經足以表達出線程同步的意思。
以下的 Counter 類用 Lock 代替 synchronized 達到了同樣的目的:
public class Counter{ private Lock lock = new Lock(); private int count = 0; public int inc(){ lock.lock(); int newCount = ++count; lock.unlock(); return newCount; } }
lock()方法會對 Lock 實例對象進行加鎖,因此所有對該對象調用 lock()方法的線程都會被阻塞,直到該 Lock 對象的 unlock()方法被調用。
這裡有一個 Lock 類的簡單實現:
public class Counter{ public class Lock{ private boolean isLocked = false; public synchronized void lock() throws InterruptedException{ while(isLocked){ wait(); } isLocked = true; } public synchronized void unlock(){ isLocked = false; notify(); } }
注意其中的 while(isLocked)循環,它又被叫做“自旋鎖”。當 isLocked 為 true 時,調用 lock()的線程在 wait()調用上阻塞等待。為防止該線程沒有收到 notify()調用也從 wait()中返回(也稱作虛假喚醒),這個線程會重新去檢查 isLocked 條件以決定當前是否可以安全地繼續執行還是需要重新保持等待,而不是認為線程被喚醒了就可以安全地繼續執行了。如果 isLocked 為 false,當前線程會退出 while(isLocked)循環,並將 isLocked 設回 true,讓其它正在調用 lock()方法的線程能夠在 Lock 實例上加鎖。
當線程完成了臨界區(位於 lock()和 unlock()之間)中的代碼,就會調用 unlock()。執行 unlock()會重新將 isLocked 設置為 false,並且通知(喚醒)其中一個(若有的話)在 lock()方法中調用了 wait()函數而處於等待狀態的線程。
Java 中的 synchronized 同步塊是可重入的。這意味著如果一個 java 線程進入了代碼中的 synchronized 同步塊,並因此獲得了該同步塊使用的同步對象對應的管程上的鎖,那麼這個線程可以進入由同一個管程對象所同步的另一個 java 代碼塊。下面是一個例子:
public class Reentrant{ public synchronized outer(){ inner(); } public synchronized inner(){ //do something } }
注意 outer()和 inner()都被聲明為 synchronized,這在 Java 中和 synchronized(this)塊等效。如果一個線程調用了 outer(),在 outer()裡調用 inner()就沒有什麼問題,因為這兩個方法(代碼塊)都由同一個管程對象(”this”)所同步。如果一個線程已經擁有了一個管程對象上的鎖,那麼它就有權訪問被這個管程對象同步的所有代碼塊。這就是可重入。線程可以進入任何一個它已經擁有的鎖所同步著的代碼塊。
前面給出的鎖實現不是可重入的。如果我們像下面這樣重寫 Reentrant 類,當線程調用 outer()時,會在 inner()方法的 lock.lock()處阻塞住。
public class Reentrant2{ Lock lock = new Lock(); public outer(){ lock.lock(); inner(); lock.unlock(); } public synchronized inner(){ lock.lock(); //do something lock.unlock(); } }
調用 outer()的線程首先會鎖住 Lock 實例,然後繼續調用 inner()。inner()方法中該線程將再一次嘗試鎖住 Lock 實例,結果該動作會失敗(也就是說該線程會被阻塞),因為這個 Lock 實例已經在 outer()方法中被鎖住了。
兩次 lock()之間沒有調用 unlock(),第二次調用 lock 就會阻塞,看過 lock()實現後,會發現原因很明顯:
public class Lock{ boolean isLocked = false; public synchronized void lock() throws InterruptedException{ while(isLocked){ wait(); } isLocked = true; } ... }
一個線程是否被允許退出 lock()方法是由 while 循環(自旋鎖)中的條件決定的。當前的判斷條件是只有當 isLocked 為 false 時 lock 操作才被允許,而沒有考慮是哪個線程鎖住了它。
為了讓這個 Lock 類具有可重入性,我們需要對它做一點小的改動:
public class Lock{ boolean isLocked = false; Thread lockedBy = null; int lockedCount = 0; public synchronized void lock() throws InterruptedException{ Thread callingThread = Thread.currentThread(); while(isLocked && lockedBy != callingThread){ wait(); } isLocked = true; lockedCount++; lockedBy = callingThread; } public synchronized void unlock(){ if(Thread.curentThread() == this.lockedBy){ lockedCount--; if(lockedCount == 0){ isLocked = false; notify(); } } } ... }
注意到現在的 while 循環(自旋鎖)也考慮到了已鎖住該 Lock 實例的線程。如果當前的鎖對象沒有被加鎖(isLocked = false),或者當前調用線程已經對該 Lock 實例加了鎖,那麼 while 循環就不會被執行,調用 lock()的線程就可以退出該方法(譯者注:“被允許退出該方法”在當前語義下就是指不會調用 wait()而導致阻塞)。
除此之外,我們需要記錄同一個線程重復對一個鎖對象加鎖的次數。否則,一次 unblock()調用就會解除整個鎖,即使當前鎖已經被加鎖過多次。在 unlock()調用沒有達到對應 lock()調用的次數之前,我們不希望鎖被解除。
現在這個 Lock 類就是可重入的了。
Java 的 synchronized 塊並不保證嘗試進入它們的線程的順序。因此,如果多個線程不斷競爭訪問相同的 synchronized 同步塊,就存在一種風險,其中一個或多個線程永遠也得不到訪問權 —— 也就是說訪問權總是分配給了其它線程。這種情況被稱作線程饑餓。為了避免這種問題,鎖需要實現公平性。本文所展現的鎖在內部是用 synchronized 同步塊實現的,因此它們也不保證公平性。
如果用 Lock 來保護臨界區,並且臨界區有可能會拋出異常,那麼在 finally 語句中調用 unlock()就顯得非常重要了。這樣可以保證這個鎖對象可以被解鎖以便其它線程能繼續對其加鎖。以下是一個示例:
lock.lock(); try{ //do critical section code, //which may throw exception } finally { lock.unlock(); }
這個簡單的結構可以保證當臨界區拋出異常時 Lock 對象可以被解鎖。如果不是在 finally 語句中調用的 unlock(),當臨界區拋出異常時,Lock 對象將永遠停留在被鎖住的狀態,這會導致其它所有在該 Lock 對象上調用 lock()的線程一直阻塞。