多线程

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方法。

本文由 创作，采用 知识共享署名4.0 国际许可协议进行许可。本站文章除注明转载/出处外，均为本站原创或翻译，转载前请务必署名。最后编辑时间为: 2021/05/17 02:53