跳到主要內容

線程池續：你必須要知道的線程池submit()實現原理之FutureTask！_台北網頁設計


※廣告預算用在刀口上，台北網頁設計公司幫您達到更多曝光效益



擁有後台管理系統的網站，將擁有強大的資料管理與更新功能，幫助您隨時新增網站的內容並節省網站開發的成本。




前言


上一篇內容寫了Java中線程池的實現原理及源碼分析，說好的是實實在在的大滿足，想通過一篇文章讓大家對線程池有個透徹的了解，但是文章寫完總覺得還缺點什麼?


上篇文章只提到線程提交的execute()方法，並沒有講解線程提交的submit()方法，submit()有一個返回值，可以獲取線程執行的結果Future<T>，這一講就那深入學習下submit()和FutureTask實現原理。


使用場景&示例


使用場景


我能想到的使用場景就是在并行計算的時候，例如一個方法中調用methodA()、methodB()，我們可以通過線程池異步去提交方法A、B，然後在主線程中獲取組裝方法A、B計算后的結果，能夠大大提升方法的吞吐量。



使用示例


/**
  * @author wangmeng
  * @date 2020/5/28 15:30
  */
 public class FutureTaskTest {
     public static void main(String[] args) throws InterruptedException, ExecutionException {
         ExecutorService threadPool = Executors.newCachedThreadPool();
 
         System.out.println("====執行FutureTask線程任務====");
         Future<String> futureTask = threadPool.submit(new Callable<String>() {
             @Override
             public String call() throws Exception {
                 System.out.println("FutureTask執行業務邏輯");
                 Thread.sleep(2000);
                 System.out.println("FutureTask業務邏輯執行完畢！");
                 return "歡迎關注: 一枝花算不算浪漫！";
             }
         });
 
         System.out.println("====執行主線程任務====");
         Thread.sleep(1000);
         boolean flag = true;
         while(flag){
             if(futureTask.isDone() && !futureTask.isCancelled()){
                 System.out.println("FutureTask異步任務執行結果：" + futureTask.get());
                 flag = false;
             }
         }
 
         threadPool.shutdown();
     }
 }

上面的使用很簡單，submit()內部傳遞的實際上是個Callable接口，我們自己實現其中的call()方法，我們通過futureTask既可以獲取到具體的返回值。


submit()實現原理


submit() 是也是提交任務到線程池，只是它可以獲取任務返回結果，返回結果是通過FutureTask來實現的，先看下ThreadPoolExecutor中代碼實現：


public class ThreadPoolExecutor extends AbstractExecutorService {
     public <T> Future<T> submit(Callable<T> task) {
         if (task == null) throw new NullPointerException();
         RunnableFuture<T> ftask = newTaskFor(task);
         execute(ftask);
         return ftask;
     }
 }
 
 public abstract class AbstractExecutorService implements ExecutorService {
     protected <T> RunnableFuture<T> newTaskFor(Callable<T> callable) {
         return new FutureTask<T>(callable);
     }
 }

提交任務還是執行execute()方法，只是task被包裝成了FutureTask ，也就是在excute()中啟動線程後會執行FutureTask.run()方法。


再來具體看下它執行的完整鏈路圖：



上圖可以看到，執行任務並返回執行結果的核心邏輯實在FutureTask中，我們以FutureTask.run/get 兩個方法為突破口，一點點剖析FutureTask的實現原理。


FutureTask源碼初探


先看下FutureTask中部分屬性：



public class FutureTask<V> implements RunnableFuture<V> {
     private volatile int state;
     private static final int NEW          = 0;
     private static final int COMPLETING   = 1;
     private static final int NORMAL       = 2;
     private static final int EXCEPTIONAL  = 3;
     private static final int CANCELLED    = 4;
     private static final int INTERRUPTING = 5;
     private static final int INTERRUPTED  = 6;
 
     private Callable<V> callable;
     private Object outcome;
     private volatile Thread runner;
     private volatile WaitNode waiters;
 }


  1. state



當前task狀態，共有7中類型。
NEW： 當前任務尚未執行
COMPLETING: 當前任務正在結束，尚未完全結束，一種臨界狀態
NORMAL：當前任務正常結束
EXCEPTIONAL: 當前任務執行過程中發生了異常。
CANCELLED: 當前任務被取消
INTERRUPTING: 當前任務中斷中..
INTERRUPTED: 當前任務已中斷




  1. callble



用戶提交任務傳遞的Callable，自定義call方法，實現業務邏輯




  1. outcome



任務結束時，outcome保存執行結果或者異常信息。




  1. runner



當前任務被線程執行期間，保存當前任務的線程對象引用




  1. waiters



因為會有很多線程去get當前任務的結果，所以這裏使用了一種stack數據結構來保存



FutureTask.run()實現原理


我們已經知道在線程池runWorker()中最終會調用到FutureTask.run()方法中，我們就來看下它的執行原理吧：



具體代碼如下：


public class FutureTask<V> implements RunnableFuture<V> {
     public void run() {
         if (state != NEW ||
             !UNSAFE.compareAndSwapObject(this, runnerOffset, null, Thread.currentThread()))
             return;
         try {
             Callable<V> c = callable;
             if (c != null && state == NEW) {
                 V result;
                 boolean ran;
                 try {
                     result = c.call();
                     ran = true;
                 } catch (Throwable ex) {
                     result = null;
                     ran = false;
                     setException(ex);
                 }
                 if (ran)
                     set(result);
             }
         } finally {
             runner = null;
             int s = state;
             if (s >= INTERRUPTING)
                 handlePossibleCancellationInterrupt(s);
         }
     }
 }

首先是判斷FutureTask中state狀態，必須是NEW才可以繼續執行。


然後通過CAS修改runner引用為當前線程。


接着執行用戶自定義的call()方法，將返回結果設置到result中，result可能為正常返回也可能為異常信息。這裏主要是調用set()/setException()


FutureTask.set()實現原理


set()方法的實現很簡單，直接看下代碼：


public class FutureTask<V> implements RunnableFuture<V> {
     protected void set(V v) {
         if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
             outcome = v;
             UNSAFE.putOrderedInt(this, stateOffset, NORMAL);
             finishCompletion();
         }
     }
 }

將call()返回的數據賦值給全局變量outcome上，然後修改state狀態為NORMAL，最後調用finishCompletion()來做掛起線程的喚醒操作，這個方法等到get()後面再來講解。


FutureTask.get()實現原理



接着看下代碼：


public class FutureTask<V> implements RunnableFuture<V> {
     public V get() throws InterruptedException, ExecutionException {
         int s = state;
         if (s <= COMPLETING)
             s = awaitDone(false, 0L);
         return report(s);
     }
 }

如果FutureTask中state為NORMAL或者COMPLETING，說明當前任務並沒有執行完成，調用get()方法會被阻塞，具體的阻塞邏輯在awaitDone()方法：


private int awaitDone(boolean timed, long nanos) throws InterruptedException {
 
         final long deadline = timed ? System.nanoTime() + nanos : 0L;
         WaitNode q = null;
         boolean queued = false;
         for (;;) {
             if (Thread.interrupted()) {
                 removeWaiter(q);
                 throw new InterruptedException();
             }
 
             int s = state;
             if (s > COMPLETING) {
                 if (q != null)
                     q.thread = null;
                 return s;
             }
             else if (s == COMPLETING)
                 Thread.yield();
             else if (q == null)
                 q = new WaitNode();
             else if (!queued)
                 queued = UNSAFE.compareAndSwapObject(this, waitersOffset, q.next = waiters, q);
             else if (timed) {
                 nanos = deadline - System.nanoTime();
                 if (nanos <= 0L) {
                     removeWaiter(q);
                     return state;
                 }
                 LockSupport.parkNanos(this, nanos);
             }
             else
                 LockSupport.park(this);
         }
     }

這個方法可以說是FutureTask中最核心的方法了，一步步來分析：


如果timed不為空，這說明指定nanos時間還未返回結果，線程就會退出。


※廣告預算用在刀口上，台北網頁設計公司幫您達到更多曝光效益



擁有後台管理系統的網站，將擁有強大的資料管理與更新功能，幫助您隨時新增網站的內容並節省網站開發的成本。




q是一個WaitNode對象，是將當前引用線程封裝在一個stack數據結構中，WaitNode對象屬性如下：


 static final class WaitNode {
     volatile Thread thread;
     volatile WaitNode next;
     WaitNode() { thread = Thread.currentThread(); }
 }

接着判斷當前線程是否中斷，如果中斷則拋出中斷異常。


下面就進入一輪輪的if... else if...判斷邏輯，我們還是採用分支的方式去分析。


分支一：if (s > COMPLETING) {


此時get()方法已經有結果了，無論是正常返回的結果，還是異常、中斷、取消等，此時直接返回state狀態，然後執行report()方法。


分支二：else if (s == COMPLETING)


條件成立，說明當前任務接近完成狀態，這裏讓當前線程再釋放cpu，進行下一輪搶佔cpu。


分支三：else if (q == null)


第一次自旋執行，WaitNode還沒有初始化，初始化q=new WaitNode();


分支四：else if (!queued){


queued代表當前線程是否入棧，如果沒有入棧則進行入棧操作，順便將全局變量waiters指向棧頂元素。


分支五/六：LockSupport.park


如果設置了超時時間，則使用parkNanos來掛起當前線程，否則使用park()


經過這麼一輪自旋循環后，如果執行call()還沒有返回結果，那麼調用get()方法的線程都會被掛起。


被掛起的線程會等待run()返回結果后依次喚醒，具體的執行邏輯在finishCompletion()中。


最終stack結構中數據如下：



FutureTask.finishCompletion()實現原理



具體實現代碼如下：


private void finishCompletion() {
     for (WaitNode q; (q = waiters) != null;) {
         if (UNSAFE.compareAndSwapObject(this, waitersOffset, q, null)) {
             for (;;) {
                 Thread t = q.thread;
                 if (t != null) {
                     q.thread = null;
                     LockSupport.unpark(t);
                 }
                 WaitNode next = q.next;
                 if (next == null)
                     break;
                 q.next = null;
                 q = next;
             }
             break;
         }
     }
 
     done();
 
     callable = null;
 }

代碼實現很簡單，看過get()方法后，我們知道所有調用get()方法的線程，在run()還沒有返回結果前，都會保存到一個有WaitNode構成的statck數據結構中，而且每個線程都會被掛起。


這裡是遍歷waiters棧頂元素，然後依次查詢起next節點進行喚醒，喚醒后的節點接着會往後調用report()方法。


FutureTask.report()實現原理



具體代碼如下：


private V report(int s) throws ExecutionException {
     Object x = outcome;
     if (s == NORMAL)
         return (V)x;
     if (s >= CANCELLED)
         throw new CancellationException();
     throw new ExecutionException((Throwable)x);
 }

這個方法很簡單，因為執行到了這裏，說明當前state狀態肯定大於COMPLETING，判斷如果是正常返回，那麼返回outcome數據。


如果state是取消狀態，拋出CancellationException異常。


如果狀態都不滿足，則說明執行中出現了差錯，直接拋出ExecutionException


FutureTask.cancel()實現原理



public boolean cancel(boolean mayInterruptIfRunning) {
     if (!(state == NEW && UNSAFE.compareAndSwapInt(this, stateOffset, NEW, mayInterruptIfRunning ? INTERRUPTING : CANCELLED)))
         return false;
     try {
         if (mayInterruptIfRunning) {
             try {
                 Thread t = runner;
                 if (t != null)
                     t.interrupt();
             } finally {
                 UNSAFE.putOrderedInt(this, stateOffset, INTERRUPTED);
             }
         }
     } finally {
         finishCompletion();
     }
     return true;
 }

cancel()方法的邏輯很簡單，就是修改state狀態為CANCELLED，然後調用finishCompletion()來喚醒等待的線程。


這裏如果mayInterruptIfRunning，就會先中斷當前線程，然後再去喚醒等待的線程。


總結


FutureTask的實現原理其實很簡單，每個方法基本上都畫了一個簡單的流程圖來方便立即。


後面還打算分享一個BlockingQueue相關的源碼解讀，這樣線程池也可以算是完結了。


在這之前可能會先分享一個SpringCloud常見配置代碼分析、最佳實踐等手冊，方便工作中使用，也是對之前看過的源碼一種總結。敬請期待！
歡迎關注：

本站聲明:網站內容來源於博客園,如有侵權,請聯繫我們,我們將及時處理

※廣告預算用在刀口上，台北網頁設計公司幫您達到更多曝光效益



擁有後台管理系統的網站，將擁有強大的資料管理與更新功能，幫助您隨時新增網站的內容並節省網站開發的成本。




Orignal From: 線程池續：你必須要知道的線程池submit()實現原理之FutureTask！_台北網頁設計

留言

張貼留言

這個網誌中的熱門文章

Python 併發總結，多線程，多進程，異步IO

1 測量函數運行時間 import time def profile(func): def wrapper(*args, ** kwargs): import time start = time.time() func( *args, ** kwargs) end = time.time() print ' COST: {} ' .format(end - start) return wrapper @profile def fib(n): if n<= 2 : return 1 return fib(n-1) + fib(n-2 ) fib( 35 )   2 啟動多個線程，並等待完成   2.1 使用threading.enumerate() import threading for i in range(2 ): t = threading.Thread(target=fib, args=(35 ,)) t.start() main_thread = threading.currentThread() for t in threading.enumerate(): if t is main_thread: continue t.join()   2.2 先保存啟動的線程 threads = [] for i in range(5 ): t = Thread(target=foo, args= (i,)) threads.append(t) t.start() for t in threads: t.join()   3 使用信號量，限制同時能有幾個線程訪問臨界區 from threading import Semaphore import time sema = Semaphor...
張貼留言
閱讀完整內容

高雄十大包子名店出爐

, 圖文：吳恩文 高雄包子大賽落幕了，我只能就我個人意見， 介紹一下前十名這些包子，但是不能代表其他四位評審的意見，雖然身為評審長，我通常不會第一個表示意見，以免影響其他評審， 我主要工作是負責發問。   這次參賽的素包子很少，而且都不夠細致，又偏油，我不愛， 但是第一名的甜芝麻包－熔岩黑金包，竟然是素食得名－ 漢來蔬食巨蛋店。   這包子賣相太好，竹炭粉的黑色外皮刷上金粉，一上桌，眾人驚呼， 搶拍照，內餡是芝麻餡，混一點花生醬增稠，加入白糖芝麻油， 熔岩爆漿的程度剛剛好，我一直以為芝麻要配豬油才行、 但是選到好的黑芝麻油一樣不減香醇， 當下有二位評審就想宅配回家。   尤其特別的是，黑芝麻餡室溫易化，師傅必須要輪班躲在冷藏室內， 穿著大外套才能包，一天包不了多少，我笑說，漢來美食，集團餐廳那麼多，實力雄厚，根本是「 奧運選手報名參加村裡運動會」嘛，其他都是小包子店啊， 但是沒辦法，顯然大家都覺得它好看又好吃， 目前限定漢來蔬食高雄巨蛋店，二顆88元，可以冷凍宅配， 但是要排一陣子，因為供不應求，聽說，四月份， 台北sogo店開始會賣。   第二名的包子，左營寬來順早餐店，顯然平易近人的多，一顆肉包， 十塊錢，是所有參賽者中最便宜的，當然，個頭也小， 它的包子皮明顯和其他不同，灰灰的老麵，薄但紮實有嚼勁， 肉餡新鮮帶汁，因為打了些水，味道極其簡單，就是蔥薑，塩， 香油，薑味尤其明顯，是老眷村的味道， 而特別的是老闆娘是台灣本省人， 當年完全是依據眷村老兵的口味一步一步調整而來，沒有加什麼糖、 五香粉，胡椒粉，油蔥酥。就是蔥薑豬肉和老麵香，能得名， 應該是它的平實無華，鮮美簡單，打動人心。   這是標準的心靈美食，可以撫慰人心，得名之前，寛來順已經天天排隊，現在，恐怕要排更久了， 建議大家六七點早點上門。   第三名，「專十一」很神奇，我記得比賽最後， 大家連吃了幾家不能引起共鳴的包子，有些累，到了專十一， 就坐著等包子，其他評審一吃，就催我趕快試，我一吃， 也醒了大半。   它的包子皮厚薄適中，但是高筋麵粉高些，老麵加一點點酵母， 我心中，它的皮屬一屬二，至於餡又多又好吃，蛋黃還是切丁拌入， 不是整顆放，吃起來「美味、均衡、飽滿」。一顆二十元。   老闆是陸軍專科十一期畢業取名專十一，...
張貼留言
閱讀完整內容

韋伯連續劇終於更新 期待第一季順利完結

　　地球天文學界的跳票大王詹姆斯·韋伯空間望遠鏡 (James Webb Space Telescope，縮寫為 JWST)自 1996 年以來斷斷續續不按劇本演出的連續劇終於讓焦慮的觀眾們又等到了一次更新：五層遮陽罩測試順利完成。 裝配完成的韋伯望遠鏡與好夥伴遮陽罩同框啦。Credit: NASA 　　嚴格的測試是任何空間任務順利成功的重中之重。遮陽罩，這個韋伯望遠鏡異常重要的親密夥伴，要是無法正常運轉的話，韋伯的這一季天文界連續劇說不準就要一直拖更了。 　　詹姆斯·韋伯空間望遠鏡是歷史上造出的最先進的空間望遠鏡。它不僅是一架紅外望遠鏡，還具有特別高的靈敏度。但想要達到辣么高的靈敏度來研究系外行星和遙遠的宇宙童年，韋伯童鞋必須非常"冷靜"，體溫升高的話，靈敏度會大大折損。這個時候，遮陽罩就要大顯身手啦。 　　遮陽罩在韋伯的設計中至關重要。韋伯望遠鏡會被發射到拉格朗日 L2 點，運行軌道很高，遠離太陽、地球與月球。太陽是韋伯的主要熱量干擾的來源，其次是地球與月球。遮陽罩會有效阻斷來自這三大熱源的能量並保護韋伯維持在工作溫度正常運轉。這個工作溫度指的是零下 220 攝氏度(-370 華氏度；50 開爾文)。 上圖中我們可以看出，韋伯望遠鏡的配置大致可分為兩部分：紅色較熱的一面溫度為 85 攝氏度，藍色較冷的一面溫度達到零下 233 攝氏度。紅色的這部分中，儀器包括太陽能板、通信設備、計算機、以及轉向裝置。藍色部分的主要裝置包括鏡面、探測器、濾光片等。Credit: STSci. 　　遮陽罩的那一部分和望遠鏡的鏡面這部分可以產生非常極端的溫差。遮陽的這面溫度可以達到 110 攝氏度，足以煮熟雞蛋，而背陰處的部分溫度極低，足以凍結氧氣。 　　工程師們剛剛完成了五層遮陽罩的測試，按照韋伯在 L2 時的運行狀態安裝了遮陽罩。L2 距離地球約 160 萬公里。NASA 表示這些測試使用了航天器的自帶系統來展開遮陽罩，測試目前都已成功完成。韋伯望遠鏡遮陽罩負責人 James Cooper 介紹說這是遮陽罩"第一次在望遠鏡系統的电子設備的控制下展開。儘管這個任務非常艱巨，難度高，但測試順利完成，遮陽罩展開時的狀態非常驚艷"。 　　遮陽罩由五層 Kapton 製成。Kapton 是一種聚酰亞胺薄膜材料， 耐高溫絕...
張貼留言
閱讀完整內容