CountDownLatch应用场景实践

一.CountDownLatch介绍

        CountDownLatch是一个并发工具类， 作用是允许一个或多个线程等待其他线程完成操作。我们有时会称之为发令枪。有一个个形象的例子能说明其功能：公司召开一个全体股东会，需要所有股东到场了才能正式开始。会议室准备了一个花名册，记录了需要到会的总人数，每一个股东来到会议室就要签名确认，同时总人数减一，当需要到会的总人数为0时，会议就正式开始。所以，CountDownLatch就有充当计数器和发令枪的功能。

二.结合ExecutorService实现拆分任务并发执行

       一般我们业务场景中经常会有执行一个大任务，如果我们串行执行，则效率较慢。我们应用分而治之的思想，将其拆分，启动多个线程来并行。但是如果目前不阻塞主线程，可能在子线程进行中的过程中，主线程已结束。这样也无法最终汇总结果了。此时引入CountDownLatch便能解决这个痛点。

//利用countDownLatch结合ExecutorService在并发执行任务时，能够保证子线程任务结束后，主线程才继续往下走。
		CountDownLatch countDownLatch = new CountDownLatch(3);
		ExecutorService executorService = Executors.newFixedThreadPool(3);
		Runnable runnable1 = () -> {
			System.out.println(111111);
			countDownLatch.countDown();
		};
		Runnable runnable2 = () -> {
			System.out.println(222222);
			countDownLatch.countDown();
		};
		Runnable runnable3 = () -> {
			System.out.println(333333);
			countDownLatch.countDown();
		};
		executorService.execute(runnable1);
		executorService.execute(runnable2);
		executorService.execute(runnable3);
		//阻塞当前主线程
		countDownLatch.await();
		System.out.println("主线程结束返回结果");

任务拆分为几个线程来执行，则CountDownLatch就是多少。

当子线程都执行完毕了，此时CountDownLatch为0，主线程不再阻塞，继续往下走。

三.模拟高并发

有时我们写测试用例时，想要模拟一些并发场景，这时可以引入CountDownLatch。

private static final int MAX_THREAD = 10000;
private CountDownLatch cdl = new CountDownLatch(MAX_THREAD);

@Test
public void countDownLatchTest() throws InterruptedException {
		//模拟并发
		for (int i =0;i<MAX_THREAD;i++){
			Thread thread = new Thread(() -> {
				cdl.countDown();
				try {
					cdl.await();
				} catch (InterruptedException e) {
					e.printStackTrace();
				}
				//测试调用业务接口等
				System.out.println(new Date());
			});
			thread.start();
		}
		Thread.sleep(2000);
	}

如上便是模拟10000的并发，主线程肯定要sleep下，不然直接就结束了，子线程的结果也会来不及打印出来。