Java基础：线程池拒绝策略详解

在Java中，线程池是一个非常实用的工具，可以有效管理线程的创建和销毁，提高系统的性能和资源利用率。线程池的一个重要特性是拒绝策略（RejectedExecutionHandler），当线程池无法接受新的任务时，会触发拒绝策略。本文将详细介绍线程池拒绝策略的几种类型，并给出相应的代码示例。

线程池拒绝策略概述

线程池拒绝策略主要分为以下几种：

1. AbortPolicy（默认策略）：直接抛出RejectedExecutionException异常。
2. CallerRunsPolicy：只要线程池未关闭，就由调用者线程执行任务。
3. DiscardPolicy：默默丢弃被拒绝的任务，不抛出异常。
4. DiscardOldestPolicy：丢弃线程池中最旧的未处理的任务，然后尝试提交新任务。

代码示例

下面是一个简单的例子，展示了如何创建线程池并使用不同的拒绝策略。

import java.util.concurrent.*;

public class ThreadPoolRejectPolicyExample {
    public static void main(String[] args) {
        // 创建线程池
        int corePoolSize = 2; // 核心线程数
        int maximumPoolSize = 4; // 最大线程数
        long keepAliveTime = 10; // 线程最大空闲时间
        TimeUnit unit = TimeUnit.SECONDS; // 时间单位
        BlockingQueue<Runnable> workQueue = new LinkedBlockingQueue<>(2); // 工作队列，最大存放任务为2

        // 使用AbortPolicy拒绝策略
        ThreadPoolExecutor executor = new ThreadPoolExecutor(
                corePoolSize,
                maximumPoolSize,
                keepAliveTime,
                unit,
                workQueue,
                new ThreadPoolExecutor.AbortPolicy() // 拒绝策略
        );

        // 提交10个任务
        for (int i = 0; i < 10; i++) {
            final int taskId = i;
            executor.execute(() -> {
                System.out.println("执行任务 " + taskId + "，线程名：" + Thread.currentThread().getName());
                try {
                    Thread.sleep(1000); // 模拟任务执行时间
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            });
        }

        executor.shutdown();
        try {
            executor.awaitTermination(1, TimeUnit.MINUTES);
        } catch (InterruptedException e) {
            System.err.println("任务被中断");
        }
    }
}

测试不同的拒绝策略

在上面的示例中，我们使用了AbortPolicy作为拒绝策略。您可以轻松地替换new ThreadPoolExecutor.AbortPolicy()为其他策略，如CallerRunsPolicy、DiscardPolicy和DiscardOldestPolicy。

这是一个示例，展示了如何使用CallerRunsPolicy：

ThreadPoolExecutor executor = new ThreadPoolExecutor(
        corePoolSize,
        maximumPoolSize,
        keepAliveTime,
        unit,
        workQueue,
        new ThreadPoolExecutor.CallerRunsPolicy() // 拒绝策略
);

总结

通过使用不同的拒绝策略，您可以灵活地控制线程池对任务的处理方式。AbortPolicy适合需要立即反馈的场景，CallerRunsPolicy适合需要让调用者亲自处理某些任务的场景，而DiscardPolicy和DiscardOldestPolicy则适用于可以容忍丢失部分任务的场合。选择合适的拒绝策略是性能优化的重要环节。在实际应用中，需要根据具体的业务需求进行合理的选择。