1. 程式人生 > 程式設計 >Java CountDownLatch應用場景程式碼例項

Java CountDownLatch應用場景程式碼例項

Java的concurrent包裡面的CountDownLatch其實可以把它看作一個計數器,只不過這個計數器的操作是原子操作,同時只能有一個執行緒去操作這個計數器,也就是同時只能有一個執行緒去減這個計數器裡面的值。

你可以向CountDownLatch物件設定一個初始的數字作為計數值,任何呼叫這個物件上的await()方法都會阻塞,直到這個計數器的計數值被其他的執行緒減為0為止。

CountDownLatch的一個非常典型的應用場景是:有一個任務想要往下執行,但必須要等到其他的任務執行完畢後才可以繼續往下執行。假如我們這個想要繼續往下執行的任務呼叫一個CountDownLatch物件的await()方法,其他的任務執行完自己的任務後呼叫同一個CountDownLatch物件上的countDown()方法,這個呼叫await()方法的任務將一直阻塞等待,直到這個CountDownLatch物件的計數值減到0為止。

舉個例子,有三個工人在為老闆幹活,這個老闆有一個習慣,就是當三個工人把一天的活都幹完了的時候,他就來檢查所有工人所幹的活。記住這個條件:三個工人先全部幹完活,老闆才檢查。所以在這裡用Java程式碼設計兩個類,Worker代表工人,Boss代表老闆,具體的程式碼實現如下:

Java程式碼

package org.zapldy.concurrent; 
 
import java.util.Random; 
import java.util.concurrent.CountDownLatch; 
import java.util.concurrent.TimeUnit; 
 
public class Worker implements Runnable{ 
   
  private CountDownLatch downLatch; 
  private String name; 
   
  public Worker(CountDownLatch downLatch,String name){ 
    this.downLatch = downLatch; 
    this.name = name; 
  } 
   
  public void run() { 
    this.doWork(); 
    try{ 
      TimeUnit.SECONDS.sleep(new Random().nextInt(10)); 
    }catch(InterruptedException ie){ 
    } 
    System.out.println(this.name + "活幹完了!"); 
    this.downLatch.countDown(); 
     
  } 
   
  private void doWork(){ 
    System.out.println(this.name + "正在幹活!"); 
  } 
   
}

Java程式碼

package org.zapldy.concurrent; 
 
import java.util.concurrent.CountDownLatch; 
 
public class Boss implements Runnable { 
 
  private CountDownLatch downLatch; 
   
  public Boss(CountDownLatch downLatch){ 
    this.downLatch = downLatch; 
  } 
   
  public void run() { 
    System.out.println("老闆正在等所有的工人幹完活......"); 
    try { 
      this.downLatch.await(); 
    } catch (InterruptedException e) { 
    } 
    System.out.println("工人活都幹完了,老闆開始檢查了!"); 
  } 
 
} 

Java程式碼

package org.zapldy.concurrent;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CountDownLatchDemo {

	public static void main(String[] args) {
		ExecutorService executor = Executors.newCachedThreadPool();
		
		CountDownLatch latch = new CountDownLatch(3);
		
		Worker w1 = new Worker(latch,"張三");
		Worker w2 = new Worker(latch,"李四");
		Worker w3 = new Worker(latch,"王二");
		
		Boss boss = new Boss(latch);
		
		executor.execute(w3);
		executor.execute(w2);
		executor.execute(w1);
		executor.execute(boss);
		
		executor.shutdown();
	}

}

當你執行CountDownLatchDemo這個物件的時候,你會發現是等所有的工人都幹完了活,老闆才來檢查,下面是我本地機器上執行的一次結果,可以肯定的每次執行的結果可能與下面不一樣,但老闆檢查永遠是在後面的。

王二正在幹活!
李四正在幹活!
老闆正在等所有的工人幹完活......
張三正在幹活!
張三活幹完了!
王二活幹完了!
李四活幹完了!
工人活都幹完了,老闆開始檢查了!

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。