首页 > 编程语言 >Java异步判断线程池所有任务是否执行完成的方法

Java异步判断线程池所有任务是否执行完成的方法

时间:2024-07-11 15:19:19浏览次数:16  
标签:异步 Java 任务 Future 线程 完成 executorService 等待

1.使用ExecutorServiceCountDownLatch的方法示例

在Java中,当我们使用线程池(如ExecutorService)来执行异步任务时,常常需要知道所有任务是否都已经完成。ExecutorService接口提供了几种方式来处理这种情况,但最常用的是shutdown()awaitTermination()方法的组合,或者使用FutureCompletionService。这里我将提供一个使用ExecutorServiceCountDownLatch的示例,因为CountDownLatch提供了一种直观的方式来等待一组线程完成。

首先,我们定义几个任务,然后使用ExecutorService来异步执行它们,并使用CountDownLatch来等待所有任务完成。

import java.util.concurrent.*;  
  
public class ThreadPoolExample {  
  
    public static void main(String[] args) throws InterruptedException {  
        // 创建一个包含固定数量线程的线程池  
        ExecutorService executorService = Executors.newFixedThreadPool(4);  
  
        // 定义任务数量  
        int taskCount = 10;  
  
        // 使用CountDownLatch来等待所有任务完成  
        final CountDownLatch latch = new CountDownLatch(taskCount);  
  
        // 提交任务到线程池  
        for (int i = 0; i < taskCount; i++) {  
            int taskId = i;  
            executorService.submit(() -> {  
                // 模拟任务执行  
                try {  
                    Thread.sleep(1000); // 假设每个任务需要1秒  
                } catch (InterruptedException e) {  
                    Thread.currentThread().interrupt();  
                }  
                System.out.println("任务 " + taskId + " 完成");  
                // 每完成一个任务,计数减一  
                latch.countDown();  
            });  
        }  
  
        // 等待所有任务完成  
        System.out.println("等待所有任务完成...");  
        latch.await(); // 阻塞当前线程,直到latch的计数达到零  
        System.out.println("所有任务完成!");  
  
        // 关闭线程池  
        executorService.shutdown();  
  
        // 可选:等待线程池中的线程都执行完毕  
        try {  
            if (!executorService.awaitTermination(60, TimeUnit.SECONDS)) {  
                // 线程池没有在规定时间内关闭,则强制关闭  
                executorService.shutdownNow();  
            }  
        } catch (InterruptedException e) {  
            // 当前线程在等待过程中被中断  
            executorService.shutdownNow();  
            Thread.currentThread().interrupt();  
        }  
    }  
}

在这个例子中,我们首先创建了一个固定大小的线程池(这里使用4个线程)。然后,我们定义了一个CountDownLatch,其计数被初始化为任务的数量(这里为10)。对于每个任务,我们都向线程池提交了一个Runnable,其中包含了任务的执行逻辑和latch.countDown()调用,以确保每次任务完成时都会减少CountDownLatch的计数。

主线程通过调用latch.await()来等待,直到所有任务都调用了countDown()(即计数达到零),然后才能继续执行。这确保了主线程会等待所有任务完成后再继续。

最后,我们关闭了线程池,并通过调用awaitTermination()来可选地等待线程池中的所有线程都执行完毕。如果线程池没有在指定时间内关闭,则调用shutdownNow()来尝试立即停止所有正在执行的任务。

这个示例提供了处理异步任务并等待它们完成的一种有效方式,适用于需要等待所有任务完成再继续的场景。

2.使用ExecutorServiceinvokeAll方法和Future列表的方法示例

除了使用CountDownLatch之外,还有其他方法可以判断线程池中的所有任务是否执行完成。以下是一个使用ExecutorServiceinvokeAll方法和Future列表的示例,这种方法适用于我们有一组已知的任务(Callable)需要并行执行,并且我们需要等待所有任务完成并获取它们的结果。

import java.util.ArrayList;  
import java.util.List;  
import java.util.concurrent.*;  
  
public class ThreadPoolFutureExample {  
  
    public static void main(String[] args) throws InterruptedException, ExecutionException {  
        // 创建一个包含固定数量线程的线程池  
        ExecutorService executorService = Executors.newFixedThreadPool(4);  
  
        // 创建一个Callable任务列表  
        List<Callable<String>> tasks = new ArrayList<>();  
        for (int i = 0; i < 10; i++) {  
            final int taskId = i;  
            tasks.add(() -> {  
                // 模拟任务执行  
                Thread.sleep(1000); // 假设每个任务需要1秒  
                return "任务 " + taskId + " 完成";  
            });  
        }  
  
        // 使用invokeAll提交所有任务,这将返回一个Future列表  
        List<Future<String>> futures = executorService.invokeAll(tasks);  
  
        // 遍历Future列表,获取每个任务的结果  
        for (Future<String> future : futures) {  
            // get()会阻塞,直到对应的任务完成  
            System.out.println(future.get());  
        }  
  
        // 关闭线程池  
        executorService.shutdown();  
  
        // 可选:等待线程池中的线程都执行完毕  
        try {  
            if (!executorService.awaitTermination(60, TimeUnit.SECONDS)) {  
                // 线程池没有在规定时间内关闭,则强制关闭  
                executorService.shutdownNow();  
            }  
        } catch (InterruptedException e) {  
            // 当前线程在等待过程中被中断  
            executorService.shutdownNow();  
            Thread.currentThread().interrupt();  
        }  
    }  
}  
  
// 注意:这里使用了Lambda表达式和方法引用来简化Callable的创建  
// 实际使用中,你可能需要实现Callable接口或使用匿名内部类

在这个例子中,我们创建了一个ExecutorService和一个Callable任务列表。每个Callable任务都会返回一个字符串,表示任务完成的信息。我们使用invokeAll方法提交了所有任务,并立即获得了一个Future列表,每个Future都代表了一个任务的执行结果。

然后,我们遍历这个Future列表,并对每个Future调用get()方法。get()方法会阻塞当前线程,直到对应的任务完成并返回结果。这样,我们就能确保在继续执行之前,所有任务都已经完成。

最后,我们关闭了线程池,并等待所有线程都执行完毕(或超时后强制关闭)。

请注意,虽然这个示例使用了CallableFuture,但它并没有直接提供一个“是否所有任务都已完成”的布尔值。然而,通过遍历Future列表并调用get(),我们实际上已经达到了等待所有任务完成的效果。如果我们只需要知道是否所有任务都已开始执行(而不是等待它们完成),那么我们可能需要采用不同的策略,比如使用execute方法结合其他同步机制(如CountDownLatch)。

3.使用ExecutorService来异步执行多个Callable任务方法示例

以下是一个详细完整的代码示例,该示例使用了ExecutorService来异步执行多个Callable任务,并通过遍历Future列表来等待所有任务完成并获取它们的结果。

import java.util.ArrayList;  
import java.util.List;  
import java.util.concurrent.*;  
  
public class ThreadPoolFutureCompleteExample {  
  
    public static void main(String[] args) {  
        // 创建一个包含固定数量线程的线程池  
        ExecutorService executorService = Executors.newFixedThreadPool(4);  
  
        // 创建一个Callable任务列表  
        List<Callable<String>> tasks = new ArrayList<>();  
        for (int i = 0; i < 10; i++) {  
            final int taskId = i;  
            tasks.add(new Callable<String>() {  
                @Override  
                public String call() throws Exception {  
                    // 模拟任务执行  
                    TimeUnit.SECONDS.sleep(1); // 假设每个任务需要1秒  
                    return "任务 " + taskId + " 完成";  
                }  
            });  
  
            // 或者使用Lambda表达式(如果你使用的是Java 8或更高版本)  
            // tasks.add(() -> {  
            //     TimeUnit.SECONDS.sleep(1);  
            //     return "任务 " + taskId + " 完成";  
            // });  
        }  
  
        try {  
            // 使用invokeAll提交所有任务,这将返回一个Future列表  
            List<Future<String>> futures = executorService.invokeAll(tasks);  
  
            // 遍历Future列表,获取每个任务的结果  
            for (Future<String> future : futures) {  
                // get()会阻塞,直到对应的任务完成  
                System.out.println(future.get());  
            }  
  
            // 关闭线程池  
            executorService.shutdown();  
  
            // 等待线程池中的所有线程都执行完毕(可选)  
            // 注意:由于我们已经调用了invokeAll并等待了所有Future的完成,这一步通常是多余的  
            // 但为了完整性,我还是展示了如何等待线程池关闭  
            boolean terminated = executorService.awaitTermination(60, TimeUnit.SECONDS);  
            if (!terminated) {  
                // 如果线程池没有在规定时间内关闭,则强制关闭  
                System.err.println("线程池没有在规定时间内关闭,尝试强制关闭...");  
                executorService.shutdownNow();  
                // 注意:shutdownNow()不保证已经提交的任务会被取消  
                // 它会尝试停止正在执行的任务,但已经开始执行的任务可能无法被中断  
            }  
  
        } catch (InterruptedException | ExecutionException e) {  
            // 处理异常  
            e.printStackTrace();  
  
            // 如果当前线程在等待过程中被中断,尝试关闭线程池  
            if (!executorService.isShutdown()) {  
                executorService.shutdownNow();  
            }  
  
            // 根据需要,可能还需要重新设置中断状态  
            Thread.currentThread().interrupt();  
        }  
    }  
}

在这个示例中,我使用了传统的匿名内部类来创建Callable任务(同时也提供了Lambda表达式的注释),以便与各种Java版本兼容。然而,如果我们正在使用Java 8或更高版本,我强烈推荐我们使用Lambda表达式来简化代码。

请注意,invokeAll方法会阻塞调用它的线程,直到所有任务都完成,或者直到等待超时(如果我们提供了超时时间)。但是,在这个示例中,我们没有为invokeAll提供超时时间,因此它会一直等待,直到所有任务都完成。

另外,请注意,在catch块中,如果捕获到InterruptedException,我们检查了线程池是否已经被关闭(使用isShutdown方法)。如果没有,我们调用shutdownNow方法来尝试关闭线程池并停止正在执行的任务。然而,需要注意的是,shutdownNow方法并不保证能够停止所有已经开始执行的任务,因为某些任务可能无法被中断。

最后,如果在捕获到InterruptedException后,我们确定当前线程需要被重新中断(比如,我们在一个循环中等待某个条件,而中断是用来退出循环的),那么我们应该调用Thread.currentThread().interrupt()来重新设置中断状态。在这个示例中,我们没有这样做,因为main方法不需要重新中断。但是,在更复杂的场景中,这可能是必要的。

标签:异步,Java,任务,Future,线程,完成,executorService,等待
From: https://www.cnblogs.com/TS86/p/18296282

相关文章

  • 【java】实现sse调用websocket接口,忽略wss证书并控制sse吐字速度
    maven<dependency><groupId>org.java-websocket</groupId><artifactId>Java-WebSocket</artifactId><version>1.5.3</version></dependency>AsyncConfigpackag......
  • Linux设备驱动器 之二 线程同步第二篇
    Linux设备驱动器之二线程同步第二篇mutex数据结构LinuxAPIs在Linux驱动器中的应用NXPfreescale系列QSPI驱动器变量定义初始化存取数据semaphore数据结构LinuxAPIs在Linux驱动器中的应用ELAN的Uxxx系列驱动器变量定义初始化同步操作mutex数据结构stru......
  • 异步编程秘籍:一探std::future的魔法,让代码飞起来
    ......
  • JAVA生成验证码(字母+数字)
    /***验证码、邀请码工具类*/@ComponentpublicclassVerificationCodeUtil{/***生成验证码:字母+数字*@paramlength验证码长度*@return验证码字符串*/publicstaticStringgetVerificationCode(intlength){Str......
  • 基于java+ssm+jsp的连锁干洗店后台管理系统+vue录像毕业设计实战项目+lw文档
    前言......
  • 基于javaweb jsp ssm汽车服务商城系统设计与实现+vue录像(源码+lw+部署文档+讲解等)
    前言......
  • Java面试题系列 - 第9天
    题目:深入探讨Java中的设计模式及其应用场景背景说明:设计模式是软件工程中解决问题的常见方案,它们提供了经过验证的模板,帮助开发者解决在软件设计过程中遇到的特定问题。在Java中,熟悉并正确应用设计模式能够显著提升代码的可读性、可维护性和可扩展性。问题要求:解释设计模式......
  • 最开始学java时写的文档
    一,java基础1.常用cmd窗口操作E:切换到e盘dir查看当前路径下的内容cd目录进入单级目录cd..退回到上一级目录cd目录1\目录2\进入多级目录cls清屏但不会退出当前目录exit关闭cmd窗口想要在任意目录下都可以打开指定软件可以把软件路径配置到环境变量中2.jav......
  • 初识Java
    一、java语言概述1.java是什么java是一门面向对象的编程语言,它不仅吸收了C++的各种优点,还摒弃了其难以理解的多继承、指针等概念,因此,java具有功能强大又简单易用的两个特制,作为面向对象的编程语言的代表,极好地实现了面向对象理论,允许程序员以优雅的思维方式进行复杂的编程......
  • 初学java
    Java作为一种广泛使用的高级编程语言,它以其跨平台性、面向对象设计、健壮的标准库、自动内存管理和安全性等优点而受到青睐。这些特性使得Java适用于多种计算平台,并且能够构建大型、复杂的应用程序,同时保证程序的稳定性和安全性。其也是目前使用最多的编程语言。Java有Java.ES,Ja......