快捷導(dǎo)航

詳解Java線程同步器CountDownLatch

更新時(shí)間：2020年09月12日 10:31:47 作者：java小新人

這篇文章主要介紹了Java線程同步器CountDownLatch的相關(guān)資料，幫助大家更好的理解和學(xué)習(xí)Java，感興趣的朋友可以了解下

　　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)全部啟動(dò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)全部啟動(dò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中的state
public 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)前線程拋出異常InterruptedException
public 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)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: