CountDownLatch、CyclicBarrier、Semaphore 用法總結

CountDownLatch

  • CountDownLatch 類位於 java.util.concurrent 包下,利用它能夠實現相似計數器的功能。好比有一個任務A,它要等待其餘4個任務執行完畢以後才能執行,此時就能夠利用CountDownLatch來實現這種功能了。
  • CountDownLatch類只提供了一個構造器:
public CountDownLatch(int count) {  };  //參數count爲計數值
  • 而後下面這3個方法是CountDownLatch類中最重要的方法:
public void await() throws InterruptedException { };   //調用await()方法的線程會被掛起,它會等待直到count值爲0才繼續執行
public boolean await(long timeout, TimeUnit unit) throws InterruptedException { };  //和await()相似,只不過等待必定的時間後count值還沒變爲0的話就會繼續執行
public void countDown() { };  //將count值減1
  • 代碼實現
package sychronized;

import static net.mindview.util.Print.*;
import java.util.concurrent.*;

class Task implements Runnable{
    private static int count = 0;
    private final int id = count++;
    final CountDownLatch latch ;
    public Task(CountDownLatch latch){
        this.latch = latch;
    }
    
    @Override
    public void run(){
        try {
            print(this+"正在執行");
            TimeUnit.MILLISECONDS.sleep(3000);
            print(this+"執行完畢");
            latch.countDown();
        } catch (InterruptedException e) {
            print(this + " 被中斷");
        }
    }

    @Override
    public String toString() {
        return "Task-"+id;
    }
}

public class Test {
    public static void main(String[] args) {
        final CountDownLatch latch = new CountDownLatch(2);
        ExecutorService exec = Executors.newCachedThreadPool();
        
        exec.execute(new Task(latch));
        exec.execute(new Task(latch));
        
        try {
            print("等待2個子線程執行完畢...");
            long start = System.currentTimeMillis();
            latch.await();
            long end = System.currentTimeMillis();
            
            print("2個子線程已經執行完畢 "+(end - start));
            print("繼續執行主線程");
        }catch (InterruptedException e){
            print("主線程被中斷");
        }
        exec.shutdown();
    }
}

#輸出結果:
等待2個子線程執行完畢...
Task-0正在執行
Task-1正在執行
Task-0執行完畢
Task-1執行完畢
2個子線程已經執行完畢 3049
繼續執行主線程

CyclicBarrier

  • 字面意思迴環柵欄,經過它能夠實現讓一組線程等待至某個狀態以後再所有同時執行。叫作迴環是由於當全部等待線程都被釋放之後,CyclicBarrier能夠被重用。咱們暫且把這個狀態就叫作barrier,當調用await()方法以後,線程就處於barrier了。
  • CyclicBarrier類位於java.util.concurrent包下,CyclicBarrier提供2個構造器:
  • 參數parties指讓多少個線程或者任務等待至barrier狀態
  • 參數barrierAction爲當這些線程都達到barrier狀態時會執行的內容
public CyclicBarrier(int parties, Runnable barrierAction) {}
public CyclicBarrier(int parties) {}
  • 而後CyclicBarrier中最重要的方法就是 await 方法,它有2個重載版本:
  • 第一個版本比較經常使用,用來掛起當前線程,直至全部線程都到達barrier狀態再同時執行後續任務;
  • 第二個版本是讓這些線程等待至必定的時間,若是還有線程沒有到達barrier狀態就直接讓到達barrier的線程執行後續任務。
public int await() throws InterruptedException, BrokenBarrierException { };
public int await(long timeout, TimeUnit unit)throws InterruptedException,BrokenBarrierException,TimeoutException { };
  • 代碼展現
package sychronized;

import java.util.Random;
import java.util.concurrent.*;
import static net.mindview.util.Print.*;

class WriteTask implements Runnable{
    private static int count = 0;
    private final int id = count++;
    private CyclicBarrier barrier ;
    private static Random random = new Random(47);
    public WriteTask(CyclicBarrier cyclicBarrier) {
        this.barrier = cyclicBarrier;
    }
    
    @Override
    public void run() {
        print(this+"開始寫入數據...");
        try {
            TimeUnit.MILLISECONDS.sleep(random.nextInt(5000));      //以睡眠來模擬寫入數據操做
            print(this+"寫入數據完畢,等待其餘線程寫入完畢"+" "+System.currentTimeMillis());
            barrier.await();
        } catch (InterruptedException e) {
            print(this + "is interrupted!");
        }catch(BrokenBarrierException e){
           throw new RuntimeException(e);
        }
        print("全部任務寫入完畢,繼續處理其餘任務... "+System.currentTimeMillis());
    }

    @Override
    public String toString() {
        return getClass().getSimpleName()+"-"+id;
    }
}

public class CyclicBarrierTest {
    public static void main(String[] args) {
        int N = 4;
        CyclicBarrier barrier  = new CyclicBarrier(N);
        ExecutorService exec = Executors.newCachedThreadPool();
        for(int i = 0; i < N; ++i){
            exec.execute(new WriteTask(barrier));
        }
        exec.shutdown();
    }
}

#輸出結果:
WriteTask-3 開始寫入數據...
WriteTask-2 開始寫入數據...
WriteTask-1 開始寫入數據...
WriteTask-0 開始寫入數據...
WriteTask-2 寫入數據完畢,等待其餘線程寫入完畢 1512048648904
WriteTask-1 寫入數據完畢,等待其餘線程寫入完畢 1512048650042
WriteTask-0 寫入數據完畢,等待其餘線程寫入完畢 1512048650209
WriteTask-3 寫入數據完畢,等待其餘線程寫入完畢 1512048652606
全部任務寫入完畢,繼續處理其餘任務... 1512048652607
全部任務寫入完畢,繼續處理其餘任務... 1512048652607
全部任務寫入完畢,繼續處理其餘任務... 1512048652607
全部任務寫入完畢,繼續處理其餘任務... 1512048652607

**java

  • 若是說想在全部線程寫入操做完以後,進行額外的其餘操做能夠爲CyclicBarrier提供Runnable參數:

**dom

package sychronized;

import java.util.Random;
import java.util.concurrent.*;
import static net.mindview.util.Print.*;

class WriteTask implements Runnable{
    private static int count = 0;
    private final int id = count++;
    private CyclicBarrier barrier ;
    private static Random random = new Random(47);
    public WriteTask(CyclicBarrier cyclicBarrier) {
        this.barrier = cyclicBarrier;
    }
    
    @Override
    public void run() {
        print(this+" 開始寫入數據...");
        try {
            TimeUnit.MILLISECONDS.sleep(random.nextInt(5000));      //以睡眠來模擬寫入數據操做
            print(this+" 寫入數據完畢,等待其餘線程寫入完畢"+" "+System.currentTimeMillis());
            barrier.await();
            TimeUnit.MILLISECONDS.sleep(10);
        } catch (InterruptedException e) {
            print(this + "is interrupted!");
        }catch(BrokenBarrierException e){
           throw new RuntimeException(e);
        }
        print("全部任務寫入完畢,繼續處理其餘任務... "+System.currentTimeMillis()+Thread.currentThread());
    }

    @Override
    public String toString() {
        return getClass().getSimpleName()+"-"+id;
    }
}

public class CyclicBarrierTest {
    public static void main(String[] args) {
        int N = 4;
        CyclicBarrier barrier  = new CyclicBarrier(N, new Runnable() {
            @Override
            public void run() {
                print(Thread.currentThread());
            }
        });
     
        ExecutorService exec = Executors.newCachedThreadPool();
        for(int i = 0; i < N; ++i){
            exec.execute(new WriteTask(barrier));
        }
        exec.shutdown();
    }
}
#輸出結果爲:
WriteTask-3 開始寫入數據...
WriteTask-1 開始寫入數據...
WriteTask-2 開始寫入數據...
WriteTask-0 開始寫入數據...
WriteTask-1 寫入數據完畢,等待其餘線程寫入完畢 1512049061954
WriteTask-2 寫入數據完畢,等待其餘線程寫入完畢 1512049063092
WriteTask-0 寫入數據完畢,等待其餘線程寫入完畢 1512049063261
WriteTask-3 寫入數據完畢,等待其餘線程寫入完畢 1512049065657
Thread[pool-1-thread-4,5,main]
全部任務寫入完畢,繼續處理其餘任務... 1512049065668Thread[pool-1-thread-2,5,main]
全部任務寫入完畢,繼續處理其餘任務... 1512049065668Thread[pool-1-thread-1,5,main]
全部任務寫入完畢,繼續處理其餘任務... 1512049065668Thread[pool-1-thread-4,5,main]
全部任務寫入完畢,繼續處理其餘任務... 1512049065668Thread[pool-1-thread-3,5,main]

從結果能夠看出,當四個線程都到達barrier狀態後,會從四個線程中選擇一個線程去執行Runnable。ide

  • 另外CyclicBarrier是能夠重用的,看下面這個例子:
package sychronized;

import java.util.Random;
import java.util.concurrent.*;
import static net.mindview.util.Print.*;

class WriteTask implements Runnable{
    private static int count = 0;
    private final int id = count++;
    private CyclicBarrier barrier ;
    private static Random random = new Random(47);
    public WriteTask(CyclicBarrier cyclicBarrier) {
        this.barrier = cyclicBarrier;
    }
    
    @Override
    public void run() {
        
        while (!Thread.interrupted()){
            print(this+" 開始寫入數據...");
            try {
                TimeUnit.MILLISECONDS.sleep(random.nextInt(5000));      //以睡眠來模擬寫入數據操做
                print(this+" 寫入數據完畢,等待其餘線程寫入完畢"+" "+System.currentTimeMillis());
                barrier.await();
                TimeUnit.MILLISECONDS.sleep(10);
            } catch (InterruptedException e) {
                print(this + "is interrupted!");
            }catch(BrokenBarrierException e){
                throw new RuntimeException(e);
            }
            print("全部任務寫入完畢,繼續處理其餘任務... "+System.currentTimeMillis());
        }
       
    }

    @Override
    public String toString() {
        return getClass().getSimpleName()+"-"+id;
    }
}

class CyclicBarrierManager implements Runnable{
    private CyclicBarrier barrier ;
    private ExecutorService exec;
    public CyclicBarrierManager(CyclicBarrier barrier, ExecutorService exec,int N){
        this.barrier = barrier ;
        this.exec = exec;
        for (int i = 0; i < N-1; ++i){
            exec.execute(new WriteTask(barrier));
        }
    }
    
    @Override
    public void run(){
        while (!Thread.interrupted()){
            try {
                barrier.await();
            }catch (InterruptedException e){
                print(getClass().getSimpleName()+" 被中斷了!");
            }catch (BrokenBarrierException e){
                throw new RuntimeException(e);
            }
        }
    }
}

public class CyclicBarrierTest {
    public static void main(String[] args) throws Exception{
        int N = 4;
        CyclicBarrier barrier  = new CyclicBarrier(N);
        ExecutorService exec = Executors.newCachedThreadPool();
        exec.execute(new CyclicBarrierManager(barrier,exec,N));
        exec.shutdown();
    }
}

#輸出結果:
WriteTask-1 開始寫入數據...
WriteTask-2 開始寫入數據...
WriteTask-0 開始寫入數據...
WriteTask-2 寫入數據完畢,等待其餘線程寫入完畢 1512051484365
WriteTask-0 寫入數據完畢,等待其餘線程寫入完畢 1512051485503
WriteTask-1 寫入數據完畢,等待其餘線程寫入完畢 1512051488068
全部任務寫入完畢,繼續處理其餘任務... 1512051488078
全部任務寫入完畢,繼續處理其餘任務... 1512051488078
WriteTask-2 開始寫入數據...
全部任務寫入完畢,繼續處理其餘任務... 1512051488078
WriteTask-1 開始寫入數據...
WriteTask-0 開始寫入數據...
WriteTask-0 寫入數據完畢,等待其餘線程寫入完畢 1512051488513
WriteTask-1 寫入數據完畢,等待其餘線程寫入完畢 1512051489045
WriteTask-2 寫入數據完畢,等待其餘線程寫入完畢 1512051489945
全部任務寫入完畢,繼續處理其餘任務... 1512051489955
WriteTask-0 開始寫入數據...
全部任務寫入完畢,繼續處理其餘任務... 1512051489955
全部任務寫入完畢,繼續處理其餘任務... 1512051489955
WriteTask-2 開始寫入數據...
WriteTask-1 開始寫入數據...
WriteTask-2 寫入數據完畢,等待其餘線程寫入完畢 1512051490155
WriteTask-1 寫入數據完畢,等待其餘線程寫入完畢 1512051494477
WriteTask-0 寫入數據完畢,等待其餘線程寫入完畢 1512051494823
全部任務寫入完畢,繼續處理其餘任務... 1512051494833
全部任務寫入完畢,繼續處理其餘任務... 1512051494833
WriteTask-0 開始寫入數據...
全部任務寫入完畢,繼續處理其餘任務... 1512051494833
WriteTask-1 開始寫入數據...
WriteTask-2 開始寫入數據...
WriteTask-2 寫入數據完畢,等待其餘線程寫入完畢 1512051494961
WriteTask-0 寫入數據完畢,等待其餘線程寫入完畢 1512051496040
WriteTask-1 寫入數據完畢,等待其餘線程寫入完畢 1512051498121
全部任務寫入完畢,繼續處理其餘任務... 1512051498132
全部任務寫入完畢,繼續處理其餘任務... 1512051498132
WriteTask-1 開始寫入數據...
全部任務寫入完畢,繼續處理其餘任務... 1512051498132

Semaphore

  • Semaphore翻譯成字面意思爲 信號量,Semaphore 能夠同時讓多個線程同時訪問共享資源,經過 acquire() 獲取一個許可,若是沒有就等待,而 release() 釋放一個許可。
  • Semaphore類位於java.util.concurrent包下,它提供了2個構造器:
public Semaphore(int permits) {          //參數permits表示許可數目,即同時能夠容許多少線程進行訪問
    sync = new NonfairSync(permits);
}
public Semaphore(int permits, boolean fair) {    //這個多了一個參數fair表示是不是公平的,即等待時間越久的越先獲取許可
    sync = (fair)? new FairSync(permits) : new NonfairSync(permits);
}
  • 下面說一下Semaphore類中比較重要的幾個方法,首先是acquire()、release()方法:
public void acquire() throws InterruptedException {  }     //獲取一個許可
public void acquire(int permits) throws InterruptedException { }    //獲取permits個許可
public void release() {}          //釋放一個許可
public void release(int permits) {}    //釋放permits個許可
  • acquire()用來獲取一個許可,若無許可可以得到,則會一直等待,直到得到許可。
  • release()用來釋放許可
  • 注意,在釋放許可以前,必須先獲得到許可。

 這4個方法都會被阻塞,若是想當即獲得執行結果,可使用下面幾個方法:ui

public boolean tryAcquire() { };    //嘗試獲取一個許可,若獲取成功,則當即返回true,若獲取失敗,則當即返回false
public boolean tryAcquire(long timeout, TimeUnit unit) throws InterruptedException { };  //嘗試獲取一個許可,若在指定的時間內獲取成功,則當即返回true,不然則當即返回false
public boolean tryAcquire(int permits) { }; //嘗試獲取permits個許可,若獲取成功,則當即返回true,若獲取失敗,則當即返回false
public boolean tryAcquire(int permits, long timeout, TimeUnit unit) throws InterruptedException { }; //嘗試獲取permits個許可,若在指定的時間內獲取成功,則當即返回true,不然則當即返回false
package sychronized;
import java.util.Random;
import java.util.concurrent.*;
import static net.mindview.util.Print.*;

class Worker implements Runnable{
    private static int count = 0;
    private final int id = count++;
    private int finished = 0;
    private Random random = new Random(47);
    private Semaphore semaphore;
    public Worker(Semaphore semaphore){
        this.semaphore = semaphore;
    }
    
    @Override 
    public void run(){
        try {
            while (!Thread.interrupted()){
                semaphore.acquire();
                print(this+" 佔用一個機器在生產...   ");
                TimeUnit.MILLISECONDS.sleep(random.nextInt(2000));
                synchronized (this){
                    print(" 已經生產了"+(++finished)+"個產品,"+"釋放出機器");
                }
                semaphore.release();
            }
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    @Override
    public String toString() {
        return getClass().getSimpleName()+"-"+id;
    }
}

public class SemaphoreTest {
    public static void main(String[] args) {
        int N = 8;            //工人數
        Semaphore semaphore = new Semaphore(5); //機器數目
        ExecutorService exec = Executors.newCachedThreadPool();
        for (int i = 0; i < N; ++i){
            exec.execute(new Worker(semaphore));
        }
        exec.shutdown();
    }
}

總結

CountDownLatch和CyclicBarrier都可以實現線程之間的等待,只不過它們側重點不一樣:this

  • CountDownLatch 通常用於某個線程A等待若干個其餘線程執行完任務以後,它才執行;
  • CyclicBarrier 通常用於一組線程互相等待至某個狀態,而後這一組線程再同時執行;
  • CountDownLatch 是不可以重用的,而 CyclicBarrier 是能夠重用的。

Semaphore 其實和鎖有點相似,它通常用於控制對 某組 資源的訪問權限,而鎖是控制對 某個 資源的訪問權限。線程

相關文章
相關標籤/搜索