探索 Java 中的线程池自定义技巧：高效、灵活地管理并发任务！

在并发编程中，线程池无疑是提升性能、合理使用系统资源的关键工具之一。Java 提供了 Executors 工具类来简化线程池的创建，但如果不根据实际需求调整配置，默认线程池可能会带来性能瓶颈，甚至导致资源耗尽的问题。因此，了解如何根据实际场景自定义线程池的策略至关重要。今天，我们就来学习如何灵活高效地自定义线程池，合理管理并发任务。

---

为什么需要自定义线程池？

Executors.newFixedThreadPool() 和 Executors.newCachedThreadPool() 提供了便捷的线程池创建方式，但这些默认实现可能在一些场景下无法满足需求。例如：

• 性能问题：如果任务密集而线程数不足，会出现任务堆积。
• 资源浪费：线程数过多会导致 CPU 切换频繁，影响系统响应。
• 控制需求：需要限制等待任务队列的大小，避免内存爆炸。

自定义线程池可以根据实际场景进行调整，有效地提升系统性能。

---

1. 基本步骤：使用 ThreadPoolExecutor 构建线程池

在 Java 中，ThreadPoolExecutor 是最核心的线程池实现类。使用它自定义线程池时，可以手动指定各项参数，如核心线程数、最大线程数、队列容量等。基本构造方法如下：

ThreadPoolExecutor threadPool = new ThreadPoolExecutor(
        corePoolSize,      // 核心线程数
        maximumPoolSize,   // 最大线程数
        keepAliveTime,     // 线程空闲保持时间
        TimeUnit.SECONDS,  // 时间单位
        new LinkedBlockingQueue<>(queueCapacity),  // 任务队列
        new ThreadFactory() {  // 线程工厂
            @Override
            public Thread newThread(Runnable r) {
                return new Thread(r, "custom-thread");
            }
        },
        new ThreadPoolExecutor.CallerRunsPolicy()  // 拒绝策略
);

2. 合理设置核心线程数与最大线程数

• 核心线程数 (corePoolSize)：核心线程数决定了线程池在没有空闲线程的情况下立即启动多少线程。一般来说，可以根据任务的 CPU 密集度来调整。如果任务是 IO 密集型，可以设置更多的线程数；如果是 CPU 密集型，则保持较少的线程数（通常为 CPU 核数的 1-2 倍）。

• 最大线程数 (maximumPoolSize)：当任务数量超过核心线程数时，线程池会继续创建新线程，直到达到最大线程数。在高负载的情况下，可以将其设为核心线程数的 2 倍或更高，但需谨慎，避免过多线程争夺 CPU 资源。

实例：创建一个适合 IO 密集型任务的线程池

int corePoolSize = 10;
int maximumPoolSize = 20;
long keepAliveTime = 60;
ThreadPoolExecutor ioThreadPool = new ThreadPoolExecutor(
        corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.SECONDS,
        new LinkedBlockingQueue<>(100)
);

---

3. 选择合适的任务队列

任务队列决定了多余任务的存储方式。以下是常用的几种任务队列及其应用场景：

• SynchronousQueue：没有容量，每提交一个任务必须有空闲线程来执行。适合需要动态调整线程数量的情况。
• LinkedBlockingQueue：无界队列，任务较多时可以避免因队列已满而拒绝任务，适合任务处理速度快、任务数量大的情况。
• ArrayBlockingQueue：有界队列，设定固定容量，适合需要控制任务数量，防止内存溢出的场景。

实例：使用 ArrayBlockingQueue 防止任务过多时占用过多内存

ThreadPoolExecutor boundedThreadPool = new ThreadPoolExecutor(
        corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.SECONDS,
        new ArrayBlockingQueue<>(50)
);

---

4. 巧妙利用线程工厂自定义线程

通过自定义 ThreadFactory，可以为线程池内的线程设置特殊的属性，比如线程命名、优先级、是否为守护线程等。这有助于调试和识别不同线程池的线程。

ThreadFactory customThreadFactory = new ThreadFactory() {
    private final AtomicInteger threadNumber = new AtomicInteger(1);
    
    @Override
    public Thread newThread(Runnable r) {
        Thread thread = new Thread(r, "custom-thread-" + threadNumber.getAndIncrement());
        thread.setPriority(Thread.NORM_PRIORITY);  // 设置优先级
        thread.setDaemon(false);                   // 设置是否为守护线程
        return thread;
    }
};

ThreadPoolExecutor threadPool = new ThreadPoolExecutor(
        corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.SECONDS,
        new LinkedBlockingQueue<>(),
        customThreadFactory
);

---

5. 选择合适的拒绝策略

当线程池的任务队列已满，且线程数已达到最大值时，新的任务会被拒绝处理。此时可以选择合适的拒绝策略，来保证程序的稳定性：

• AbortPolicy：直接抛出异常，这是默认策略。
• CallerRunsPolicy：由调用线程（通常是主线程）执行该任务，以减缓任务提交速度。
• DiscardPolicy：直接丢弃任务，不抛出异常。
• DiscardOldestPolicy：丢弃队列中最早的任务，然后尝试重新提交任务。

实例：使用 CallerRunsPolicy 降低拒绝任务的发生

ThreadPoolExecutor resilientThreadPool = new ThreadPoolExecutor(
        corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.SECONDS,
        new LinkedBlockingQueue<>(100),
        new ThreadPoolExecutor.CallerRunsPolicy()
);

这种策略可以有效防止任务被抛弃或异常终止，使系统更加稳健。

---

6. 监控线程池状态

自定义线程池后，监控其状态也非常重要。可以定期检查线程池的活跃线程数、完成任务数等信息，及时调整参数。示例代码如下：

System.out.println("Active Threads: " + threadPool.getActiveCount());
System.out.println("Completed Tasks: " + threadPool.getCompletedTaskCount());
System.out.println("Queue Size: " + threadPool.getQueue().size());

也可以使用监控工具，比如 Spring Actuator 和 Prometheus，实时监控线程池状态。

---

7. 合理设置线程的超时释放机制

对于非核心线程，ThreadPoolExecutor 提供了 keepAliveTime 配置，当线程空闲时间超过指定时长时会自动释放。这有助于避免线程池在任务高峰期后一直占用资源：

threadPool.allowCoreThreadTimeOut(true);

设置 allowCoreThreadTimeOut(true) 后，核心线程在空闲超过 keepAliveTime 时也会被释放。

---

结语

自定义线程池在高并发场景中可以显著提升应用的响应速度和稳定性。通过合理设置核心线程数、任务队列、拒绝策略和线程工厂等参数，可以更好地控制线程池行为，让系统更加灵活、可靠。希望这些技巧能帮助你打造更加高效的线程池，实现并发任务的合理管理！

推荐阅读文章

• 由 Spring 静态注入引发的一个线上T0级别事故（真的以后得避坑）
• 如何理解 HTTP 是无状态的，以及它与 Cookie 和 Session 之间的联系
• HTTP、HTTPS、Cookie 和 Session 之间的关系
• 什么是 Cookie？简单介绍与使用方法
• 什么是 Session？如何应用？
• 使用 Spring 框架构建 MVC 应用程序：初学者教程
• 有缺陷的 Java 代码：Java 开发人员最常犯的 10 大错误
• 如何理解应用 Java 多线程与并发编程？
• 把握Java泛型的艺术：协变、逆变与不可变性一网打尽
• Java Spring 中常用的 @PostConstruct 注解使用总结
• 如何理解线程安全这个概念？
• 理解 Java 桥接方法
• Spring 整合嵌入式 Tomcat 容器
• Tomcat 如何加载 SpringMVC 组件
• “在什么情况下类需要实现 Serializable，什么情况下又不需要（一）？”
• “避免序列化灾难：掌握实现 Serializable 的真相！（二）”
• 如何自定义一个自己的 Spring Boot Starter 组件（从入门到实践）
• 解密 Redis：如何通过 IO 多路复用征服高并发挑战！
• 线程 vs 虚拟线程：深入理解及区别
• 深度解读 JDK 8、JDK 11、JDK 17 和 JDK 21 的区别
• 10大程序员提升代码优雅度的必杀技，瞬间让你成为团队宠儿！
• “打破重复代码的魔咒：使用 Function 接口在 Java 8 中实现优雅重构！”
• Java 中消除 If-else 技巧总结
• 线程池的核心参数配置(仅供参考)
• 【人工智能】聊聊Transformer，深度学习的一股清流(13)
• Java 枚举的几个常用技巧，你可以试着用用