线程同步之CyclicBarrier使用示例

前几天写了一个CountDownLatch使用示例，今天把CyclicBarrier的示例补上。

话说这日部门组织大家去吃大排档，有18个人。大排档的桌小，一桌只能4个人。大家工作完成时间不定，有人来的早有人来的晚，但约好所有人到齐才开吃。呃，这里假设所有人都一定会来（现实中可能有些人因为这样那样的原因去不了）。

因为18个人来齐才能开饭，我这里用CountDownLatch来对18人到的事件计数。

final CountDownLatch countDown = new CountDownLatch(18); // 18个人

并在主线程里

		try {
			countDown.await(); <span style="font-family: Arial, Helvetica, sans-serif;">// 等所有人到齐</span>
		} catch (InterruptedException e) {
			e.printStackTrace();
		}

我们用一个CyclicBarrier对象表示能容下4个人的小桌人满、通知大排档老板

		final CountDownLatch firstSignal = new CountDownLatch(1); // 第一桌
		// 每4人一桌
		final CyclicBarrier barrier = new CyclicBarrier(4, new Runnable() {
			@Override
			public void run() {
				if (firstSignal.getCount() > 0) {
					System.out.println("老板，先开一桌！");
				} else {
					System.out.println("老板，再开一桌！");
				}
				firstSignal.countDown();
			}
		});

每个人来的时候报个到，18个人最后2人只能他们两人一桌，barrier对象指定的规则不适用，需要reset()，这将产生一个BrokenBarrierException。

		ExecutorService pool = Executors.newCachedThreadPool();
		for (int i = 0; i < 18; i++) {
			final int idx = i + 1;
			pool.execute(new Runnable() {
				@Override
				public void run() {
					try {
						Thread.sleep(random.nextInt(3000));
					} catch (InterruptedException e) {
						e.printStackTrace();
					}
					System.out.println("第" + idx + "人到了");
					countDown.countDown();
					try {
						barrier.await(); // 等这桌人满或所有人到齐。
					} catch (InterruptedException e) {
						e.printStackTrace();
					} catch (BrokenBarrierException e) { // 最后2人会产生此异常。
						// e.printStackTrace();
						System.out.println("你来的可真晚，" + idx); // 来的晚要被抱怨
					}
				}
			});
		}
		pool.shutdown();

在所有人到齐之后，显然有2人被barrier阻塞了，不能上桌需要如下处理

		try {
			countDown.await(); // 等所有人到齐
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		if (barrier.getNumberWaiting() != 0) {
			System.out.println("老板，给最后" + barrier.getNumberWaiting() + "个人单开一桌！");
			barrier.reset();
		}
		System.out.println("老板，我们人都到齐了，上菜吧");

程序输出如下：

第14人到了
第13人到了
第9人到了
第12人到了
老板，先开一桌！
第1人到了
第2人到了
第5人到了
第18人到了
老板，再开一桌！
第16人到了
第15人到了
第10人到了
第17人到了
老板，再开一桌！
第4人到了
第8人到了
第7人到了
第6人到了
老板，再开一桌！
第11人到了
第3人到了
老板，给最后2个人单开一桌！
老板，我们人都到齐了，上菜吧
你来的可真晚，11
你来的可真晚，3

代码下载请到我的github上：https://github.com/DowenLiu126/syncDemo