1. 程式人生 > >如何在 Java 中正確使用 wait, notify 和 notifyAll – 以生產者消費者模型為例

如何在 Java 中正確使用 wait, notify 和 notifyAll – 以生產者消費者模型為例

wait, notify 和 notifyAll,這些在多執行緒中被經常用到的保留關鍵字,在實際開發的時候很多時候卻並沒有被大家重視。本文對這些關鍵字的使用進行了描述。

在 Java 中可以用 wait、notify 和 notifyAll 來實現執行緒間的通訊。。舉個例子,如果你的Java程式中有兩個執行緒——即生產者和消費者,那麼生產者可以通知消費者,讓消費者開始消耗資料,因為佇列緩衝區中有內容待消費(不為空)。相應的,消費者可以通知生產者可以開始生成更多的資料,因為當它消耗掉某些資料後緩衝區不再為滿。

我們可以利用wait()來讓一個執行緒在某些條件下暫停執行。例如,在生產者消費者模型中,生產者執行緒在緩衝區為滿的時候,消費者在緩衝區為空的時候,都應該暫停執行。如果某些執行緒在等待某些條件觸發,那當那些條件為真時,你可以用 notify 和 notifyAll 來通知那些等待中的執行緒重新開始執行。不同之處在於,notify 僅僅通知一個執行緒,並且我們不知道哪個執行緒會收到通知,然而 notifyAll 會通知所有等待中的執行緒。換言之,如果只有一個執行緒在等待一個訊號燈,notify和notifyAll都會通知到這個執行緒。但如果多個執行緒在等待這個訊號燈,那麼notify只會通知到其中一個,而其它執行緒並不會收到任何通知,而notifyAll會喚醒所有等待中的執行緒。

如何使用Wait

儘管關於wait和notify的概念很基礎,它們也都是Object類的函式,但用它們來寫程式碼卻並不簡單。如果你在面試中讓應聘者來手寫程式碼,用wait和notify解決生產者消費者問題,我幾乎可以肯定他們中的大多數都會無所適從或者犯下一些錯誤,例如在錯誤的地方使用 synchronized 關鍵詞,沒有對正確的物件使用wait,或者沒有遵循規範的程式碼方法。說實話,這個問題對於不常使用它們的程式設計師來說確實令人感覺比較頭疼。

第一個問題就是,我們怎麼在程式碼裡使用wait()呢?因為wait()並不是Thread類下的函式,我們並不能使用Thread.call()。事實上很多Java程式設計師都喜歡這麼寫,因為它們習慣了使用Thread.sleep(),所以他們會試圖使用wait() 來達成相同的目的,但很快他們就會發現這並不能順利解決問題。正確的方法是對在多執行緒間共享的那個Object來使用wait。在生產者消費者問題中,這個共享的Object就是那個緩衝區佇列。

第二個問題是,既然我們應該在synchronized的函式或是物件裡呼叫wait,那哪個物件應該被synchronized呢?答案是,那個你希望上鎖的物件就應該被synchronized,即那個在多個執行緒間被共享的物件。在生產者消費者問題中,應該被synchronized的就是那個緩衝區佇列。

永遠在迴圈(loop)裡呼叫 wait 和 notify,不是在 If 語句

現在你知道wait應該永遠在被synchronized的背景下和那個被多執行緒共享的物件上呼叫,下一個一定要記住的問題就是,你應該永遠在while迴圈,而不是if語句中呼叫wait。因為執行緒是在某些條件下等待的——在我們的例子裡,即“如果緩衝區佇列是滿的話,那麼生產者執行緒應該等待”,你可能直覺就會寫一個if語句。但if語句存在一些微妙的小問題,導致即使條件沒被滿足,你的執行緒你也有可能被錯誤地喚醒。所以如果你不線上程被喚醒後再次使用while迴圈檢查喚醒條件是否被滿足,你的程式就有可能會出錯——例如在緩衝區為滿的時候生產者繼續生成資料,或者緩衝區為空的時候消費者開始小號資料。所以記住,永遠在while迴圈而不是if語句中使用wait!

Java wait(), notify(), notifyAll() 範例

下面我們提供一個使用wait和notify的範例程式。在這個程式裡,我們使用了上文所述的一些程式碼規範。我們有兩個執行緒,分別名為PRODUCER(生產者)和CONSUMER(消費者),他們分別繼承了了Producer和Consumer類,而Producer和Consumer都繼承了Thread類。Producer和Consumer想要實現的程式碼邏輯都在run()函式內。Main執行緒開始了生產者和消費者執行緒,並聲明瞭一個LinkedList作為緩衝區佇列(在Java中,LinkedList實現了佇列的介面)。生產者在無限迴圈中持續往LinkedList裡插入隨機整數直到LinkedList滿。我們在while(queue.size == maxSize)迴圈語句中檢查這個條件。請注意到我們在做這個檢查條件之前已經在佇列物件上使用了synchronized關鍵詞,因而其它執行緒不能在我們檢查條件時改變這個佇列。如果佇列滿了,那麼PRODUCER執行緒會在CONSUMER執行緒消耗掉佇列裡的任意一個整數,並用notify來通知PRODUCER執行緒之前持續等待。在我們的例子中,wait和notify都是使用在同一個共享物件上的。
package com.niepengfei.test4;

import java.util.Queue;
import java.util.Random;

public class MyProcduer implements Runnable{

	private Queue<Integer> quene;
	
	private int maxSize;
	
	public MyProcduer(Queue<Integer> quene,int maxSize) {
		super();
		this.quene = quene;
		this.maxSize = maxSize;
	}

	@Override
	public void run() {
		while (true) {
			synchronized (quene) {
				while (quene.size() == maxSize) {
					System.out.println("緩衝區已經滿了,請消費者消費");
					try {
						quene.wait();//當前執行緒暫停,並釋放鎖
					} catch (InterruptedException e) {
						e.printStackTrace();
					}
				}
				//產生一個隨進數,放進緩衝區
				Random random = new Random(); 
                int i = random.nextInt(); 
				quene.add(i);
				System.out.println("生產了數字:"+i);
				quene.notifyAll();
			}
		}
	}

}

package com.niepengfei.test4;

import java.util.Queue;

public class MyConsumer implements Runnable{

	private Queue<Integer> quene;
	
	private int maxSize;
	
	public MyConsumer(Queue<Integer> quene,int maxSize) {
		super();
		this.quene = quene;
		this.maxSize = maxSize;
	}
	
	@Override
	public void run() {
		while(true){
			synchronized (quene) {
				while (quene.isEmpty()) {
					try {
						System.out.println("緩衝區已經沒有數字了,請生產者生產");
						quene.wait();
					} catch (InterruptedException e) {
						e.printStackTrace();
					}
				}
				
				System.out.println("消費者消費了:" + quene.remove());
				quene.notifyAll();
			}
		}
	}

}

package com.niepengfei.test4;

import java.util.LinkedList;
import java.util.Queue;

public class Test {
	
	public static void main(String[] args) {
		Queue<Integer> queue = new LinkedList<Integer>();
		
		MyProcduer procduer = new MyProcduer(queue, 10);
		MyConsumer consumer = new MyConsumer(queue, 10);
		
		new Thread(procduer).start();
		new Thread(consumer).start();
		
	}

}

本文重點:

1. 你可以使用wait和notify函式來實現執行緒間通訊。你可以用它們來實現多執行緒(>3)之間的通訊。

2. 永遠在synchronized的函式或物件裡使用wait、notify和notifyAll,不然Java虛擬機器會生成 IllegalMonitorStateException。

3. 永遠在while迴圈裡而不是if語句下使用wait。這樣,迴圈會線上程睡眠前後都檢查wait的條件,並在條件實際上並未改變的情況下處理喚醒通知。

4. 永遠在多執行緒間共享的物件(在生產者消費者模型裡即緩衝區佇列)上使用wait。

5. 基於前文提及的理由,更傾向用 notifyAll(),而不是 notify()。