【Java】已解决java.util.concurrent.RejectedExecutionException异常

屿小夏

发布于 2024-06-21 13:25:51

1680

发布于 2024-06-21 13:25:51

文章被收录于专栏：IT杂谈学习

已解决java.util.concurrent.RejectedExecutionException异常

一、问题背景

java.util.concurrent.RejectedExecutionException是Java并发编程中常见的一个异常，它通常发生在使用ExecutorService（如ThreadPoolExecutor）执行异步任务时，当任务提交到线程池但线程池无法处理新任务（比如，因为已经关闭了或者达到了其最大容量）时，就会抛出此异常。

在Web应用、后台服务或其他需要处理大量并发请求的系统中，经常可以看到此类异常的出现。下面，我们将探讨如何解决这一问题。

二、可能出错的原因

线程池已满且RejectedExecutionHandler的默认策略是抛出异常。
线程池已关闭，无法接受新任务。
提交的任务数量超过了线程池的最大处理能力。

三、错误代码示例

假设我们有一个固定大小的线程池，但没有正确处理拒绝策略：

import java.util.concurrent.*;  
  
public class ThreadPoolExample {  
    public static void main(String[] args) {  
        ExecutorService executor = Executors.newFixedThreadPool(10); // 创建一个固定大小为10的线程池  
  
        // 假设我们提交了大量的任务  
        for (int i = 0; i < Integer.MAX_VALUE; i++) {  
            executor.execute(() -> {  
                // 模拟一个长时间运行的任务  
                try {  
                    Thread.sleep(1000);  
                } catch (InterruptedException e) {  
                    e.printStackTrace();  
                }  
            });  
        }  
  
        // 这里并没有关闭线程池  
    }  
}

在这个例子中，因为循环是无限的，线程池很快就会饱和，并且由于没有设置自定义的拒绝策略，当线程池无法接受新任务时，就会抛出RejectedExecutionException。

四、正确代码示例

为了解决这个问题，我们可以采取以下措施：

合理地设置线程池的大小。
使用合适的队列来存储待处理的任务。
实现一个自定义的拒绝策略。

import java.util.concurrent.*;  
  
public class ThreadPoolExampleWithRejectionHandler {  
    public static void main(String[] args) {  
        // 创建一个固定大小为10的线程池，并使用一个无界队列（这里仅作为示例，通常不推荐使用无界队列）  
        ExecutorService executor = new ThreadPoolExecutor(  
                10, // 核心线程数  
                10, // 最大线程数  
                0L, TimeUnit.MILLISECONDS, // 空闲线程等待新任务的最长时间  
                new LinkedBlockingQueue<>(), // 使用无界队列  
                Executors.defaultThreadFactory(),  
                new ThreadPoolExecutor.CallerRunsPolicy() // 使用调用者运行策略作为拒绝策略  
        );  
  
        // 提交任务（注意这里应有一个合理的控制，防止无限制地提交任务）  
        // ...  
  
        // 当不再需要提交新任务时，优雅地关闭线程池  
        executor.shutdown(); // 开始关闭过程，不再接受新任务  
        try {  
            // 等待所有任务完成，或者超时等待  
            if (!executor.awaitTermination(60, TimeUnit.SECONDS)) {  
                // 超时了，可以取消未完成的任务  
                executor.shutdownNow(); // 尝试停止所有正在执行的任务，停止处理正在等待的任务，并返回等待执行的任务列表  
            }  
        } catch (InterruptedException ie) {  
            // 如果当前线程在等待过程中被中断，则重新中断线程池中的线程  
            executor.shutdownNow();  
            // 保存中断状态  
            Thread.currentThread().interrupt();  
        }  
    }  
}

在这个例子中，我们使用了ThreadPoolExecutor的构造函数来创建一个线程池，并指定了一个CallerRunsPolicy的拒绝策略。当线程池不能接受新任务时，它会尝试在调用execute的线程中运行该任务。