CountDownLatch 压测教程

在实战项目中，我们除了使用 jemter 等工具进行压测外，还可以自己动手使用 CountDownLatch 类编写压测代码。可以说 jemter 的并发压测背后也是使用的 CountDownLatch。可见掌握 CountDownLatch 类的使用是有多么的重要。

CountDownLatch是Java多线程同步器的四大金刚之一，CountDownLatch能够使一个线程等待其他线程完成各自的工作后再执行。本文介绍 CountDownLatch 的相关用法。

使用CountDownLatch让线程集中执行，代码如下：

private static void latchTest() throws InterruptedException {
	final CountDownLatch start = new CountDownLatch(1);
	final CountDownLatch end = new CountDownLatch(300);
	ExecutorService exce = Executors.newFixedThreadPool(300);
	for (int i = 0; i < 300; i++) {
		Runnable run = new Runnable() {
			@Override
			public void run() {
				try {
					start.await();
					// 测试执行具体的方法
					testLoad();
				} catch (InterruptedException e) {
					e.printStackTrace();
				} finally {
					end.countDown();
				}
			}
		};
		exce.submit(run);
	}
	start.countDown();
	end.await();
	exce.shutdown();
}

简简单单的几行代码就可以实现300的压测。

除了 CountDownLatch 之外，我们还可以使用 java 线程的 join 方法。

join用于让当前执行线程等待join线程执行结束。其实现原理是不停检查join线程是否存活，如果join线程存活则让当前线程永远wait，代码片段如下，wait(0)表示永远等待下去。

while (isAlive()) {
	wait(0);
}

直到join线程中止后，线程的this.notifyAll会被调用，调用notifyAll是在JVM里实现的，所以JDK里看不到，有兴趣的同学可以看看JVM源码。JDK不推荐在线程实例上使用wait，notify和notifyAll方法。

： » CountDownLatch 压测教程