android 等待线程池所有任务执行完毕

之为兄与三S

于 2024-08-05 03:32:58 发布

阅读量39

点赞数

如何在 Android 中等待线程池所有任务执行完毕

在 Android 开发中，线程池能够有效地管理和执行多线程任务，提高应用性能。为了确保在所有线程池中的任务执行完毕后，我们能够继续后续操作，了解如何正确地等待线程池的所有任务完成是至关重要的。本文将为您提供清晰的步骤指导和代码示例，以帮助您实现这一功能。

流程概述

下面的表格展示了等待线程池所有任务执行完毕的基本步骤：

步骤	描述
1	创建一个线程池
2	提交任务到线程池
3	使用 `CountDownLatch` 或 `Future` 等机制等待任务完成
4	关闭线程池

步骤详细解说

1. 创建一个线程池

首先，您需要创建一个线程池。Java 提供了 Executors 类来方便地创建线程池。以下是创建一个固定大小的线程池的代码示例：

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

ExecutorService threadPool = Executors.newFixedThreadPool(5); // 创建一个固定大小为5的线程池

2. 提交任务到线程池

您可以通过 execute 或 submit 方法将任务提交到线程池。以下是使用 submit 方法提交任务的示例：

import java.util.concurrent.Future;

Future<?> future = threadPool.submit(() -> {
    // 模拟耗时任务
    try {
        Thread.sleep(2000); // 线程休眠2秒，模拟任务执行
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
});

3. 等待任务完成

为了确保主线程等待线程池中所有任务完成，我们可以使用 CountDownLatch 或收集 Future 对象并调用它们的 get 方法。

使用 `CountDownLatch`

import java.util.concurrent.CountDownLatch;

int taskCount = 5; // 设定任务总数
CountDownLatch latch = new CountDownLatch(taskCount);

for (int i = 0; i < taskCount; i++) {
    threadPool.submit(() -> {
        try {
            // 模拟耗时任务
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            latch.countDown(); // 任务完成，计数器减一
        }
    });
}

try {
    latch.await(); // 等待直到计数器为0，即所有任务完成
} catch (InterruptedException e) {
    e.printStackTrace();
}

使用 `Future`

如果您不希望使用 CountDownLatch，可以将每个 Future 存储在列表中，然后调用其 get 方法：

import java.util.ArrayList;
import java.util.List;

List<Future<?>> futures = new ArrayList<>();

for (int i = 0; i < taskCount; i++) {
    Future<?> future = threadPool.submit(() -> {
        // 模拟耗时任务
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    });
    futures.add(future); // 将每个任务的 Future 存入列表
}

for (Future<?> future : futures) {
    try {
        future.get(); // 等待每个任务完成
    } catch (Exception e) {
        e.printStackTrace();
    }
}