詳解Java線程同步器CountDownLatch
Java程序有的時(shí)候在主線程中會創(chuàng)建多個(gè)線程去執(zhí)行任務(wù),然后在主線程執(zhí)行完畢之前,把所有線程的任務(wù)進(jìn)行匯總,以前可以用線程的join方法,但是這個(gè)方法不夠靈活,我們可以使用CountDownLatch類,實(shí)現(xiàn)更優(yōu)雅,而且使用線程池的話,可沒有辦法調(diào)用線程的join方法的呀!
一.簡單使用CountDownLatch
直接使用線程:
package com.example.demo.study;import java.util.concurrent.CountDownLatch;public class Study0215 { //這里相當(dāng)于新建一個(gè)初始值為2的計(jì)數(shù)器 private static volatile CountDownLatch countDownLatch = new CountDownLatch(2); public static void main(String[] args) throws InterruptedException {new Thread(()->{ try {Thread.sleep(1000);System.out.println('線程一執(zhí)行完畢'); } catch (Exception e) { }finally {//每調(diào)用這個(gè)方法計(jì)數(shù)器減一countDownLatch.countDown(); } }).start();new Thread(()->{ try {Thread.sleep(1000);System.out.println('線程二執(zhí)行完畢'); } catch (Exception e) { }finally {countDownLatch.countDown(); } }).start();System.out.println('兩個(gè)線程已經(jīng)全部啟動'); //只要調(diào)用了這個(gè)方法之后,主線程會阻塞,直到計(jì)數(shù)器countDownLatch變成0就會返回 countDownLatch.await(); System.out.println('執(zhí)行完畢'); } }
實(shí)際中盡量少直接操作線程,而是使用線程池:
package com.example.demo.study;import java.util.concurrent.CountDownLatch;import java.util.concurrent.ExecutorService;import java.util.concurrent.Executors;public class Study0215 { // 這里相當(dāng)于新建一個(gè)初始值為2的計(jì)數(shù)器 private static volatile CountDownLatch countDownLatch = new CountDownLatch(2); public static void main(String[] args) throws InterruptedException { //創(chuàng)建線程池 ExecutorService pool = Executors.newFixedThreadPool(2); //將任務(wù)一丟進(jìn)線程池 pool.submit(() -> { try {Thread.sleep(1000);System.out.println('線程一執(zhí)行完畢'); } catch (Exception e) { } finally {// 每調(diào)用這個(gè)方法計(jì)數(shù)器減一countDownLatch.countDown(); } }); //任務(wù)二丟進(jìn)線程池 pool.submit(() -> { try {Thread.sleep(1000);System.out.println('線程二執(zhí)行完畢'); } catch (Exception e) { } finally {countDownLatch.countDown(); } }); System.out.println('兩個(gè)線程已經(jīng)全部啟動'); // 只要調(diào)用了這個(gè)方法之后,主線程會阻塞,直到計(jì)數(shù)器countDownLatch變成0就會返回 countDownLatch.await(); System.out.println('執(zhí)行完畢'); }}
二.await方法
看下面的圖,可以知道這個(gè)CountDownLatch類內(nèi)部有個(gè)工具類Sync實(shí)現(xiàn)了AQS,然后CountDownLatch中的方法都是調(diào)用工具類Sync去操作的,emmm....跟前面說過的ReentrantLock類結(jié)構(gòu)是一樣的;
我們看看CountDownLatch構(gòu)造器傳遞的數(shù)其實(shí)就是設(shè)置AQS中state的值:
//實(shí)際上調(diào)用把值傳遞給了Sync,也就是設(shè)置了AQS中的statepublic CountDownLatch(int count) { if (count < 0) throw new IllegalArgumentException('count < 0'); this.sync = new Sync(count);}Sync(int count) { setState(count);}
我們再看看await方法:
//當(dāng)前線程調(diào)用了await方法之后,當(dāng)前線程就會給阻塞,直到以下兩種情況://1.其他線程調(diào)用了countDown方法將計(jì)數(shù)器減到0之后,該線程就返回了;//2.其他線程調(diào)用了當(dāng)前的線程的中斷方法,當(dāng)前線程拋出異常InterruptedExceptionpublic void await() throws InterruptedException { sync.acquireSharedInterruptibly(1);}public final void acquireSharedInterruptibly(int arg) throws InterruptedException { //當(dāng)前線程被中斷就拋出異常 if (Thread.interrupted()) throw new InterruptedException(); //查看計(jì)數(shù)器中的值是不是0,不過不是0,就進(jìn)入AQS等待隊(duì)列等待; if (tryAcquireShared(arg) < 0) doAcquireSharedInterruptibly(arg);}protected int tryAcquireShared(int acquires) { return (getState() == 0) ? 1 : -1;}
三.countDown方法
public void countDown() { sync.releaseShared(1);}public final boolean releaseShared(int arg) { //tryReleaseShared方法返回false,說明當(dāng)前計(jì)數(shù)器的值減一成功 //返回true,說明計(jì)數(shù)器的值此時(shí)為0,那就要喚醒因?yàn)檎{(diào)用了CountDownLatch而阻塞的線程 if (tryReleaseShared(arg)) { doReleaseShared(); return true; } return false;}protected boolean tryReleaseShared(int releases) { //一個(gè)無限循環(huán) for (;;) { //獲取state的值 int c = getState(); //如果state為0,返回false if (c == 0) return false; //否則就把state減一然后用CAS更新到state int nextc = c-1; if (compareAndSetState(c, nextc)) return nextc == 0; }}
四.getState方法
這個(gè)方法獲取計(jì)數(shù)器的值,其實(shí)就是獲取AQS中的state的值;
int getCount() { return getState();}protected final int getState() { return state;}
其實(shí)CountDownLatch比較容易,功能和Thread的join方法一樣,只不過更靈活,基于AQS實(shí)現(xiàn),在初始化的時(shí)候設(shè)置state的值,當(dāng)線程調(diào)用CountDownLatch的await方法的時(shí)候,當(dāng)前線程就會被丟到AQS的阻塞隊(duì)列掛起;然后當(dāng)其他線程調(diào)用了countDown方法,其實(shí)就是將state減一,當(dāng)state等于0的時(shí)候,就會喚醒所有因?yàn)檎{(diào)用await方法而阻塞的線程;
以上就是詳解Java線程同步器CountDownLatch的詳細(xì)內(nèi)容,更多關(guān)于Java CountDownLatch的資料請關(guān)注好吧啦網(wǎng)其它相關(guān)文章!
相關(guān)文章:
1. moment轉(zhuǎn)化時(shí)間戳出現(xiàn)Invalid Date的問題及解決2. python爬蟲實(shí)戰(zhàn)之制作屬于自己的一個(gè)IP代理模塊3. python如何實(shí)現(xiàn)word批量轉(zhuǎn)HTML4. PHP如何打印跟蹤調(diào)試信息5. Java8內(nèi)存模型PermGen Metaspace實(shí)例解析6. 小技巧處理div內(nèi)容溢出7. 匹配模式 - XSL教程 - 48. HTML <!DOCTYPE> 標(biāo)簽9. 存儲于xml中需要的HTML轉(zhuǎn)義代碼10. CSS3實(shí)例分享之多重背景的實(shí)現(xiàn)(Multiple backgrounds)
