前言
線程池,相信很多人都有用過,沒用過相信的也有學習過。但是,線程池的拒絕策略,相信知道的人會少許多。
四種線程池拒絕策略
當線程池的任務緩存隊列已滿並且線程池中的線程數目達到maximumPoolSize時,如果還有任務到來就會采取任務拒絕策略,通常有以下四種策略:
ThreadPoolExecutor.AbortPolicy:丟棄任務並拋出RejectedExecutionException異常。ThreadPoolExecutor.DiscardPolicy:丟棄任務,但是不拋出異常。ThreadPoolExecutor.DiscardOldestPolicy:丟棄隊列最前面的任務,然后重新提交被拒絕的任務ThreadPoolExecutor.CallerRunsPolicy:由調用線程(提交任務的線程)處理該任務
線程池默認的拒絕策略
既然有四種拒絕策略可以選擇,那么線程池的默認拒絕策略是什么呢?查看java.util.concurrent.ThreadPoolExecutor類的源碼,我們可以看到:
/** * The default rejected execution handler */private static final RejectedExecutionHandler defaultHandler = new AbortPolicy();
線程池的默認拒絕策略為AbortPolicy,即丟棄任務並拋出RejectedExecutionException異常。我們可以通過代碼來驗證這一點,現有如下代碼:
public class ThreadPoolTest { public static void main(String[] args) { BlockingQueue queue = new ArrayBlockingQueue<>(100); ThreadFactory factory = r -> new Thread(r, "test-thread-pool"); ThreadPoolExecutor executor = new ThreadPoolExecutor(5, 5, 0L, TimeUnit.SECONDS, queue, factory); while (true) { executor.submit(() -> { try { System.out.println(queue.size()); Thread.sleep(10000); } catch (InterruptedException e) { e.printStackTrace(); } }); } }}
這里是一個默認的線程池,沒有設置拒絕策略,設置了最大線程隊列是100。運行代碼:

結果是符合預期的,這也證明了線程池的默認拒絕策略是ThreadPoolExecutor.AbortPolicy:丟棄任務並拋出RejectedExecutionException異常。
設置線程池拒絕策略
如果我們想要根據實際業務場景需要,設置其他的線程池拒絕策略,可以通過ThreadPoolExecutor重載的構造方法進行設置:

現在的開發中,相信大家都有使用spring,其實我們也可以通過spring提供的org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor構建線程池。如下:

@Configurationpublic class TaskExecutorConfig implements AsyncConfigurer { /** * Set the ThreadPoolExecutor's core pool size. */ private static final int CORE_POOL_SIZE = 5; /** * Set the ThreadPoolExecutor's maximum pool size. */ private static final int MAX_POOL_SIZE = 5; /** * Set the capacity for the ThreadPoolExecutor's BlockingQueue. */ private static final int QUEUE_CAPACITY = 1000; /** * 通過重寫getAsyncExecutor方法,制定默認的任務執行由該方法產生 *
* 配置類實現AsyncConfigurer接口並重寫getAsyncExcutor方法,並返回一個ThreadPoolTaskExevutor * 這樣我們就獲得了一個基於線程池的TaskExecutor */ @Override public Executor getAsyncExecutor() { ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor(); taskExecutor.setCorePoolSize(CORE_POOL_SIZE); taskExecutor.setMaxPoolSize(MAX_POOL_SIZE); taskExecutor.setQueueCapacity(QUEUE_CAPACITY); taskExecutor.initialize(); taskExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.AbortPolicy()); return taskExecutor; }}
通過ThreadPoolTaskExecutor的setRejectedExecutionHandler設置拒絕策略即可。
拒絕策略場景分析
AbortPolicy
ThreadPoolExecutor.AbortPolicy:丟棄任務並拋出RejectedExecutionException異常。
A handler for rejected tasks that throws a {@code RejectedExecutionException}.
這是線程池默認的拒絕策略,在任務不能再提交的時候,拋出異常,及時反饋程序運行狀態。如果是比較關鍵的業務,推薦使用此拒絕策略,這樣子在系統不能承載更大的並發量的時候,能夠及時的通過異常發現。
DiscardPolicy
ThreadPoolExecutor.DiscardPolicy:丟棄任務,但是不拋出異常。如果線程隊列已滿,則后續提交的任務都會被丟棄,且是靜默丟棄。
A handler for rejected tasks that silently discards therejected task.
使用此策略,可能會使我們無法發現系統的異常狀態。建議是一些無關緊要的業務采用此策略。例如,本人的博客網站統計閱讀量就是采用的這種拒絕策略。
DiscardOldestPolicy
ThreadPoolExecutor.DiscardOldestPolicy:丟棄隊列最前面的任務,然后重新提交被拒絕的任務。
A handler for rejected tasks that discards the oldest unhandled request and then retries {@code execute}, unless the executor is shut down, in which case the task is discarded.
此拒絕策略,是一種喜新厭舊的拒絕策略。是否要采用此種拒絕策略,還得根據實際業務是否允許丟棄老任務來認真衡量。
CallerRunsPolicy
ThreadPoolExecutor.CallerRunsPolicy:由調用線程處理該任務
A handler for rejected tasks that runs the rejected task directly in the calling thread of the {@code execute} method, unless the executor has been shut down, in which case the task is discarded.
如果任務被拒絕了,則由調用線程(提交任務的線程)直接執行此任務,我們可以通過代碼來驗證這一點:
把之前的代碼修改如下:
public static void main(String[] args) { BlockingQueue queue = new ArrayBlockingQueue<>(10); ThreadFactory factory = r -> new Thread(r, "test-thread-pool"); ThreadPoolExecutor executor = new ThreadPoolExecutor(5, 5, 0L, TimeUnit.SECONDS, queue, factory, new ThreadPoolExecutor.CallerRunsPolicy()); for (int i = 0; i < 1000; i++) { executor.submit(() -> { try { System.out.println(Thread.currentThread().getName() + ":執行任務"); Thread.sleep(1000); } catch (InterruptedException e) { e.printStackTrace(); } }); }}

把隊列最大值改為10,打印輸出線程的名稱。執行結果如下:

通過結果可以看到,主線程main也執行了任務,這正說明了此拒絕策略由調用線程(提交任務的線程)直接執行被丟棄的任務的。
總結
本文介紹和演示了四種線程池拒絕策略,具體使用哪種策略,還得根據實際業務場景才能做出抉擇。