閱讀699 返回首頁    go 阿裏雲 go 技術社區[雲棲]


信號量

Semaphore(信號量) 是一個線程同步結構,用於在線程間傳遞信號,以避免出現信號丟失(譯者注:下文會具體介紹),或者像鎖一樣用於保護一個關鍵區域。自從5.0開始,jdk 在java.util.concurrent包裏提供了Semaphore 的官方實現,因此大家不需要自己去實現Semaphore。但是還是很有必要 去熟悉如何使用Semaphore及其背後的原理

本文的涉及的主題如下:

  1. 簡單的Semaphore實現
  2. 使用Semaphore來發出信號
  3. 可計數的Semaphore
  4. 有上限的Semaphore
  5. 把Semaphore當鎖來使用

一、簡單的Semaphore實現

下麵是一個信號量的簡單實現:

public class Semaphore {

private boolean signal = false;

public synchronized void take() {

this.signal = true;

this.notify();

}

public synchronized void release() throws InterruptedException{

while(!this.signal) wait();

this.signal = false;

}

}

Take方法發出一個被存放在Semaphore內部的信號,而Release方法則等待一個信號,當其接收到信號後,標記位signal被清空,然後該方法終止。

使用這個semaphore可以避免錯失某些信號通知。用take方法來代替notify,release方法來代替wait。如果某線程在調用 release等待之前調用take方法,那麼調用release方法的線程仍然知道take方法已經被某個線程調用過了,因為該Semaphore內部 保存了take方法發出的信號。而wait和notify方法就沒有這樣的功能。

當用semaphore來產生信號時,take和release這兩個方法名看起來有點奇怪。這兩個名字來源於後麵把semaphore當做鎖的例子,後麵會詳細介紹這個例子,在該例子中,take和release這兩個名字會變得很合理。

二、使用Semaphore來產生信號

下麵的例子中,兩個線程通過Semaphore發出的信號來通知對方

Semaphore semaphore = new Semaphore();

SendingThread sender = new SendingThread(semaphore);

ReceivingThread receiver = new ReceivingThread(semaphore);

receiver.start();

sender.start();

public class SendingThread {

Semaphore semaphore = null;

public SendingThread(Semaphore semaphore){

this.semaphore = semaphore;

}

public void run(){

while(true){

//do something, then signal

this.semaphore.take();

}

}

}

public class RecevingThread {

Semaphore semaphore = null;

public ReceivingThread(Semaphore semaphore){

this.semaphore = semaphore;

}

public void run(){

while(true){

this.semaphore.release();

//receive signal, then do something...

}

}

}

三、可計數的Semaphore

上麵提到的Semaphore的簡單實現並沒有計算通過調用take方法所產生信號的數量。可以把它改造成具有計數功能的Semaphore。下麵是一個可計數的Semaphore的簡單實現。

public class CountingSemaphore {

private int signals = 0;

public synchronized void take() {

this.signals++;

this.notify();

}

public synchronized void release() throws InterruptedException{

while(this.signals == 0) wait();

this.signals--;

}

}

四、有上限的Semaphore

上麵的CountingSemaphore並沒有限製信號的數量。下麵的代碼將CountingSemaphore改造成一個信號數量有上限的BoundedSemaphore。

public class BoundedSemaphore {

private int signals = 0;

private int bound   = 0;

public BoundedSemaphore(int upperBound){

this.bound = upperBound;

}

public synchronized void take() throws InterruptedException{

while(this.signals == bound) wait();

this.signals++;

this.notify();

}

public synchronized void release() throws InterruptedException{

while(this.signals == 0) wait();

this.signals--;

this.notify();

}

}

在BoundedSemaphore中,當已經產生的信號數量達到了上限,take方法將阻塞新的信號產生請求,直到某個線程調用release方法後,被阻塞於take方法的線程才能傳遞自己的信號。

五、把Semaphore當鎖來使用

當信號量的數量上限是1時,Semaphore可以被當做鎖來使用。通過take和release方法來保護關鍵區域。請看下麵的例子:

BoundedSemaphore semaphore = new BoundedSemaphore(1);

...

semaphore.take();

try{

//critical section

} finally {

semaphore.release();

}

在前麵的例子中,Semaphore被用來在多個線程之間傳遞信號,這種情況下,take和release分別被不同的線程調用。但是在鎖這個例子 中,take和release方法將被同一線程調用,因為隻允許一個線程來獲取信號(允許進入關鍵區域的信號),其它調用take方法獲取信號的線程將被 阻塞,知道第一個調用take方法的線程調用release方法來釋放信號。對release方法的調用永遠不會被阻塞,這是因為任何一個線程都是先調用 take方法,然後再調用release。

通過有上限的Semaphore可以限製進入某代碼塊的線程數量。設想一下,在上麵的例子中,如果BoundedSemaphore 上限設為5將 會發生什麼?意味著允許5個線程同時訪問關鍵區域,但是你必須保證,這個5個線程不會互相衝突。否則你的應用程序將不能正常運行。

必須注意,release方法應當在finally塊中被執行。這樣可以保在關鍵區域的代碼拋出異常的情況下,信號也一定會被釋放。


文章轉自 並發編程網-ifeve.com

最後更新:2017-05-22 18:01:48

  上一篇:go  顛覆大數據分析之Storm簡介
  下一篇:go  阿裏雲總裁胡曉明40個精彩回答 涉及CDN、價格戰、生態競爭等