java高并發(fā)ScheduledThreadPoolExecutor類深度解析
正文
在【高并發(fā)專題】的專欄中，我們深度分析了ThreadPoolExecutor類的源代碼，而ScheduledThreadPoolExecutor類是ThreadPoolExecutor類的子類。今天我們就來一起手撕ScheduledThreadPoolExecutor類的源代碼。
構造方法
我們先來看下ScheduledThreadPoolExecutor的構造方法，源代碼如下所示。
public ScheduledThreadPoolExecutor(int corePoolSize) {
super(corePoolSize, Integer.MAX_VALUE, 0, NANOSECONDS, new DelayedWorkQueue());
}
?
public ScheduledThreadPoolExecutor(int corePoolSize, ThreadFactory threadFactory) {
super(corePoolSize, Integer.MAX_VALUE, 0, NANOSECONDS,
new DelayedWorkQueue(), threadFactory);
}
?
public ScheduledThreadPoolExecutor(int corePoolSize, RejectedExecutionHandler handler) {
super(corePoolSize, Integer.MAX_VALUE, 0, NANOSECONDS,
new DelayedWorkQueue(), handler);
}
?
public ScheduledThreadPoolExecutor(int corePoolSize, ThreadFactory threadFactory, RejectedExecutionHandler handler) {
super(corePoolSize, Integer.MAX_VALUE, 0, NANOSECONDS,
new DelayedWorkQueue(), threadFactory, handler);
}
從代碼結構上來看，ScheduledThreadPoolExecutor類是ThreadPoolExecutor類的子類，ScheduledThreadPoolExecutor類的構造方法實際上調用的是ThreadPoolExecutor類的構造方法。
schedule方法
接下來，我們看一下ScheduledThreadPoolExecutor類的schedule方法，源代碼如下所示。
public ScheduledFuture<?> schedule(Runnable command, long delay, TimeUnit unit) {
//如果傳遞的Runnable對象和TimeUnit時間單位為空
//拋出空指針異常
if (command == null || unit == null)
throw new NullPointerException();
//封裝任務對象，在decorateTask方法中直接返回ScheduledFutureTask對象
RunnableScheduledFuture<?> t = decorateTask(command, new ScheduledFutureTask<Void>(command, null, triggerTime(delay, unit)));
//執(zhí)行延時任務
delayedExecute(t);
//返回任務
return t;
}
?
public <V> ScheduledFuture<V> schedule(Callable<V> callable, long delay, TimeUnit unit)
//如果傳遞的Callable對象和TimeUnit時間單位為空
//拋出空指針異常
if (callable == null || unit == null)
throw new NullPointerException();
//封裝任務對象，在decorateTask方法中直接返回ScheduledFutureTask對象
RunnableScheduledFuture<V> t = decorateTask(callable,
new ScheduledFutureTask<V>(callable, triggerTime(delay, unit)));
//執(zhí)行延時任務
delayedExecute(t);
//返回任務
return t;
}
從源代碼可以看出，ScheduledThreadPoolExecutor類提供了兩個重載的schedule方法，兩個schedule方法的第一個參數不同?？梢詡鬟fRunnable接口對象，也可以傳遞Callable接口對象。在方法內部，會將Runnable接口對象和Callable接口對象封裝成RunnableScheduledFuture對象，本質上就是封裝成ScheduledFutureTask對象。并通過delayedExecute方法來執(zhí)行延時任務。
在源代碼中，我們看到兩個schedule都調用了decorateTask方法，接下來，我們就看看decorateTask方法。
decorateTask方法
decorateTask方法源代碼如下所示。
protected <V> RunnableScheduledFuture<V> decorateTask(Runnable runnable, RunnableScheduledFuture<V> task) {
return task;
}
protected <V> RunnableScheduledFuture<V> decorateTask(Callable<V> callable, RunnableScheduledFuture<V> task) {
return task;
}
通過源碼可以看出decorateTask方法的實現比較簡單，接收一個Runnable接口對象或者Callable接口對象和封裝的RunnableScheduledFuture任務，兩個方法都是將RunnableScheduledFuture任務直接返回。在ScheduledThreadPoolExecutor類的子類中可以重寫這兩個方法。
接下來，我們繼續(xù)看下scheduleAtFixedRate方法。
scheduleAtFixedRate方法
scheduleAtFixedRate方法源代碼如下所示。
public ScheduledFuture<?> scheduleAtFixedRate(Runnable command, long initialDelay, long period, TimeUnit unit) {
//傳入的Runnable對象和TimeUnit為空，則拋出空指針異常
if (command == null || unit == null)
throw new NullPointerException();
//如果執(zhí)行周期period傳入的數值小于或者等于0
//拋出非法參數異常
if (period <= 0)
throw new IllegalArgumentException();
//將Runnable對象封裝成ScheduledFutureTask任務，
//并設置執(zhí)行周期
ScheduledFutureTask<Void> sft =
new ScheduledFutureTask<Void>(command, null, triggerTime(initialDelay, unit), unit.toNanos(period));
//調用decorateTask方法，本質上還是直接返回ScheduledFutureTask對象
RunnableScheduledFuture<Void> t = decorateTask(command, sft);
//設置執(zhí)行的任務
sft.outerTask = t;
//執(zhí)行延時任務
delayedExecute(t);
//返回執(zhí)行的任務
return t;
}
通過源碼可以看出，scheduleAtFixedRate方法將傳遞的Runnable對象封裝成ScheduledFutureTask任務對象，并設置了執(zhí)行周期，下一次的執(zhí)行時間相對于上一次的執(zhí)行時間來說，加上了period時長，時長的具體單位由TimeUnit決定。采用固定的頻率來執(zhí)行定時任務。
ScheduledThreadPoolExecutor類中另一個定時調度任務的方法是scheduleWithFixedDelay方法，接下來，我們就一起看看scheduleWithFixedDelay方法。
scheduleWithFixedDelay方法
scheduleWithFixedDelay方法的源代碼如下所示。
public ScheduledFuture<?> scheduleWithFixedDelay(Runnable command, long initialDelay, long delay, TimeUnit unit) {
//傳入的Runnable對象和TimeUnit為空，則拋出空指針異常
if (command == null || unit == null)
throw new NullPointerException();
//任務延時時長小于或者等于0，則拋出非法參數異常
if (delay <= 0)
throw new IllegalArgumentException();
//將Runnable對象封裝成ScheduledFutureTask任務
//并設置固定的執(zhí)行周期來執(zhí)行任務
ScheduledFutureTask<Void> sft =
new ScheduledFutureTask<Void>(command, null,triggerTime(initialDelay, unit), unit.toNanos(-delay));
//調用decorateTask方法，本質上直接返回ScheduledFutureTask任務
RunnableScheduledFuture<Void> t = decorateTask(command, sft);
//設置執(zhí)行的任務
sft.outerTask = t;
//執(zhí)行延時任務
delayedExecute(t);
//返回任務
return t;
}
從scheduleWithFixedDelay方法的源代碼，我們可以看出在將Runnable對象封裝成ScheduledFutureTask時，設置了執(zhí)行周期，但是此時設置的執(zhí)行周期與scheduleAtFixedRate方法設置的執(zhí)行周期不同。此時設置的執(zhí)行周期規(guī)則為：下一次任務執(zhí)行的時間是上一次任務完成的時間加上delay時長，時長單位由TimeUnit決定。也就是說，具體的執(zhí)行時間不是固定的，但是執(zhí)行的周期是固定的，整體采用的是相對固定的延遲來執(zhí)行定時任務。
如果大家細心的話，會發(fā)現在scheduleWithFixedDelay方法中設置執(zhí)行周期時，傳遞的delay值為負數，如下所示。
ScheduledFutureTask<Void> sft =
new ScheduledFutureTask<Void>(command, null, triggerTime(initialDelay, unit), unit.toNanos(-delay));
這里的負數表示的是相對固定的延遲。
在ScheduledFutureTask類中，存在一個setNextRunTime方法，這個方法會在run方法執(zhí)行完任務后調用，這個方法更能體現scheduleAtFixedRate方法和scheduleWithFixedDelay方法的不同，setNextRunTime方法的源碼如下所示。
private void setNextRunTime() {
//距離下次執(zhí)行任務的時長
long p = period;
//固定頻率執(zhí)行，
//上次執(zhí)行任務的時間
//加上任務的執(zhí)行周期
if (p > 0)
time += p;
//相對固定的延遲
//使用的是系統(tǒng)當前時間
//加上任務的執(zhí)行周期
else
time = triggerTime(-p);
}
在setNextRunTime方法中通過對下次執(zhí)行任務的時長進行判斷來確定是固定頻率執(zhí)行還是相對固定的延遲。
triggerTime方法
在ScheduledThreadPoolExecutor類中提供了兩個triggerTime方法，用于獲取下一次執(zhí)行任務的具體時間。triggerTime方法的源碼如下所示。
private long triggerTime(long delay, TimeUnit unit) {
return triggerTime(unit.toNanos((delay < 0) ? 0 : delay));
}
?
long triggerTime(long delay) {
return now() +
((delay < (Long.MAX_VALUE >> 1)) ? delay : overflowFree(delay));
}
這兩個triggerTime方法的代碼比較簡單，就是獲取下一次執(zhí)行任務的具體時間。有一點需要注意的是：delay < (Long.MAX_VALUE >> 1判斷delay的值是否小于Long.MAX_VALUE的一半，如果小于Long.MAX_VALUE值的一半，則直接返回delay，否則需要處理溢出的情況。
我們看到在triggerTime方法中處理防止溢出的邏輯使用了overflowFree方法，接下來，我們就看看overflowFree方法的實現。
overflowFree方法
overflowFree方法的源代碼如下所示。
private long overflowFree(long delay) {
//獲取隊列中的節(jié)點
Delayed head = (Delayed) super.getQueue().peek();
//獲取的節(jié)點不為空，則進行后續(xù)處理
if (head != null) {
//從隊列節(jié)點中獲取延遲時間
long headDelay = head.getDelay(NANOSECONDS);
//如果從隊列中獲取的延遲時間小于0，并且傳遞的delay
//值減去從隊列節(jié)點中獲取延遲時間小于0
if (headDelay < 0 && (delay - headDelay < 0))
//將delay的值設置為Long.MAX_VALUE + headDelay
delay = Long.MAX_VALUE + headDelay;
}
//返回延遲時間
return delay;
}
通過對overflowFree方法的源碼分析，可以看出overflowFree方法本質上就是為了限制隊列中的所有節(jié)點的延遲時間在Long.MAX_VALUE值之內，防止在ScheduledFutureTask類中的compareTo方法中溢出。
ScheduledFutureTask類中的compareTo方法的源碼如下所示。
public int compareTo(Delayed other) {
if (other == this) // compare zero if same object
return 0;
if (other instanceof ScheduledFutureTask) {
ScheduledFutureTask<?> x = (ScheduledFutureTask<?>)other;
long diff = time - x.time;
if (diff < 0)
return -1;
else if (diff > 0)
return 1;
else if (sequenceNumber < x.sequenceNumber)
return -1;
else
return 1;
}
long diff = getDelay(NANOSECONDS) - other.getDelay(NANOSECONDS);
return (diff < 0) ? -1 : (diff > 0) ? 1 : 0;
}
compareTo方法的主要作用就是對各延遲任務進行排序，距離下次執(zhí)行時間靠前的任務就排在前面。
delayedExecute方法
delayedExecute方法是ScheduledThreadPoolExecutor類中延遲執(zhí)行任務的方法，源代碼如下所示。
private void delayedExecute(RunnableScheduledFuture<?> task) {
//如果當前線程池已經關閉
//則執(zhí)行線程池的拒絕策略
if (isShutdown())
reject(task);
//線程池沒有關閉
else {
//將任務添加到阻塞隊列中
super.getQueue().add(task);
//如果當前線程池是SHUTDOWN狀態(tài)
//并且當前線程池狀態(tài)下不能執(zhí)行任務
//并且成功從阻塞隊列中移除任務
if (isShutdown() &&
!canRunInCurrentRunState(task.isPeriodic()) &&
remove(task))
//取消任務的執(zhí)行，但不會中斷執(zhí)行中的任務
task.cancel(false);
else
//調用ThreadPoolExecutor類中的ensurePrestart()方法
ensurePrestart();
}
}
可以看到在delayedExecute方法內部調用了canRunInCurrentRunState方法，canRunInCurrentRunState方法的源碼實現如下所示。
boolean canRunInCurrentRunState(boolean periodic) {
return isRunningOrShutdown(periodic ? continueExistingPeriodicTasksAfterShutdown : executeExistingDelayedTasksAfterShutdown);
}
可以看到canRunInCurrentRunState方法的邏輯比較簡單，就是判斷線程池當前狀態(tài)下能夠執(zhí)行任務。
另外，在delayedExecute方法內部還調用了ThreadPoolExecutor類中的ensurePrestart()方法，接下來，我們看下ThreadPoolExecutor類中的ensurePrestart()方法的實現，如下所示。
void ensurePrestart() {
int wc = workerCountOf(ctl.get());
if (wc < corePoolSize)
addWorker(null, true);
else if (wc == 0)
addWorker(null, false);
}
在ThreadPoolExecutor類中的ensurePrestart()方法中，首先獲取當前線程池中線程的數量，如果線程數量小于corePoolSize則調用addWorker方法傳遞null和true，如果線程數量為0，則調用addWorker方法傳遞null和false。
關于addWork()方法的源碼解析，大家可以參考【高并發(fā)專題】中的《高并發(fā)之——通過ThreadPoolExecutor類的源碼深度解析線程池執(zhí)行任務的核心流程》一文，這里，不再贅述。
reExecutePeriodic方法
reExecutePeriodic方法的源代碼如下所示。
void reExecutePeriodic(RunnableScheduledFuture<?> task) {
//線程池當前狀態(tài)下能夠執(zhí)行任務
if (canRunInCurrentRunState(true)) {
//將任務放入隊列
super.getQueue().add(task);
//線程池當前狀態(tài)下不能執(zhí)行任務，并且成功移除任務
if (!canRunInCurrentRunState(true) && remove(task))
//取消任務
task.cancel(false);
else
//調用ThreadPoolExecutor類的ensurePrestart()方法
ensurePrestart();
}
}
總體來說reExecutePeriodic方法的邏輯比較簡單，但是，這里需要注意和delayedExecute方法的不同點：調用reExecutePeriodic方法的時候已經執(zhí)行過一次任務，所以，并不會觸發(fā)線程池的拒絕策略；傳入reExecutePeriodic方法的任務一定是周期性的任務。
onShutdown方法
onShutdown方法是ThreadPoolExecutor類中的鉤子函數，它是在ThreadPoolExecutor類中的shutdown方法中調用的，而在ThreadPoolExecutor類中的onShutdown方法是一個空方法，如下所示。
void onShutdown() {
}
ThreadPoolExecutor類中的onShutdown方法交由子類實現，所以ScheduledThreadPoolExecutor類覆寫了onShutdown方法，實現了具體的邏輯，ScheduledThreadPoolExecutor類中的onShutdown方法的源碼實現如下所示。
@Override
void onShutdown() {
//獲取隊列
BlockingQueue<Runnable> q = super.getQueue();
//在線程池已經調用shutdown方法后，是否繼續(xù)執(zhí)行現有延遲任務
boolean keepDelayed = getExecuteExistingDelayedTasksAfterShutdownPolicy();
//在線程池已經調用shutdown方法后，是否繼續(xù)執(zhí)行現有定時任務
boolean keepPeriodic = getContinueExistingPeriodicTasksAfterShutdownPolicy();
//在線程池已經調用shutdown方法后，不繼續(xù)執(zhí)行現有延遲任務和定時任務
if (!keepDelayed && !keepPeriodic) {
//遍歷隊列中的所有任務
for (Object e : q.toArray())
//取消任務的執(zhí)行
if (e instanceof RunnableScheduledFuture<?>)
((RunnableScheduledFuture<?>) e).cancel(false);
//清空隊列
q.clear();
}
//在線程池已經調用shutdown方法后，繼續(xù)執(zhí)行現有延遲任務和定時任務
else {
//遍歷隊列中的所有任務
for (Object e : q.toArray()) {
//當前任務是RunnableScheduledFuture類型
if (e instanceof RunnableScheduledFuture) {
//將任務強轉為RunnableScheduledFuture類型
RunnableScheduledFuture<?> t = (RunnableScheduledFuture<?>)e;
//在線程池調用shutdown方法后不繼續(xù)的延遲任務或周期任務
//則從隊列中刪除并取消任務
if ((t.isPeriodic() ? !keepPeriodic : !keepDelayed) ||
t.isCancelled()) {
if (q.remove(t))
t.cancel(false);
}
}
}
}
//最終調用tryTerminate()方法
tryTerminate();
}
ScheduledThreadPoolExecutor類中的onShutdown方法的主要邏輯就是先判斷線程池調用shutdown方法后，是否繼續(xù)執(zhí)行現有的延遲任務和定時任務，如果不再執(zhí)行，則取消任務并清空隊列；如果繼續(xù)執(zhí)行，將隊列中的任務強轉為RunnableScheduledFuture對象之后，從隊列中刪除并取消任務。大家需要好好理解這兩種處理方式。
最后調用ThreadPoolExecutor類的tryTerminate方法。有關ThreadPoolExecutor類的tryTerminate方法的源碼解析，大家可以參考【高并發(fā)專題】中的《高并發(fā)之——通過源碼深度分析線程池中Worker線程的執(zhí)行流程》一文，這里不再贅述。
至此，ScheduledThreadPoolExecutor類中的核心方法的源代碼，我們就分析完了，更多關于java高并發(fā)ScheduledThreadPoolExecutor的資料請關注腳本之家其它相關文章！
相關文章
解決java.util.NoSuchElementException異常正確方法
java.util.NoSuchElementException是Java中的一種異常,表示在迭代器或枚舉中找不到元素,這篇文章主要給大家介紹了關于解決java.util.NoSuchElementException異常的相關資料,需要的朋友可以參考下2023-11-11
這篇文章主要介紹了詳解關于java文件下載文件名亂碼問題解決方案，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-01-01
Jackson?是?Java?中最流行的?JSON?處理庫之一，它提供了許多注解來簡化?JSON?的序列化和反序列化過程。這篇文章將介紹一些?Jackson?常用的注解，以幫助您更輕松地處理?JSON?數據2023-05-05
Liquibase 在 Spring Boot 中的使用詳細介紹
Liquibase 提供了靈活的變更集機制,支持創(chuàng)建表、修改列、填充數據、回滾變更等多種操作,本文將通過多個豐富的示例,詳細講解如何在 Spring Boot項目中使用 Liquibase,感興趣的朋友跟隨小編一起看看吧2024-12-12
這篇文章主要介紹了java Unicode和UTF-8之間轉換實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-09-09
這篇文章主要介紹了spring boot 圖片上傳與顯示功能實例詳解,需要的朋友可以參考下2017-04-04
這篇文章主要介紹了Java進程cpu占用過高問題解決,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-04-04
這篇文章主要介紹了淺談java中BigDecimal類的簡單用法，在開發(fā)時，如果我們需要精確計算的結果，必須使用BigDecimal類來操作。感興趣的話可以了解一下2020-07-07
Spring?JPA?deleteInBatch導致StackOverflow問題
這篇文章主要介紹了Spring?JPA?deleteInBatch導致StackOverflow問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-05-05
這篇文章主要介紹了并發(fā)編程Java內存模型順序一致性，順序一致性內存模型是一個理論參考模型，處理器的內存模型和編程語言的內存模型都會以順序一致性內存模型作為參照，下面我們一起進入文章看看學校內容,需要的朋友可以參考一下2021-11-11