分析java並發中的wait notify notifyAll

一、前言

java 面試是否有被問到過,sleepwait 方法的區別,關於這個問題其實不用多說,大多數人都能回答出最主要的兩點區別:

  • sleep 是線程的方法, wait / notify / notifyAll 是 Object 類的方法;
  • sleep 不會釋放當前線程持有的鎖,到時間後程序會繼續執行,wait 會釋放線程持有的鎖並掛起,直到通過 notify 或者 notifyAll 重新獲得鎖。

另外還有一些參數、異常等區別,不細說瞭。本文重點記錄一下 wait / notify / notifyAll 的相關知識。

二、常見的同步場景

開發中常常遇到這樣的場景:

一個線程執行過程中,需要開啟另外一個子線程去做某個耗時的操作(通過休眠3秒模擬),並且**等待**子線程返回結果,主線程再根據返回的結果繼續往下執行。

這裡註意我上面加*兩個字“等待”。如果不需要等待,單純隻是對子線程的結果做處理,我們大可註冊回調方法解決問題,此文不再贅述接口回調。

此處場景就是主線程停下來等待子線程執行完畢後,主線程再繼續執行。針對該場景下面給出實現:

2.1、設置一個判斷的標志位

volatile boolean flag = false;

public void test(){
    //...

    Thread t1 = new Thread(() -> {
        try {
            Thread.sleep(3000);
            System.out.println("--- 休眠 3 秒");
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            flag = true;
        }
    });
    t1.start();

    while(!flag){

    }
    System.out.println("--- work thread run");
}

上面的代碼,執行結果:

強調一點,聲明標志位的時候,一定註意 volatile 關鍵字不能忘,如果不加該關鍵字修飾,程序可能進入死循環。這是同步中的可見性問題,在 《java 並發——內置鎖》 中有記錄。

顯然,這個實現方案並不好,本來主線程什麼也不用做,卻一直在競爭資源,做空循環,性能上不好,所以並不推薦。

2.2、線程的 join 方法

public void test(){
    //...

    Thread t1 = new Thread(() -> {
        try {
            Thread.sleep(3000);
            System.out.println("--- 休眠 3 秒");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    });

    t1.start();

    try {
        t1.join();
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    System.out.println("--- work thread continue");
}

上面的代碼,執行結果同上。利用 Thread 類的 join 方法實現瞭同步,達到瞭效果,但是 join 方法不能一定保證效果,在不同的 cpu 上,可能呈現出意想不到的結果,所以盡量不要用上述方法。

2.3、使用閉鎖 CountDownLatch

不清楚閉鎖的新同學可以瞭解下java並發中的線程。

public void test(){
    //...

    final CountDownLatch countDownLatch = new CountDownLatch(1);

    Thread t1 = new Thread(() -> {
        try {
            Thread.sleep(3000);
            System.out.println("--- 休眠 3 秒");
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            countDownLatch.countDown();
        }
    });

    t1.start();

    try {
        countDownLatch.await();
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    System.out.println("--- work thread run");
}

上面的代碼,執行結果同上。同樣可以實現上述效果,執行結果和上面一樣。該方法推薦使用。

2.4、利用 wait / notify 優化標志位方法

為瞭方便對比,首先給 2.1 中的循環方法增加一些打印。修改後的代碼如下:

volatile boolean flag = false;

public void test() {
    //...
    Thread t1 = new Thread(() -> {
        try {
            Thread.sleep(3000);
            System.out.println("--- 休眠 3 秒");
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            flag = true;
        }
    });
    t1.start();

    while (!flag) {
        try {
            System.out.println("---while-loop---");
            Thread.sleep(500);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
    System.out.println("--- work thread run");
}

執行結果如下:

事實證明,while 循環確實一直在執行。

為瞭使該線程再不需要執行的時候不搶占資源,我們可以利用 wait 方法將其掛起,在需要它執行的時候,再利用 notify 方法將其喚醒。這樣達到優化的目的,優化後的代碼如下:

volatile boolean flag = false;

public void test() {
    //...
    final Object obj = new Object();
    Thread t1 = new Thread(() -> {
        synchronized (obj) {
            try {
                Thread.sleep(3000);
                System.out.println("--- 休眠 3 秒");
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                flag = true;
            }
            obj.notify();
        }
    });
    t1.start();

    synchronized (obj) {
        while (!flag) {
            try {
                System.out.println("---while-loop---");
                Thread.sleep(500);
                obj.wait();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
    System.out.println("--- work thread run");

}

執行結果:

結果證明,優化後的程序,循環隻執行瞭一次。

三、理解 wait / notify / notifyAll

在Java中,每個對象都有兩個池,鎖(monitor)池和等待池

3.1、鎖池

鎖池:假設線程A已經擁有瞭某個對象的鎖,而其它的線程想要調用這個對象的某個synchronized方法(或者synchronized塊),由於這些線程在進入對象的synchronized方法之前必須先獲得該對象的鎖的擁有權,但是該對象的鎖目前正被線程A擁有,所以這些線程就進入瞭該對象的鎖池中。

3.2、等待池

等待池:假設一個線程A調用瞭某個對象的wait()方法,線程A就會釋放該對象的鎖(因為wait()方法必須出現在synchronized中,這樣自然在執行wait()方法之前線程A就已經擁有瞭該對象的鎖),同時線程A就進入到瞭該對象的等待池中。如果另外的一個線程調用瞭相同對象的notifyAll()方法,那麼處於該對象的等待池中的線程就會全部進入該對象的鎖池中,準備爭奪鎖的擁有權。如果另外的一個線程調用瞭相同對象的notify()方法,那麼僅僅有一個處於該對象的等待池中的線程(隨機)會進入該對象的鎖池.

3.3、notify 和 notifyAll 的區別

3.3.1、wait()

public final void wait() throws InterruptedException,IllegalMonitorStateException
該方法用來將當前線程置入休眠狀態,直到接到通知或被中斷為止。在調用 wait()之前,線程必須要獲得該對象的對象級別鎖,即隻能在同步方法或同步塊中調用 wait()方法。進入 wait()方法後,當前線程釋放鎖。在從 wait()返回前,線程與其他線程競爭重新獲得鎖。如果調用 wait()時,沒有持有適當的鎖,則拋出 IllegalMonitorStateException,它是 RuntimeException 的一個子類,因此,不需要 try-catch 結

3.3.2、notify()

public final native void notify() throws IllegalMonitorStateException
該方法也要在同步方法或同步塊中調用,即在調用前,線程也必須要獲得該對象的對象級別鎖,的如果調用 notify()時沒有持有適當的鎖,也會拋出 IllegalMonitorStateException。
該方法用來通知那些可能等待該對象的對象鎖的其他線程。如果有多個線程等待,則線程規劃器任意挑選出其中一個 wait()狀態的線程來發出通知,並使它等待獲取該對象的對象鎖(notify 後,當前線程不會馬上釋放該對象鎖,wait 所在的線程並不能馬上獲取該對象鎖,要等到程序退出 synchronized 代碼塊後,當前線程才會釋放鎖,wait所在的線程也才可以獲取該對象鎖),但不驚動其他同樣在等待被該對象notify的線程們。當第一個獲得瞭該對象鎖的 wait 線程運行完畢以後,它會釋放掉該對象鎖,此時如果該對象沒有再次使用 notify 語句,則即便該對象已經空閑,其他 wait 狀態等待的線程由於沒有得到該對象的通知,會繼續阻塞在 wait 狀態,直到這個對象發出一個 notify 或 notifyAll。這裡需要註意:它們等待的是被 notify 或 notifyAll,而不是鎖。這與下面的 notifyAll()方法執行後的情況不同。

3.3.3、notifyAll()

public final native void notifyAll() throws IllegalMonitorStateException

該方法與 notify ()方法的工作方式相同,重要的一點差異是:

notifyAll 使所有原來在該對象上 wait 的線程統統退出 wait 的狀態(即全部被喚醒,不再等待 notify 或 notifyAll,但由於此時還沒有獲取到該對象鎖,因此還不能繼續往下執行),變成等待獲取該對象上的鎖,一旦該對象鎖被釋放(notifyAll 線程退出調用瞭 notifyAll 的 synchronized 代碼塊的時候),他們就會去競爭。如果其中一個線程獲得瞭該對象鎖,它就會繼續往下執行,在它退出 synchronized 代碼塊,釋放鎖後,其他的已經被喚醒的線程將會繼續競爭獲取該鎖,一直進行下去,直到所有被喚醒的線程都執行完畢。

四、生產者與消費者模式

生產者與消費者問題是並發編程裡面的經典問題。接下來說說利用wait()和notify()來實現生產者和消費者並發問題:
顯然要保證生產者和消費者並發運行不出亂,主要要解決:當生產者線程的緩存區為滿的時候,就應該調用wait()來停止生產者繼續生產,而當生產者滿的緩沖區被消費者消費掉一塊時,則應該調用notify()喚醒生產者,通知他可以繼續生產;同樣,對於消費者,當消費者線程的緩存區為空的時候,就應該調用wait()停掉消費者線程繼續消費,而當生產者又生產瞭一個時就應該調用notify()來喚醒消費者線程通知他可以繼續消費瞭。

下面是一個簡單的代碼實現:

package com.sharpcj;

import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class Test {
    public static void main(String[] args) {
        Reposity reposity = new Reposity(600);
        ExecutorService threadPool = Executors.newCachedThreadPool();
        for(int i = 0; i < 10; i++){
            threadPool.submit(new Producer(reposity));
        }

        for(int i = 0; i < 10; i++){
            threadPool.submit(new Consumer(reposity));
        }
        threadPool.shutdown();
    }
}


class Reposity {
    private static final int MAX_NUM = 2000;
    private int currentNum;

    private final Object obj = new Object();

    public Reposity(int currentNum) {
        this.currentNum = currentNum;
    }

    public void in(int inNum) {
        synchronized (obj) {
            while (currentNum + inNum > MAX_NUM) {
                try {
                    System.out.println("入貨量 " + inNum + " 線程 " + Thread.currentThread().getId() + "被掛起...");
                    obj.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            try {
                Thread.sleep(200);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            currentNum += inNum;
            System.out.println("線程: " + Thread.currentThread().getId() + ",入貨:inNum = [" + inNum + "], currentNum = [" + currentNum + "]");
            obj.notifyAll();
        }
    }

    public void out(int outNum) {
        synchronized (obj) {
            while (currentNum < outNum) {
                try {
                    System.out.println("出貨量 " + outNum + " 線程 " + Thread.currentThread().getId() + "被掛起...");
                    obj.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            try {
                Thread.sleep(200);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            currentNum -= outNum;
            System.out.println("線程: " + Thread.currentThread().getId() + ",出貨:outNum = [" + outNum + "], currentNum = [" + currentNum + "]");
            obj.notifyAll();
        }
    }
}

class Producer implements Runnable {
    private Reposity reposity;

    public Producer(Reposity reposity) {
        this.reposity = reposity;
    }

    @Override
    public void run() {
        reposity.in(200);
    }
}

class Consumer implements Runnable {
    private Reposity reposity;

    public Consumer(Reposity reposity) {
        this.reposity = reposity;
    }

    @Override
    public void run() {
        reposity.out(200);
    }
}

執行結果:

五、總結

1.調用wait方法和notify、notifyAll方法前必須獲得對象鎖,也就是必須寫在synchronized(鎖對象){……}代碼塊中。

2.當線程調用瞭wait方法後就釋放瞭對象鎖,否則其他線程無法獲得對象鎖。

3.當調用 wait() 方法後,線程必須再次獲得對象鎖後才能繼續執行。

4.如果另外兩個線程都在 wait,則正在執行的線程調用notify方法隻能喚醒一個正在wait的線程(公平競爭,由JVM決定)。

5.當使用notifyAll方法後,所有wait狀態的線程都會被喚醒,但是隻有一個線程能獲得鎖對象,必須執行完while(condition){this.wait();}後才釋放對象鎖。其餘的需要等待該獲得對象鎖的線程執行完釋放對象鎖後才能繼續執行。

6.當某個線程調用notifyAll方法後,雖然其他線程被喚醒瞭,但是該線程依然持有著對象鎖,必須等該同步代碼塊執行完(右大括號結束)後才算正式釋放瞭鎖對象,另外兩個線程才有機會執行。

7.第5點中說明, wait 方法的調用前的條件判斷需放在循環中,否則可能出現邏輯錯誤。另外,根據程序邏輯合理使用 wait 即 notify 方法,避免如先執行 notify ,後執行 wait 方法,線程一直掛起之類的錯誤。

以上就是分析java並發中的wait notify notifyAll的詳細內容,更多關於java並發wait notify notifyAll的資料請關註WalkonNet其它相關文章!

推薦閱讀: