簡談java并發(fā)FutureTask的實現(xiàn)

更新時間：2019年06月14日 15:03:28 作者：宸小朔

這篇文章主要介紹了簡談java并發(fā)FutureTask的實現(xiàn),FutureTask都是用于獲取線程執(zhí)行的返回結(jié)果。文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，,需要的朋友可以參考下

概述

在使用java多線程解決問題的時候，為了提高效率，我們常常會異步處理一些計算任務(wù)并在最后異步的獲取計算結(jié)果，這個過程的實現(xiàn)離不開Future接口及其實現(xiàn)類FutureTask。FutureTask類實現(xiàn)了Runnable, Future接口，接下來我會通過源碼對該類的實現(xiàn)進(jìn)行詳解。

使用

我們先看下FutureTask中的主要方法如下，可以看出FutureTask實現(xiàn)了任務(wù)及異步結(jié)果的集合功能?？吹竭@塊的方法，大家肯定會有疑問，Runnable任務(wù)的run方法返回空，F(xiàn)utureTask如何依靠該方法獲取線程異步執(zhí)行結(jié)果，這個問題，我們在下面給大家介紹。

//以下五個方法實現(xiàn)接口Future中方法
public boolean isCancelled(); 
public boolean isDone(); 
public boolean cancel();
public V get() throws InterruptedException, ExecutionException;
public V get(long timeout, TimeUnit unit);
//實現(xiàn)接口Runnable中方法
public void run();

我們在使用中會構(gòu)造一個FutureTask對象，然后將FutureTask扔到另一個線程中執(zhí)行，而主線程繼續(xù)執(zhí)行其他業(yè)務(wù)邏輯，一段時間后主線程調(diào)用FutureTask的get方法獲取執(zhí)行結(jié)果。下面我們看一個簡單的例子：

/**
* Created by yuanqiongqiong on 2019/4/9.
*/
public class FutureTaskTest {
private static ExecutorService executorService = Executors.newFixedThreadPool(1);
public static void main(String []args) {
Callable callable = new AccCallable(1, 2);
FutureTask futureTask = new FutureTask(callable);
executorService.execute(futureTask);
System.out.println("go to do other things in main thread");
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("go back in main thread");
try {
int result = (int) futureTask.get();
System.out.println("result is " + result);
} catch (InterruptedException e) {
e.printStackTrace();
} catch (ExecutionException e) {
e.printStackTrace();
}
}
static class AccCallable implements Callable<Integer> {
private int a;
private int b;
public AccCallable(int a, int b) {
this.a = a;
this.b = b;
}
@Override
public Integer call() throws Exception {
System.out.println("acc a and b in threadId = " + Thread.currentThread().getName());
return a + b;
}
}
}

輸出結(jié)果為：

go to do other things in main thread
acc a and b in threadId = pool-1-thread-1
go back in main thread
result is 3

實現(xiàn)分析

在分析實現(xiàn)前，我們先想下如果讓我們實現(xiàn)一個類似FutureTask的功能，我們會如何做？因為需要獲取執(zhí)行結(jié)果，需要一個Object對象來存執(zhí)行結(jié)果。任務(wù)執(zhí)行時間不可控性，我們需要一個變量表示執(zhí)行狀態(tài)。其他線程會調(diào)用get方法獲取結(jié)果，在沒達(dá)到超時的時候需要將線程阻塞或掛起。

因此需要一個隊列類似的結(jié)構(gòu)存儲等待該結(jié)果的線程信息，這樣在任務(wù)執(zhí)行線程完成后就可以喚醒這些阻塞或掛起的線程，得到結(jié)果。FutureTask的實際實現(xiàn)也是類似的邏輯，具體如下。

首先看下FutureTask的主要成員變量如下：

//futureTask執(zhí)行狀態(tài)
private volatile int state;
//具體的執(zhí)行任務(wù)，會在run方法中抵用callable.call()
private Callable<V> callable;
//執(zhí)行結(jié)果
private Object outcome; 
//獲取結(jié)果的等待線程節(jié)點
private volatile WaitNode waiters;

對于執(zhí)行狀態(tài)，在源碼中已經(jīng)有了非常清晰的解釋，這里我只是貼出源碼，不在進(jìn)行說明，具體如下：

/**
* Possible state transitions:
* NEW -> COMPLETING -> NORMAL
* NEW -> COMPLETING -> EXCEPTIONAL
* NEW -> CANCELLED
* NEW -> INTERRUPTING -> INTERRUPTED
*/
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;

然后我們看下FutureTask的構(gòu)造函數(shù)，如下：

public FutureTask(Callable<V> callable) {
if (callable == null)
throw new NullPointerException();
this.callable = callable;
this.state = NEW; // ensure visibility of callable
}
public FutureTask(Runnable runnable, V result) {
//構(gòu)造函數(shù)傳入runnable對象時調(diào)用靜態(tài)工具類Executors的方法轉(zhuǎn)換為一個callable對象
this.callable = Executors.callable(runnable, result);
this.state = NEW; // ensure visibility of callable
}

如前所述，F(xiàn)utureTask的執(zhí)行線程中會調(diào)用其run()方法執(zhí)行任務(wù)，我們看下這塊邏輯：

public void run() {
//1.如果執(zhí)行狀態(tài)不是NEW或者有其他線程執(zhí)行該任務(wù)，直接返回
if (state != NEW ||
!UNSAFE.compareAndSwapObject(this, runnerOffset,
null, Thread.currentThread()))
return;
try {
Callable<V> c = callable;
//2.如果執(zhí)行狀態(tài)是NEW,即任務(wù)還沒執(zhí)行，直接調(diào)用callable.call()方法獲取執(zhí)行結(jié)果
if (c != null && state == NEW) {
V result;
boolean ran;
try {
result = c.call();
ran = true;
} catch (Throwable ex) {
result = null;
ran = false;
//3.發(fā)生異常，更新status為EXCEPTIONAL，喚醒掛起線程
setException(ex);
}
//4.如果結(jié)果成功返回，調(diào)用set方法將設(shè)置outcome，更改status執(zhí)行狀態(tài)，喚醒掛起線程
if (ran)
set(result);
}
} finally {
// runner must be non-null until state is settled to
// prevent concurrent calls to run()
runner = null;
// state must be re-read after nulling runner to prevent
// leaked interrupts
int s = state;
if (s >= INTERRUPTING)
handlePossibleCancellationInterrupt(s);
}
}

我們看下set函數(shù)的實現(xiàn)，具體看下4中的執(zhí)行：

protected void set(V v) {
//將執(zhí)行狀態(tài)變更為COMPLETING
if (UNSAFE.compareAndSwapInt(this, stateOffset, NEW, COMPLETING)) {
//設(shè)置執(zhí)行結(jié)果
outcome = v;
//設(shè)置執(zhí)行狀態(tài)為NORMAL
UNSAFE.putOrderedInt(this, stateOffset, NORMAL); // final state
//執(zhí)行完成后處理操作，具體就是遍歷阻塞鏈表，刪除鏈表節(jié)點，并喚醒每個節(jié)點關(guān)聯(lián)的線程
finishCompletion();
}
}

以上就是任務(wù)執(zhí)行線程做的邏輯，以上邏輯也回答了FutureTask如何得到執(zhí)行結(jié)果的疑問。下面我們看下用戶調(diào)用get方法獲取執(zhí)行結(jié)果時的實現(xiàn)邏輯，這個時候FutureTask可能處理各種狀態(tài)，即可能沒有執(zhí)行，執(zhí)行中，已完成，發(fā)生異常等，具體如下：

public V get(long timeout, TimeUnit unit)
throws InterruptedException, ExecutionException, TimeoutException {
if (unit == null)
throw new NullPointerException();
int s = state;
//執(zhí)行狀態(tài)是NEW或者COMPLETING時執(zhí)行awaitDone將線程加入等待隊列中并掛起線程
if (s <= COMPLETING &&
(s = awaitDone(true, unit.toNanos(timeout))) <= COMPLETING)
throw new TimeoutException();
//根據(jù)執(zhí)行狀態(tài)status進(jìn)行結(jié)果封裝
return report(s);
}
//我理解這塊是get的核心邏輯
private int awaitDone(boolean timed, long nanos)
throws InterruptedException {
//如果設(shè)置了超時時間，計算還有多長時間超時
final long deadline = timed ? System.nanoTime() + nanos : 0L;
WaitNode q = null;
boolean queued = false;
for (;;) {
//如果當(dāng)前線程被中斷，刪除等待隊列中的節(jié)點，并拋出異常
if (Thread.interrupted()) {
removeWaiter(q);
throw new InterruptedException();
}
int s = state;
//如果執(zhí)行狀態(tài)已經(jīng)完成或者發(fā)生異常，直接跳出自旋返回
if (s > COMPLETING) {
if (q != null)
q.thread = null;
return s;
}
//如果執(zhí)行狀態(tài)是正在執(zhí)行，說明線程已經(jīng)被加入到等待隊列中，放棄cpu進(jìn)入下次循環(huán)(真正的自旋)
else if (s == COMPLETING) // cannot time out yet
Thread.yield();
//第一次進(jìn)入循環(huán)，創(chuàng)建節(jié)點
else if (q == null)
q = new WaitNode();
//將節(jié)點加入到等待隊列中，waiters相當(dāng)于頭階段，不斷將頭結(jié)點更新為新節(jié)點
else if (!queued)
queued = UNSAFE.compareAndSwapObject(this, waitersOffset,
q.next = waiters, q);
else if (timed) {
//如果設(shè)置了超時時間，在進(jìn)行下次循環(huán)前查看是否已經(jīng)超時，如果超時刪除該節(jié)點進(jìn)行返回
nanos = deadline - System.nanoTime();
if (nanos <= 0L) {
removeWaiter(q);
return state;
}
//掛起當(dāng)前節(jié)點
LockSupport.parkNanos(this, nanos);
}
else
LockSupport.park(this);
}
}

這里需要說明一點，F(xiàn)utureTask中的阻塞隊列新加入的節(jié)點都在頭結(jié)點并且next指向之前的頭結(jié)點，waitars指針總是指向新加入節(jié)點，通過waitars可以遍歷整個等待隊列，具體截圖如下。此外等待隊列節(jié)點結(jié)構(gòu)很簡單成員變量只有線程引用和next指針，這里再列出器接口。