Java子线程执行完毕如何通知主线程？

Java子线程完成后的主线程通知机制

在Java多线程编程中，常常需要子线程执行完毕后通知主线程。本文将介绍两种常用的方法：CompletableFuture.allOf() 和 CountDownLatch。

方法一：使用CompletableFuture.allOf()

C

ompletableFuture 提供了 allOf() 方法，可以等待多个 CompletableFuture 对象全部完成。 我们可以为每个子线程创建一个 CompletableFuture，并在子线程执行完成后完成该 CompletableFuture。

import java.util.Arrays;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CopyOnWriteArrayList;

public class Main {
    public static void main(String[] args) {
        List list = new CopyOnWriteArrayList<>();

        CompletableFuture futureA = CompletableFuture.runAsync(() -> {
            // 子线程 A 的计算逻辑
            list.addAll(Arrays.asList("A1", "A2", "A3"));
        });

        CompletableFuture futureB = CompletableFuture.runAsync(() -> {
            // 子线程 B 的计算逻辑
            list.addAll(Arrays.asList("B1", "B2", "B3"));
        });

        CompletableFuture.allOf(futureA, futureB).thenRun(() -> {
            System.out.println("所有子线程执行完毕，开始处理 list: " + list);
        });
    }
}

方法二：使用CountDownLatch

CountDownLatch 也是一种线程同步工具。构造函数接收一个整数参数，表示需要等待的线程数量。子线程执行完成后调用 countDown() 方法递减计数器。主线程调用 await() 方法等待计数器变为零。

import java.util.Arrays;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.CopyOnWriteArrayList;

public class Main {
    public static void main(String[] args) throws InterruptedException {
        List list = new CopyOnWriteArrayList<>();
        CountDownLatch latch = new CountDownLatch(2);

        Thread threadA = new Thread(() -> {
            // 子线程 A 的计算逻辑
            list.addAll(Arrays.asList("A1", "A2", "A3"));
            latch.countDown();
        });

        Thread threadB = new Thread(() -> {
            // 子线程 B 的计算逻辑
            list.addAll(Arrays.asList("B1", "B2", "B3"));
            latch.countDown();
        });

        threadA.start();
        threadB.start();

        latch.await(); // 等待所有子线程完成

        System.out.println("所有子线程执行完毕，开始处理 list: " + list);
    }
}

线程安全考虑

需要注意的是，ArrayList 不是线程安全的。为了避免并发问题，我们使用了线程安全的 CopyOnWriteArrayList。 选择哪种方法取决于具体场景和需求，CompletableFuture 更适用于异步编程场景，而 CountDownLatch 更适用于需要精确控制线程同步的场景。