CyclicBarrier与CountDownLatch功能类似,也可以实现线程间的计数等待,但它的功能比CountDownLatch更加强大。
CyclicBarrier之所以翻译为循环栅栏,栅栏的意思就是,阻止线程继续执行,要求线程在栅栏出等待。循环的意思是指,这个栅栏可以反复使用。
比如,我们设置计数器为10,那么凑齐第一批10个线程之后,计数器就会归零,然后凑齐下一批的10个线程。
比如,司令下达命令,要求10个士兵一起去执行任务。这时,就会要求10个士兵先集合报道,接着,一起去执行任务。当任务执行完毕时,司令就会宣布任务执行完成。
CyclicBarrier接收一个Runnable参数作为,当计数器一次计数完成后,系统会执行的动作。接收的另外一个int型的参数,表示计数总数,也就是参与的线程总数。
CyclicBarrier.await()方法涉及到两种异常:InterrupttedException和BrokenBarrierException。一旦遇到BrokenBarrierException异常,这说明CyclicBarrier已经破损,无法等待所有的线程都到齐了。
//int parties:计数总数,也就是参与的线程总数
public CyclicBarrier(int parties,Runnable barrierAction)
public class CyclicBarrierDemo {
//士兵
public static class Soldier implements Runnable {
private String soldier; //士兵名字,即线程的标识
private final CyclicBarrier cyclic; //循环栅栏实例
public Soldier(CyclicBarrier cyclic,String soldierName) {
this.cyclic = cyclic;
this.soldier = soldierName;
}
@Override
public void run() {
try {
cyclic.await(); //等待所有的N个线程到齐,才会继续执行。CyclicBarrier的一次计数完成
doWork();
cyclic.await(); //再一次调用await方法,就会进行下一次计数。等待所有的士兵完成工作
} catch (Exception e) {
e.printStackTrace();
}
}
void doWork() {
try {
Thread.sleep(Math.abs(new Random().nextInt()%10000)); //模拟耗时任务
} catch (Exception e) {
e.printStackTrace();
}
System.out.println(soldier+": 任务完成");
}
}
public static class BarrierRun implements Runnable {
boolean flag;
int N;
public BarrierRun(boolean flag,int N) {
this.flag = flag;
this.N = N;
}
@Override
public void run() {
if (flag) {
System.out.println("司令:[士兵"+N+"个,任务完成!");
}else {
System.out.println("司令:[士兵"+N+"个,集合完毕!");
flag = true;
}
}
}
public static void main(String[] args) {
final int N = 10;
Thread[] allSoldier = new Thread[N];
boolean flag = false;
//创建CyclicBarrier实例
//N:参与的线程总数
//new BarrierRun(flag, N):当计数器一次计数完成后,系统会执行的动作
//当参与的线程总数达到N时,会执行new BarrierRun(flag, N)的方法
CyclicBarrier cyclic = new CyclicBarrier(N,new BarrierRun(flag, N));
//设置屏障点,主要是为了执行这个方法
System.out.println("集合队伍");
for (int i = 0; i < N; i++) {
System.out.println("士兵 "+i+"报道!");
allSoldier[i] = new Thread(new Soldier(cyclic, "士兵"+i));
allSoldier[i].start();
}
}
}
上述代码输出:
集合队伍
士兵 0报道!
士兵 1报道!
士兵 2报道!
士兵 3报道!
士兵 4报道!
士兵 5报道!
士兵 6报道!
士兵 7报道!
士兵 8报道!
士兵 9报道!
司令:[士兵10个,集合完毕!
士兵2: 任务完成
士兵9: 任务完成
士兵0: 任务完成
士兵8: 任务完成
士兵6: 任务完成
士兵4: 任务完成
士兵1: 任务完成
士兵7: 任务完成
士兵3: 任务完成
士兵5: 任务完成
司令:[士兵10个,任务完成!