ThreadPoolExecutor
的拒绝策略(RejectedExecutionHandler
)在线程池无法处理新的任务时会被触发。下面介绍 Java 中 ThreadPoolExecutor
提供的几种内置拒绝策略,并展示如何自定义拒绝策略。
1. 内置拒绝策略
-
AbortPolicy (默认策略)
- 当任务无法被接受时,抛出
RejectedExecutionException
异常。
new ThreadPoolExecutor.AbortPolicy()
- 当任务无法被接受时,抛出
-
CallerRunsPolicy
- 当任务无法被接受时,调用线程池的线程会执行该任务。这种策略提供了一种“回退”机制,从而降低新任务提交速度。
new ThreadPoolExecutor.CallerRunsPolicy()
-
DiscardPolicy
- 当任务无法被接受时,直接丢弃任务,不做任何处理。
new ThreadPoolExecutor.DiscardPolicy()
-
DiscardOldestPolicy
- 当任务无法被接受时,丢弃工作队列中最旧的未处理任务,然后重新尝试提交新任务。
new ThreadPoolExecutor.DiscardOldestPolicy()
2. 自定义拒绝策略
可以通过实现 RejectedExecutionHandler
接口来创建自定义的拒绝策略。例如,记录被拒绝任务的日志或进行其他处理。
import java.util.concurrent.*;
public class CustomRejectedExecutionHandler implements RejectedExecutionHandler {
@Override
public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
System.out.println("Task " + r.toString() + " rejected");
}
public static void main(String[] args) {
ThreadPoolExecutor executor = new ThreadPoolExecutor(
2, // corePoolSize
4, // maximumPoolSize
60, // keepAliveTime
TimeUnit.SECONDS, // timeUnit
new ArrayBlockingQueue<>(2), // workQueue
Executors.defaultThreadFactory(), // threadFactory
new CustomRejectedExecutionHandler() // rejectedExecutionHandler
);
for (int i = 0; i < 10; i++) {
executor.execute(new Task(i));
}
executor.shutdown();
}
static class Task implements Runnable {
private final int taskId;
public Task(int taskId) {
this.taskId = taskId;
}
@Override
public void run() {
System.out.println("Executing task " + taskId);
try {
Thread.sleep(2000); // Simulate task execution time
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
}
System.out.println("Task " + taskId + " completed");
}
@Override
public String toString() {
return "Task{" + "taskId=" + taskId + '}';
}
}
}
-
CustomRejectedExecutionHandler:自定义拒绝策略,实现
RejectedExecutionHandler
接口。在任务被拒绝时,打印被拒绝任务的信息。 -
ThreadPoolExecutor 配置:
- 核心线程数为 2。
- 最大线程数为 4。
- 工作队列大小为 2。
- 自定义拒绝策略。
-
任务提交:提交了 10 个任务。前两个任务由核心线程处理,接下来的两个任务进入工作队列。由于工作队列和线程池达到最大容量,后续任务将触发自定义拒绝策略,打印被拒绝任务的信息。
3. 总结
通过使用内置拒绝策略或实现自定义拒绝策略,可以灵活处理线程池无法接受新任务的情况,从而提高系统的健壮性和可控性。选择哪种策略取决于具体的应用场景和需求。