問題

(1)Phaser是什么？

(2)Phaser具有哪些特性？

(3)Phaser相對于CyclicBarrier和CountDownLatch的優勢？

簡介

Phaser，翻譯為階段，它適用于這樣一種場景，一個大任務可以分為多個階段完成，且每個階段的任務可以多個線程并發執行，但是必須上一個階段的任務都完成了才可以執行下一個階段的任務。

這種場景雖然使用CyclicBarrier或者CountryDownLatch也可以實現，但是要復雜的多。首先，具體需要多少個階段是可能會變的，其次，每個階段的任務數也可能會變的。相比于CyclicBarrier和CountDownLatch，Phaser更加靈活更加方便。

使用方法

下面我們看一個最簡單的使用案例：

public class PhaserTest {

public static final int PARTIES = 3;

public static final int PHASES = 4;

public static void main(String[] args) {

Phaser phaser = new Phaser(PARTIES) {

@Override

protected boolean onAdvance(int phase, int registeredParties) {

// 【本篇文章由公眾號“彤哥讀源碼”原創，請支持原創，謝謝！】

System.out.println("=======phase: " + phase + " finished=============");

return super.onAdvance(phase, registeredParties);

}

};

for (int i = 0; i < PARTIES; i++) {

new Thread(()->{

for (int j = 0; j < PHASES; j++) {

System.out.println(String.format("%s: phase: %d", Thread.currentThread().getName(), j));

phaser.arriveAndAwaitAdvance();

}

}, "Thread " + i).start();

}

}

}

這里我們定義一個需要4個階段完成的大任務，每個階段需要3個小任務，針對這些小任務，我們分別起3個線程來執行這些小任務，查看輸出結果為：

Thread 0: phase: 0

Thread 2: phase: 0

Thread 1: phase: 0

=======phase: 0 finished=============

Thread 2: phase: 1

Thread 0: phase: 1

Thread 1: phase: 1

=======phase: 1 finished=============

Thread 1: phase: 2

Thread 0: phase: 2

Thread 2: phase: 2

=======phase: 2 finished=============

Thread 0: phase: 3

Thread 2: phase: 3

Thread 1: phase: 3

=======phase: 3 finished=============

可以看到，每個階段都是三個線程都完成了才進入下一個階段。這是怎么實現的呢，讓我們一起來學習吧。

原理猜測

根據我們前面學習AQS的原理，大概猜測一下Phaser的實現原理。

首先，需要存儲當前階段phase、當前階段的任務數(參與者)parties、未完成參與者的數量，這三個變量我們可以放在一個變量state中存儲。

其次，需要一個隊列存儲先完成的參與者，當最后一個參與者完成任務時，需要喚醒隊列中的參與者。

嗯，差不多就是這樣子。

結合上面的案例帶入：

初始時當前階段為0，參與者數為3個，未完成參與者數為3；

第一個線程執行到phaser.arriveAndAwaitAdvance();時進入隊列；

第二個線程執行到phaser.arriveAndAwaitAdvance();時進入隊列；

第三個線程執行到phaser.arriveAndAwaitAdvance();時先執行這個階段的總結onAdvance()，再喚醒前面兩個線程繼續執行下一個階段的任務。

嗯，整體能說得通，至于是不是這樣呢，讓我們一起來看源碼吧。

源碼分析

主要內部類

static final class QNode implements ForkJoinPool.ManagedBlocker {

final Phaser phaser;

final int phase;

final boolean interruptible;

final boolean timed;

boolean wasInterrupted;

long nanos;

final long deadline;

volatile Thread thread; // nulled to cancel wait

QNode next;

QNode(Phaser phaser, int phase, boolean interruptible,

boolean timed, long nanos) {

this.phaser = phaser;

this.phase = phase;

this.interruptible = interruptible;

this.nanos = nanos;

this.timed = timed;

this.deadline = timed ? System.nanoTime() + nanos : 0L;

thread = Thread.currentThread();

}

}

先完成的參與者放入隊列中的節點，這里我們只需要關注thread和next兩個屬性即可，很明顯這是一個單鏈表，存儲著入隊的線程。

主要屬性

// 狀態變量，用于存儲當前階段phase、參與者數parties、未完成的參與者數unarrived_count

private volatile long state;

// 最多可以有多少個參與者，即每個階段最多有多少個任務

private static final int MAX_PARTIES = 0xffff;

// 最多可以有多少階段

private static final int MAX_PHASE = Integer.MAX_VALUE;

// 參與者數量的偏移量

private static final int PARTIES_SHIFT = 16;

// 當前階段的偏移量

private static final int PHASE_SHIFT = 32;

// 未完成的參與者數的掩碼，低16位

private static final int UNARRIVED_MASK = 0xffff; // to mask ints

// 參與者數，中間16位

private static final long PARTIES_MASK = 0xffff0000L; // to mask longs

// counts的掩碼，counts等于參與者數和未完成的參與者數的'|'操作

private static final long COUNTS_MASK = 0xffffffffL;

private static final long TERMINATION_BIT = 1L << 63;

// 一次一個參與者完成

private static final int ONE_ARRIVAL = 1;

// 增加減少參與者時使用

private static final int ONE_PARTY = 1 << PARTIES_SHIFT;

// 減少參與者時使用

private static final int ONE_DEREGISTER = ONE_ARRIVAL|ONE_PARTY;

// 沒有參與者時使用

private static final int EMPTY = 1;

// 用于求未完成參與者數量

private static int unarrivedOf(long s) {

int counts = (int)s;

return (counts == EMPTY) ? 0 : (counts & UNARRIVED_MASK);

}

// 用于求參與者數量(中間16位)，注意int的位置

private static int partiesOf(long s) {

return (int)s >>> PARTIES_SHIFT;

}

// 用于求階段數(高32位)，注意int的位置

private static int phaseOf(long s) {

return (int)(s >>> PHASE_SHIFT);

}

// 已完成參與者的數量

private static int arrivedOf(long s) {

int counts = (int)s; // 低32位

return (counts == EMPTY) ? 0 :

(counts >>> PARTIES_SHIFT) - (counts & UNARRIVED_MASK);

}

// 用于存儲已完成參與者所在的線程，根據當前階段的奇偶性選擇不同的隊列

private final AtomicReference evenQ;

private final AtomicReference oddQ;

主要屬性為state和evenQ及oddQ：

(1)state，狀態變量，高32位存儲當前階段phase，中間16位存儲參與者的數量，低16位存儲未完成參與者的數量【本篇文章由公眾號“彤哥讀源碼”原創，請支持原創，謝謝！】；

(2)evenQ和oddQ，已完成的參與者存儲的隊列，當最后一個參與者完成任務后喚醒隊列中的參與者繼續執行下一個階段的任務，或者結束任務。

構造方法

public Phaser() {

this(null, 0);

}

public Phaser(int parties) {

this(null, parties);

}

public Phaser(Phaser parent) {

this(parent, 0);

}

public Phaser(Phaser parent, int parties) {

if (parties >>> PARTIES_SHIFT != 0)

throw new IllegalArgumentException("Illegal number of parties");

int phase = 0;

this.parent = parent;

if (parent != null) {

final Phaser root = parent.root;

this.root = root;

this.evenQ = root.evenQ;

this.oddQ = root.oddQ;

if (parties != 0)

phase = parent.doRegister(1);

}

else {

this.root = this;

this.evenQ = new AtomicReference();

this.oddQ = new AtomicReference();

}

// 狀態變量state的存儲分為三段

this.state = (parties == 0) ? (long)EMPTY :

((long)phase << PHASE_SHIFT) |

((long)parties << PARTIES_SHIFT) |

((long)parties);

}

構造函數中還有一個parent和root，這是用來構造多層級階段的，不在本文的討論范圍之內，忽略之。

重點還是看state的賦值方式，高32位存儲當前階段phase，中間16位存儲參與者的數量，低16位存儲未完成參與者的數量。

下面我們一起來看看幾個主要方法的源碼：

register()方法

注冊一個參與者，如果調用該方法時，onAdvance()方法正在執行，則該方法等待其執行完畢。

public int register() {

return doRegister(1);

}

private int doRegister(int registrations) {

// state應該加的值，注意這里是相當于同時增加parties和unarrived

long adjust = ((long)registrations << PARTIES_SHIFT) | registrations;

final Phaser parent = this.parent;

int phase;

for (;;) {

// state的值

long s = (parent == null) ? state : reconcileState();

// state的低32位，也就是parties和unarrived的值

int counts = (int)s;

// parties的值

int parties = counts >>> PARTIES_SHIFT;

// unarrived的值

int unarrived = counts & UNARRIVED_MASK;

// 檢查是否溢出

if (registrations > MAX_PARTIES - parties)

throw new IllegalStateException(badRegister(s));

// 當前階段phase

phase = (int)(s >>> PHASE_SHIFT);

if (phase < 0)

break;

// 不是第一個參與者

if (counts != EMPTY) { // not 1st registration

if (parent == null || reconcileState() == s) {

// unarrived等于0說明當前階段正在執行onAdvance()方法，等待其執行完畢

if (unarrived == 0) // wait out advance

root.internalAwaitAdvance(phase, null);

// 否則就修改state的值，增加adjust，如果成功就跳出循環

else if (UNSAFE.compareAndSwapLong(this, stateOffset,

s, s + adjust))

break;

}

}

// 是第一個參與者

else if (parent == null) { // 1st root registration

// 計算state的值

long next = ((long)phase << PHASE_SHIFT) | adjust;

// 修改state的值，如果成功就跳出循環

if (UNSAFE.compareAndSwapLong(this, stateOffset, s, next))

break;

}

else {

// 多層級階段的處理方式

synchronized (this) { // 1st sub registration

if (state == s) { // recheck under lock

phase = parent.doRegister(1);

if (phase < 0)

break;

// finish registration whenever parent registration

// succeeded, even when racing with termination,

// since these are part of the same "transaction".

while (!UNSAFE.compareAndSwapLong

(this, stateOffset, s,

((long)phase << PHASE_SHIFT) | adjust)) {

s = state;

phase = (int)(root.state >>> PHASE_SHIFT);

// assert (int)s == EMPTY;

}

break;

}

}

}

}

return phase;

}

// 等待onAdvance()方法執行完畢

// 原理是先自旋一定次數，如果進入下一個階段，這個方法直接就返回了，

// 如果自旋一定次數后還沒有進入下一個階段，則當前線程入隊列，等待onAdvance()執行完畢喚醒

private int internalAwaitAdvance(int phase, QNode node) {

// 保證隊列為空

releaseWaiters(phase-1); // ensure old queue clean

boolean queued = false; // true when node is enqueued

int lastUnarrived = 0; // to increase spins upon change

// 自旋的次數

int spins = SPINS_PER_ARRIVAL;

long s;

int p;

// 檢查當前階段是否變化，如果變化了說明進入下一個階段了，這時候就沒有必要自旋了

while ((p = (int)((s = state) >>> PHASE_SHIFT)) == phase) {

// 如果node為空，注冊的時候傳入的為空

if (node == null) { // spinning in noninterruptible mode

// 未完成的參與者數量

int unarrived = (int)s & UNARRIVED_MASK;

// unarrived有變化，增加自旋次數

if (unarrived != lastUnarrived &&

(lastUnarrived = unarrived) < NCPU)

spins += SPINS_PER_ARRIVAL;

boolean interrupted = Thread.interrupted();

// 自旋次數完了，則新建一個節點

if (interrupted || --spins < 0) { // need node to record intr

node = new QNode(this, phase, false, false, 0L);

node.wasInterrupted = interrupted;

}

}

else if (node.isReleasable()) // done or aborted

break;

else if (!queued) { // push onto queue

// 節點入隊列

AtomicReference head = (phase & 1) == 0 ? evenQ : oddQ;

QNode q = node.next = head.get();

if ((q == null || q.phase == phase) &&

(int)(state >>> PHASE_SHIFT) == phase) // avoid stale enq

queued = head.compareAndSet(q, node);

}

else {

try {

// 當前線程進入阻塞狀態，跟調用LockSupport.park()一樣，等待被喚醒

ForkJoinPool.managedBlock(node);

} catch (InterruptedException ie) {

node.wasInterrupted = true;

}

}

}

// 到這里說明節點所在線程已經被喚醒了

if (node != null) {

// 置空節點中的線程

if (node.thread != null)

node.thread = null; // avoid need for unpark()

if (node.wasInterrupted && !node.interruptible)

Thread.currentThread().interrupt();

if (p == phase && (p = (int)(state >>> PHASE_SHIFT)) == phase)

return abortWait(phase); // possibly clean up on abort

}

// 喚醒當前階段阻塞著的線程

releaseWaiters(phase);

return p;

}

增加一個參與者總體的邏輯為：

(1)增加一個參與者，需要同時增加parties和unarrived兩個數值，也就是state的中16位和低16位；

(2)如果是第一個參與者，則嘗試原子更新state的值，如果成功了就退出；

(3)如果不是第一個參與者，則檢查是不是在執行onAdvance()，如果是等待onAdvance()執行完成，如果否則嘗試原子更新state的值，直到成功退出；

(4)等待onAdvance()完成是采用先自旋后進入隊列排隊的方式等待，減少線程上下文切換；

arriveAndAwaitAdvance()方法

當前線程當前階段執行完畢，等待其它線程完成當前階段。

如果當前線程是該階段最后一個到達的，則當前線程會執行onAdvance()方法，并喚醒其它線程進入下一個階段。

public int arriveAndAwaitAdvance() {

// Specialization of doArrive+awaitAdvance eliminating some reads/paths

final Phaser root = this.root;

for (;;) {

// state的值

long s = (root == this) ? state : reconcileState();

// 當前階段

int phase = (int)(s >>> PHASE_SHIFT);

if (phase < 0)

return phase;

// parties和unarrived的值

int counts = (int)s;

// unarrived的值(state的低16位)

int unarrived = (counts == EMPTY) ? 0 : (counts & UNARRIVED_MASK);

if (unarrived <= 0)

throw new IllegalStateException(badArrive(s));

// 修改state的值

if (UNSAFE.compareAndSwapLong(this, stateOffset, s,

s -= ONE_ARRIVAL)) {

// 如果不是最后一個到達的，則調用internalAwaitAdvance()方法自旋或進入隊列等待

if (unarrived > 1)

// 這里是直接返回了，internalAwaitAdvance()方法的源碼見register()方法解析

return root.internalAwaitAdvance(phase, null);

// 到這里說明是最后一個到達的參與者

if (root != this)

return parent.arriveAndAwaitAdvance();

// n只保留了state中parties的部分，也就是中16位

long n = s & PARTIES_MASK; // base of next state

// parties的值，即下一次需要到達的參與者數量

int nextUnarrived = (int)n >>> PARTIES_SHIFT;

// 執行onAdvance()方法，返回true表示下一階段參與者數量為0了，也就是結束了

if (onAdvance(phase, nextUnarrived))

n |= TERMINATION_BIT;

else if (nextUnarrived == 0)

n |= EMPTY;

else

// n 加上unarrived的值

n |= nextUnarrived;

// 下一個階段等待當前階段加1

int nextPhase = (phase + 1) & MAX_PHASE;

// n 加上下一階段的值

n |= (long)nextPhase << PHASE_SHIFT;

// 修改state的值為n

if (!UNSAFE.compareAndSwapLong(this, stateOffset, s, n))

return (int)(state >>> PHASE_SHIFT); // terminated

// 喚醒其它參與者并進入下一個階段

releaseWaiters(phase);

// 返回下一階段的值

return nextPhase;

}

}

}

arriveAndAwaitAdvance的大致邏輯為：

(1)修改state中unarrived部分的值減1；

(2)如果不是最后一個到達的，則調用internalAwaitAdvance()方法自旋或排隊等待；

(3)如果是最后一個到達的，則調用onAdvance()方法，然后修改state的值為下一階段對應的值，并喚醒其它等待的線程；

(4)返回下一階段的值；

總結

(1)Phaser適用于多階段多任務的場景，每個階段的任務都可以控制得很細；

(2)Phaser內部使用state變量及隊列實現整個邏輯【本篇文章由公眾號“彤哥讀源碼”原創，請支持原創，謝謝！】；

(3)state的高32位存儲當前階段phase，中16位存儲當前階段參與者(任務)的數量parties，低16位存儲未完成參與者的數量unarrived；

(4)隊列會根據當前階段的奇偶性選擇不同的隊列；

(5)當不是最后一個參與者到達時，會自旋或者進入隊列排隊來等待所有參與者完成任務；

(6)當最后一個參與者完成任務時，會喚醒隊列中的線程并進入下一個階段；

彩蛋

Phaser相對于CyclicBarrier和CountDownLatch的優勢？

答：優勢主要有兩點：

(1)Phaser可以完成多階段，而一個CyclicBarrier或者CountDownLatch一般只能控制一到兩個階段的任務；

(2)Phaser每個階段的任務數量可以控制，而一個CyclicBarrier或者CountDownLatch任務數量一旦確定不可修改。

推薦閱讀

歡迎關注我的公眾號“彤哥讀源碼”，查看更多源碼系列文章, 與彤哥一起暢游源碼的海洋。

總結

以上是生活随笔為你收集整理的phaser java_死磕 java同步系列之Phaser源码解析的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。