CountDownLatch源碼分析

概述

CountDownLatch可以翻譯為倒計(jì)數(shù)器,讓主調(diào)用線程等待其他一些線程工作完成后纳账,再繼續(xù)運(yùn)行绩蜻。(相當(dāng)于調(diào)用所有子線程join方法的效果)
主要有兩種使用場景:
第一種設(shè)置兩個(gè)信號厚掷,一個(gè)是啟動(dòng)信號礁苗,當(dāng)控制線程發(fā)出信號以后,所有線程才開始工作徙缴,一個(gè)是全部完成的信號试伙,當(dāng)所有工作線程完成后,控制線程才繼續(xù)工作于样。
另一個(gè)典型用法是將問題分為N個(gè)部分疏叨,每一部分用子線程跑,然后在鎖存器上遞減計(jì)數(shù)穿剖,當(dāng)所有子線程都完成后蚤蔓,調(diào)用線程將能夠繼續(xù)運(yùn)行。

例子

public static void main(String[] args) throws InterruptedException {
    final int N = 3;
    CountDownLatch startSignal = new CountDownLatch(1);
    CountDownLatch doneSignal = new CountDownLatch(N);

    for (int i = 0; i < N; ++i)
        new Thread(new Worker(startSignal, doneSignal)).start();
    doSomethingElse();            
    startSignal.countDown();      
    doSomethingElse();
    doneSignal.await();
    System.out.println(currentThread().getName() + ": end");
}

private static void doSomethingElse() {
    System.out.println(currentThread().getName() + ": do something");
}

static class Worker implements Runnable {
    private final CountDownLatch startSignal;
    private final CountDownLatch doneSignal;

    public Worker(CountDownLatch startSignal, CountDownLatch doneSignal) {
        this.startSignal = startSignal;
        this.doneSignal = doneSignal;
    }

    @Override
    public void run() {
        try {
            startSignal.await();
            doWork();
            System.out.println(currentThread().getName() + ": finish work");
            doneSignal.countDown();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

    }

    private void doWork() {
        System.out.println(currentThread().getName() + ": is working");
    }
}

運(yùn)行結(jié)果:

main: do something
main: do something
Thread-1: is working
Thread-0: is working
Thread-2: is working
Thread-2: finish work
Thread-0: finish work
Thread-1: finish work
main: end

例子github地址
主線程設(shè)置了兩個(gè)信號量糊余,startSignal秀又,doneSignal
startSignal設(shè)置的state為1,doneSignal設(shè)置的state為N需要等待的線程量
當(dāng)主線程在main函數(shù)中調(diào)用startSignal的countDown方法后贬芥,所有的子線程就可以開始做自己的事了吐辙,當(dāng)主線程調(diào)用doneSignal.await,開始等待蘸劈,知道所有子線程完成doneSignal.countDown后昏苏,主線程繼續(xù)執(zhí)行。
可以想像成一個(gè)體育老師千米測試威沫,所有學(xué)生準(zhǔn)備就緒贤惯,等老師發(fā)出開始信號startSignal.countDown,學(xué)生們都甩開膀子跑了棒掠,老師走的千米測試的終點(diǎn)后孵构,調(diào)用doneSignal.await,這時(shí)老師就要等所有學(xué)生全部跑完后烟很,才能下課浦译,開始自己的生活了。

CountDownLatch源碼

CountDownLatch使用的是AQS的共享式占用鎖溯职,定義了一個(gè)內(nèi)部類Sync繼承AQS精盅,不清楚AQS源碼的需要先看下這篇文章

//只有一個(gè)入?yún)⑹莍nt的構(gòu)造函數(shù),用來初始化Sync同步器
public CountDownLatch(int count) {
    if (count < 0) throw new IllegalArgumentException("count < 0");
    this.sync = new Sync(count);
}
private static final class Sync extends AbstractQueuedSynchronizer {
    private static final long serialVersionUID = 4982264981922014374L;

    Sync(int count) {
        setState(count);
    }

    int getCount() {
        return getState();
    }

    //重寫了父類嘗試請求加鎖的方法谜酒,判斷只有state為0叹俏,返回正數(shù),即請求加鎖成功
    protected int tryAcquireShared(int acquires) {
        return (getState() == 0) ? 1 : -1;
    }

    protected boolean tryReleaseShared(int releases) {
        // 無限循環(huán)CAS操作state的值
        for (;;) {
            int c = getState();
            //說明已經(jīng)釋放過了僻族,不需要再喚醒了
            if (c == 0)
                return false;
            int nextc = c-1;
            //當(dāng)state值減為0以后粘驰,共享鎖釋放屡谐,喚起等待節(jié)點(diǎn)
            if (compareAndSetState(c, nextc))
                return nextc == 0;
        }
    }
}
public void await() throws InterruptedException {
    sync.acquireSharedInterruptibly(1);
}
public final void acquireSharedInterruptibly(int arg)
        throws InterruptedException {
    if (Thread.interrupted())
        throw new InterruptedException();
    //調(diào)用上面的嘗試加鎖方法
    if (tryAcquireShared(arg) < 0)
        doAcquireSharedInterruptibly(arg);
}

await方法就是判斷state的值,如果state值不為0蝌数,加鎖失敗愕掏,進(jìn)入排隊(duì)等待狀態(tài)。等待其他線程釋放鎖顶伞。

public void countDown() {
    sync.releaseShared(1);
}

countDown方法比較簡單就是對state減1饵撑,當(dāng)state等于0,喚起等待的線程唆貌。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末滑潘,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子锨咙,更是在濱河造成了極大的恐慌语卤,老刑警劉巖,帶你破解...
    沈念sama閱讀 212,884評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件酪刀,死亡現(xiàn)場離奇詭異粹舵,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)骂倘,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,755評論 3 385
  • 文/潘曉璐 我一進(jìn)店門齐婴,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人稠茂,你說我怎么就攤上這事柠偶。” “怎么了睬关?”我有些...
    開封第一講書人閱讀 158,369評論 0 348
  • 文/不壞的土叔 我叫張陵诱担,是天一觀的道長。 經(jīng)常有香客問我电爹,道長蔫仙,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,799評論 1 285
  • 正文 為了忘掉前任丐箩,我火速辦了婚禮摇邦,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘屎勘。我一直安慰自己施籍,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,910評論 6 386
  • 文/花漫 我一把揭開白布概漱。 她就那樣靜靜地躺著丑慎,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上竿裂,一...
    開封第一講書人閱讀 50,096評論 1 291
  • 那天玉吁,我揣著相機(jī)與錄音,去河邊找鬼腻异。 笑死进副,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的悔常。 我是一名探鬼主播影斑,決...
    沈念sama閱讀 39,159評論 3 411
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼这嚣!你這毒婦竟也來了鸥昏?” 一聲冷哼從身側(cè)響起塞俱,我...
    開封第一講書人閱讀 37,917評論 0 268
  • 序言:老撾萬榮一對情侶失蹤姐帚,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后障涯,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體罐旗,經(jīng)...
    沈念sama閱讀 44,360評論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,673評論 2 327
  • 正文 我和宋清朗相戀三年唯蝶,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了九秀。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,814評論 1 341
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡粘我,死狀恐怖鼓蜒,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情征字,我是刑警寧澤都弹,帶...
    沈念sama閱讀 34,509評論 4 334
  • 正文 年R本政府宣布,位于F島的核電站匙姜,受9級特大地震影響畅厢,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜氮昧,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 40,156評論 3 317
  • 文/蒙蒙 一框杜、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧袖肥,春花似錦咪辱、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,882評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春选调,著一層夾襖步出監(jiān)牢的瞬間夹供,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,123評論 1 267
  • 我被黑心中介騙來泰國打工仁堪, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留哮洽,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 46,641評論 2 362
  • 正文 我出身青樓弦聂,卻偏偏與公主長得像鸟辅,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個(gè)殘疾皇子莺葫,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,728評論 2 351