答案是自定义拒绝策略可应对线程池过载。当线程池关闭或队列满且线程数达上限时,触发拒绝策略;Java 提供 AbortPolicy、CallerRunsPolicy、DiscardPolicy 和 DiscardOldestPolicy 四种内置策略;通过实现 RejectedExecutionHandler 接口可自定义行为,如记录日志、持久化任务;创建 ThreadPoolExecutor 时传入自定义处理器即可生效;应根据业务需求选择策略,保障系统稳定性与数据完整性。
当线程池中的任务数量超过其处理能力时,ThreadPoolExecutor 会触发拒绝策略。默认情况下,它会抛出 RejectedExecutionException 异常,但我们可以自定义拒绝行为来更好地应对高负载场景。实现自定义拒绝策略的关键是实现 RejectedExecutionHandler 接口。
以下情况会触发拒绝策略:
此时新提交的任务无法被处理,就会交给拒绝处理器。
Java 提供了四种常见的内置策略,均实现了 RejectedExecutionHandler:
通过实现 RejectedExecutionHandler 接口的 rejectedE
xecution(Runnable r, ThreadPoolExecutor executor) 方法,可以完全控制拒绝时的行为。
例如,记录日志并保存任务以便后续重试:
public class CustomRejectHandler implements RejectedExecutionHandler {
@Override
public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
System.err.println("任务被拒绝: " + r.toString());
System.err.println("当前线程池状态 - 线程数: " + executor.getPoolSize() +
", 队列大小: " + executor.getQueue().size());
// 可以将任务写入磁盘或消息队列用于恢复
if (r instanceof Task) {
((Task) r).saveToDisk(); // 假设 Task 有持久化方法
}
}}
配置自定义拒绝策略
在创建 ThreadPoolExecutor 时传入自定义处理器即可:
ThreadPoolExecutor executor = new ThreadPoolExecutor(
2, // corePoolSize
4, // maximumPoolSize
60L, // keepAliveTime
TimeUnit.SECONDS,
new ArrayBlockingQueue<>(10), // queue
new CustomRejectHandler() // handler
);
也可以使用内置策略之一作为构造参数:
new ThreadPoolExecutor(..., new ThreadPoolExecutor.CallerRunsPolicy())
实际应用场景建议
选择哪种策略取决于业务需求:
基本上就这些。关键是根据系统负载特征和容错能力选择或设计合适的拒绝方式,避免任务丢失或服务雪崩。