package com.yao;

import java.util.Random;
import java.util.concurrent.CyclicBarrier;

/**
 * CyclicBarrier類似于CountDownLatch也是個計數器，
 * 不同的是CyclicBarrier數的是調用了CyclicBarrier.await()進入等待的線程數，
 * 當線程數達到了CyclicBarrier初始時規定的數目時，所有進入等待狀態的線程被喚醒并繼續。
 * CyclicBarrier就象它名字的意思一樣，可看成是個障礙，
 * 所有的線程必須到齊后才能一起通過這個障礙。
 * CyclicBarrier初始時還可帶一個Runnable的參數，
 * 此Runnable任務在CyclicBarrier的數目達到后，所有其它線程被喚醒前被執行。
 */
public class CyclicBarrierTest {

 public static class ComponentThread implements Runnable {
  CyclicBarrier barrier;// 計數器
  int ID; // 組件標識
  int[] array; // 數據數組

  // 構造方法
  public ComponentThread(CyclicBarrier barrier, int[] array, int ID) {
   this.barrier = barrier;
   this.ID = ID;
   this.array = array;
  }

  public void run() {
   try {
    array[ID] = new Random().nextInt(100);
    System.out.println("Component " + ID + " generates: " + array[ID]);
    // 在這里等待Barrier處
    System.out.println("Component " + ID + " sleep...");
    barrier.await();
    System.out.println("Component " + ID + " awaked...");
    // 計算數據數組中的當前值和后續值
    int result = array[ID] + array[ID + 1];
    System.out.println("Component " + ID + " result: " + result);
   } catch (Exception ex) {
   }
  }
 }
 /**
  * 測試CyclicBarrier的用法
  */
 public static void testCyclicBarrier() {
  final int[] array = new int[3];
  CyclicBarrier barrier = new CyclicBarrier(2, new Runnable() {
   // 在所有線程都到達Barrier時執行
   public void run() {
    System.out.println("testCyclicBarrier run...");
    array[2] = array[0] + array[1];
   }
  });

  // 啟動線程
  new Thread(new ComponentThread(barrier, array, 0)).start();
  new Thread(new ComponentThread(barrier, array, 1)).start();
 }

 public static void main(String[] args) {
  CyclicBarrierTest.testCyclicBarrier();
 }
}