请详细解释Java中的线程池(ThreadPoolExecutor)的工作原理,并说明如何自定义线程池的拒绝策略

线程池(ThreadPoolExecutor)的工作原理

线程池是一种用于管理和复用线程的机制,可以有效地控制并发线程的数量,减少线程创建和销毁的开销。ThreadPoolExecutor是Java中实现线程池的核心类。

核心组件

  1. 核心线程数(corePoolSize):线程池中保持的最小线程数,即使这些线程处于空闲状态也不会被销毁。
  2. 最大线程数(maximumPoolSize):线程池中允许的最大线程数。
  3. 存活时间(keepAliveTime):当线程数超过核心线程数时,多余的空闲线程在终止前等待新任务的最长时间。
  4. 工作队列(workQueue):用于保存等待执行任务的队列。
  5. 拒绝策略(RejectedExecutionHandler):当任务无法被接受时的处理策略。

工作流程

  1. 当一个任务提交到线程池时,如果当前线程数小于核心线程数,线程池会创建一个新的线程来执行任务。
  2. 如果当前线程数已经达到核心线程数,任务会被放入工作队列中等待执行。
  3. 如果工作队列已满,但当前线程数小于最大线程数,线程池会创建一个新的线程来执行任务。
  4. 如果当前线程数已经达到最大线程数,并且工作队列已满,新提交的任务将根据拒绝策略进行处理。

自定义线程池的拒绝策略

ThreadPoolExecutor提供了四种默认的拒绝策略:

  1. AbortPolicy :默认策略,抛出RejectedExecutionException异常。
  2. CallerRunsPolicy:由调用线程执行任务。
  3. DiscardPolicy:直接丢弃任务,不抛出异常。
  4. DiscardOldestPolicy:丢弃工作队列中最老的任务,然后重新尝试执行新任务。

可以通过实现RejectedExecutionHandler接口来自定义拒绝策略。以下是一个自定义拒绝策略的示例:

java 复制代码
import java.util.concurrent.RejectedExecutionHandler; 
import java.util.concurrent.ThreadPoolExecutor; 

public class CustomRejectionHandler implements RejectedExecutionHandler {
    @Override
    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
        System.out.println("Task  " + r.toString()  + " rejected from " + executor.toString()); 
        // 可以在这里添加自定义的处理逻辑,例如记录日志、发送通知等
    }
}

使用自定义拒绝策略的示例:

ini 复制代码
import java.util.concurrent.ArrayBlockingQueue; 
import java.util.concurrent.ThreadPoolExecutor; 
import java.util.concurrent.TimeUnit; 

public class ThreadPoolExample {
    public static void main(String[] args) {
        int corePoolSize = 2;
        int maximumPoolSize = 4;
        long keepAliveTime = 10;
        TimeUnit unit = TimeUnit.SECONDS;
        ArrayBlockingQueue<Runnable> workQueue = new ArrayBlockingQueue<>(2);
        CustomRejectionHandler rejectionHandler = new CustomRejectionHandler();

        ThreadPoolExecutor executor = new ThreadPoolExecutor(
            corePoolSize,
            maximumPoolSize,
            keepAliveTime,
            unit,
            workQueue,
            rejectionHandler
        );

        for (int i = 0; i < 6; i++) {
            final int taskNumber = i;
            executor.submit(()  -> {
                System.out.println("Task  " + taskNumber + " is running.");
                try {
                    Thread.sleep(2000); 
                } catch (InterruptedException e) {
                    e.printStackTrace(); 
                }
                System.out.println("Task  " + taskNumber + " is completed.");
            });
        }

        executor.shutdown(); 
    }
}

在这个示例中,我们创建了一个自定义的拒绝策略CustomRejectionHandler,并在创建ThreadPoolExecutor时将其设置为拒绝策略。当任务无法被接受时,自定义拒绝策略会打印一条消息。

总结

线程池通过管理和复用线程,提高了系统的性能和稳定性。ThreadPoolExecutor提供了灵活的配置选项和自定义拒绝策略,可以根据具体需求进行调整。

相关推荐
77tian4 分钟前
设计模式的原理及深入解析
java·开发语言·单例模式·设计模式·代理模式·享元模式·原型模式
会飞的架狗师40 分钟前
【Spring Cloud Gateway】Nacos整合遇坑记:503 Service Unavailable
java·开发语言
重生之后端学习2 小时前
02-前端Web开发(JS+Vue+Ajax)
java·开发语言·前端·javascript·vue.js
字节源流8 小时前
关于maven的依赖下不下来的问题
java·maven
pjx9878 小时前
服务间的“握手”:OpenFeign声明式调用与客户端负载均衡
java·运维·spring·负载均衡
prinrf('千寻)9 小时前
MyBatis-Plus 的 updateById 方法不更新 null 值属性的问题
java·开发语言·mybatis
老华带你飞9 小时前
实习记录小程序|基于SSM+Vue的实习记录小程序设计与实现(源码+数据库+文档)
java·数据库·spring boot·小程序·论文·毕设·实习记录小程序
在未来等你9 小时前
互联网大厂Java求职面试:AI与大模型应用集成及云原生挑战
java·微服务·ai·kubernetes·大模型·embedding·spring ai
源码技术栈10 小时前
SaaS基于云计算、大数据的Java云HIS平台信息化系统源码
java·大数据·云计算·云his·his系统·云医院·区域his
编程、小哥哥10 小时前
互联网大厂Java面试:从Spring Boot到微服务架构的技术深挖
java·spring boot·redis·微服务·prometheus·面试技巧