天天看點

Java并發程式設計(04):線程間通信,等待/通知機制

本文源碼:GitHub·點這裡 || GitEE·點這裡

一、概念簡介

1、線程通信

在作業系統中,線程是個獨立的個體,但是線上程執行過程中,如果處理同一個業務邏輯,可能會産生資源争搶,導緻并發問題,通常使用互斥鎖來控制該邏輯。但是在還有這樣一類場景,任務執行是有順序控制的,例如常見的報表資料生成:

Java并發程式設計(04):線程間通信,等待/通知機制
  • 啟動資料分析任務,生成報表資料;
  • 報表資料存入指定位置資料容器;
  • 通知資料搬運任務,把資料寫入報表庫;

該場景在相對複雜的系統中非常常見,如果基于多線程來描述該過程,則需要線程之間通信協作,才能有條不紊的處理該場景業務。

2、等待通知機制

如上的業務場景,如果線程A生成資料過程中,線程B一直在通路資料容器,判斷該過程的資料是否已經生成,則會造成資源浪費。正常的流程應該如圖,線程A和線程B同時啟動,線程A開始處理資料生成任務,線程B嘗試擷取容器資料,資料還沒過來,線程B則進入等待狀态,當線程A的任務處理完成,則通知線程B去容器中擷取資料,這樣基于線程等待和通知的機制來協作完成任務。

3、基礎方法

等待/通知機制的相關方法是Java中Object層級的基礎方法,任何對象都有該方法:

  • notify:随機通知一個在該對象上等待的線程,使其結束wait狀态傳回;
  • notifyAll:喚醒在該對象上所有等待的線程,進入對象鎖争搶隊列中;
  • wait:線程進入waiting等待狀态,不會争搶鎖對象,也可以設定等待時間;

線程的等待通知機制,就是基于這幾個基礎方法。

二、等待通知原理

1、基本原理

等待/通知機制,該模式下指線程A在不滿足任務執行的情況下調用對象wait()方法進入等待狀态,線程B修改了線程A的執行條件,并調用對象notify()或者notifyAll()方法,線程A收到通知後從wait狀态傳回,進而執行後續操作。兩個線程通過基于對象提供的wait()/notify()/notifyAll()等方法完成等待和通知間互動,提高程式的可伸縮性。

2、實作案例

通過線程通信解決上述資料生成和存儲任務的解耦流程。

public class NotifyThread01 {

    static Object lock = new Object() ;
    static volatile List<String> dataList = new ArrayList<>();

    public static void main(String[] args) throws Exception {
        Thread saveThread = new Thread(new SaveData(),"SaveData");
        saveThread.start();
        TimeUnit.SECONDS.sleep(3);
        Thread dataThread = new Thread(new AnalyData(),"AnalyData");
        dataThread.start();
    }
    // 等待資料生成,儲存
    static class SaveData implements Runnable {
        @Override
        public void run() {
            synchronized (lock){
                while (dataList.size()==0){
                    try {
                        System.out.println(Thread.currentThread().getName()+"等待...");
                        lock.wait();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
                System.out.println("SaveData .."+ dataList.get(0)+dataList.get(1));
            }
        }
    }
    // 生成資料,通知儲存
    static class AnalyData implements Runnable {
        @Override
        public void run() {
            synchronized (lock){
                dataList.add("hello,");
                dataList.add("java");
                lock.notify();
                System.out.println("AnalyData End...");
            }
        }
    }
}
           

注意:除了dataList滿足寫條件,還要在AnalyData線程執行通知操作。

三、管道流通信

1、管道流簡介

基本概念

管道流主要用于在不同線程間直接傳送資料,一個線程發送資料到輸出管道,另一個線程從輸入管道中讀取資料,進而實作不同線程間的通信。

實作分類

管道位元組流:PipedInputStream和PipedOutputStream;

管道字元流:PipedWriter和PipedReader;

新IO管道流:Pipe.SinkChannel和Pipe.SourceChannel;

2、使用案例

public class NotifyThread02 {
    public static void main(String[] args) throws Exception {
        PipedInputStream pis = new PipedInputStream();
        PipedOutputStream pos = new PipedOutputStream();
        // 連結輸入流和輸出流
        pos.connect(pis);
        // 寫資料線程
        new Thread(new Runnable() {
            public void run() {
                BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
                // 将從鍵盤讀取的資料寫入管道流
                PrintStream ps = new PrintStream(pos);
                while (true) {
                    try {
                        System.out.print(Thread.currentThread().getName());
                        ps.println(br.readLine());
                        Thread.sleep(1000);
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            }
        }, "輸入資料線程:").start();
        // 讀資料線程
        new Thread(new Runnable() {
            public void run() {
                BufferedReader br = new BufferedReader(new InputStreamReader(pis));
                while (true) {
                    try {
                        System.out.println(Thread.currentThread().getName() + br.readLine());
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        }, "輸出資料線程:").start();
    }
}
           

寫線程向管道流寫入資料,讀線程讀取資料,完成基本通信流程。

四、生産消費模式

1、業務場景

基于線程等待通知機制:實作工廠生産一件商品,通知商店賣出一件商品的業務流程。

2、代碼實作

public class NotifyThread03 {
    public static void main(String[] args) {
        Product product = new Product();
        ProductFactory productFactory = new ProductFactory(product);
        ProductShop productShop = new ProductShop(product);
        productFactory.start();
        productShop.start();
    }
}
// 産品
class Product {
    public String name ;
    public double price ;
    // 産品是否生産完畢,預設沒有
    boolean flag ;
}
// 産品工廠:生産
class ProductFactory extends Thread {
    Product product ;
    public ProductFactory (Product product){
        this.product = product;
    }
    @Override
    public void run() {
        int i = 0 ;
        while (i < 20) {
            synchronized (product) {
                if (!product.flag){
                    if (i%2 == 0){
                        product.name = "滑鼠";
                        product.price = 79.99;
                    } else {
                        product.name = "鍵盤";
                        product.price = 89.99;
                    }
                    System.out.println("産品:"+product.name+"【價格:"+product.price+"】出廠...");
                    product.flag = true ;
                    i++;
                    // 通知消費者
                    product.notifyAll();
                } else {
                    try {
                        // 進入等待狀态
                        product.wait();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }
}
// 産品商店:銷售
class ProductShop extends Thread {
    Product product ;
    public ProductShop (Product product){
        this.product = product ;
    }
    @Override
    public void run() {
        while (true) {
            synchronized (product) {
                if (product.flag == true ){
                    System.out.println("産品:"+product.name+"【價格"+(product.price*2)+"】賣出...");
                    product.flag = false ;
                    product.notifyAll(); //喚醒生産者
                } else {
                    try {
                        product.wait();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }
}
           

流程描述:ProductFactory生成一件商品,通知商店售賣,通過flag辨別判斷控制是否進入等待狀态,商店賣出商品後,再次通知工廠生産商品。

五、源代碼位址

GitHub·位址
https://github.com/cicadasmile/java-base-parent
GitEE·位址
https://gitee.com/cicadasmile/java-base-parent
           
Java并發程式設計(04):線程間通信,等待/通知機制
序号 文章标題
01 Java并發:線程的建立方式,狀态周期管理
02 Java并發:線程核心機制,基礎概念擴充
03 Java并發:多線程并發通路,同步控制