美團面試:如何實現線程任務編排?

線程任務編排指的是對多個線程任務按照一定的邏輯順序或條件進行組織和安排,以實現協同工作、順序執行或並行執行的一種機制。

1.線程任務編排 VS 線程通訊

有同學可能會想:那線程的任務編排是不是問的就是線程間通訊啊?

線程間通訊我知道了,它的實現方式總共有以下幾種方式:

  1. Object 類下的 wait()、notify() 和 notifyAll() 方法;
  2. Condition 類下的 await()、signal() 和 signalAll() 方法;
  3. LockSupport 類下的 park() 和 unpark() 方法。

但是,線程通訊和線程的任務編排是不同的兩個概念,它們的區別如下:

  • 線程任務編排主要關注的是如何組織和管理線程執行的任務序列,確保任務按照預定的邏輯和順序執行,包括任務的啓動、停止、依賴管理、執行策略(如並行、串行)以及錯誤處理等。它是關於如何有效地規劃線程的工作流程,以達成高效和正確的程序執行目標。
  • 線程通訊則是指在多線程環境中,線程之間傳遞信息和協調工作的機制。當多個線程需要共享數據或協同完成某項任務時,它們需要通過某種方式進行溝通,以確保數據的正確性和任務的同步執行。它的重點在於解決線程間的同步問題和數據一致性問題。

簡而言之,線程任務編排側重於高層次的執行計劃和流程控制,而線程通訊則專注於底層的數據交互和同步細節。在實際應用中,有效的線程任務編排往往離不開合理的線程通訊機制,兩者相輔相成,共同支撐起復雜多線程程序的正確執行。

2.線程任務編排

線程的任務編排的實現方式主要有以下兩種:

  1. FutureTask:誕生於 JDK 1.5,它實現了 Future 接口和 Runnable 接口,設計初衷是爲了支持可取消的異步計算。它既可以承載 Runnable 任務(通過包裝成 RunnableAdapter),也可以承載 Callable 任務,從而能夠返回計算結果,使用它可以實現簡單的異步任務執行和結果的等待。
  2. CompletableFuture:誕生於 JDK 8,它不僅實現了 Future 接口,還實現了 CompletionStage 接口。CompletionStage 是對 Future 的擴展,提供了豐富的鏈式異步編程模型,支持函數式編程風格,可以更加靈活地處理異步操作的組合和依賴回調等。

2.1 FutureTask 使用

FutureTask 使用示例如下:

import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.FutureTask;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class FutureTaskDemo {
    public static void main(String[] args) {
        // 創建一個Callable任務
        Callable<Integer> task = () -> {
            Thread.sleep(2000); // 模擬任務耗時操作
            return 10; // 返回任務結果
        };

        // 創建FutureTask,並將Callable任務包裝起來
        FutureTask<Integer> futureTask = new FutureTask<>(task);

        // 創建線程池
        ExecutorService executor = Executors.newCachedThreadPool();

        // 提交FutureTask給線程池執行
        executor.submit(futureTask);

        try {
            // 獲取任務結果,get()方法會阻塞直到任務完成並返回結果
            int result = futureTask.get();
            System.out.println("任務結果:" + result);
        } catch (InterruptedException | ExecutionException e) {
            e.printStackTrace();
        }
    }
}

在上述示例中,通過創建一個 Callable 任務來模擬耗時操作,並使用 FutureTask 包裝該任務。然後將 FutureTask 提交給線程池執行,最後通過 get() 方法獲取任務的執行結果,之後纔會執行後續流程。我們可以通過 get() 方法阻塞等待程序執行結果,從而完成線程任務的簡單編排。

2.2 CompletableFuture 使用

從上面 FutureTask 實現代碼可以看出,它不但寫法麻煩,而且需要使用 get() 方法阻塞等待線程的執行結果,對於異步任務的執行來說,不夠靈活且效率也會受影響,然而 CompletableFutrue 的出現,則彌補了 FutureTask 的這些缺陷。

CompletableFutrue 提供的方法有很多,但最常用和最實用的核心方法只有以下幾個:
image.png
例如,我們現在實現一個這樣的場景:
image.png
任務描述:任務一執行完之後執行任務二,任務三和任務一和任務二一起執行,所有任務都有返回值,等任務二和任務三執行完成之後,再執行任務四,它的實現代碼如下:

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;

public class CompletableFutureExample {

    public static void main(String[] args) {
        // 任務一:返回 "Task 1 result"
        CompletableFuture<String> task1 = CompletableFuture.supplyAsync(() -> {
            try {
                // 模擬耗時操作
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
                throw new RuntimeException(e);
            }
            return "Task 1 result";
        });
        // 任務二:依賴任務一,返回 "Task 2 result" + 任務一的結果
        CompletableFuture<String> task2 = task1.handle((result1, throwable) -> {
            try {
                // 模擬耗時操作
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
                throw new RuntimeException(e);
            }
            return "Task 2 result " + result1;
        });
        // 任務三:和任務一、任務二並行執行,返回 "Task 3 result"
        CompletableFuture<String> task3 = CompletableFuture.supplyAsync(() -> {
            try {
                // 模擬耗時操作
                Thread.sleep(800); // 任務三可能比任務二先完成
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
                throw new RuntimeException(e);
            }
            return "Task 3 result";
        });
        // 任務四:依賴任務二和任務三,等待它們都完成後執行,返回 "Task 4 result" + 任務二和任務三的結果
        CompletableFuture<String> task4 = CompletableFuture.allOf(task2, task3).handle((res, throwable) -> {
            try {
                // 這裏不需要顯式等待,因爲 allOf 已經保證了它們完成
                return "Task 4 result with " + task2.get() + " and " + task3.get();
            } catch (InterruptedException | ExecutionException e) {
                throw new RuntimeException(e);
            }
        });
        // 獲取任務四的結果並打印
        String finalResult = task4.join();
        System.out.println(finalResult);
    }
}

課後思考

使用 CompletableFuture 需要配合線程池一起使用嗎?爲什麼?CompletableFuture 默認的線程池是如何實現的?

本文已收錄到我的面試小站 www.javacn.site,其中包含的內容有:Redis、JVM、併發、併發、MySQL、Spring、Spring MVC、Spring Boot、Spring Cloud、MyBatis、設計模式、消息隊列等模塊。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章