首页 > 编程语言 > 如何使用CountDownLatch同步java多线程
2020
10-09

如何使用CountDownLatch同步java多线程

最近写了一个并发幂等测试,用线程池加入多个线程,同时启动,领导觉得这样有一定的风险,要求更严格一点,把所有的线程加入池中,然后同时启动。

本来有多种方法,因为我们需要从多个线程中获取返回值,所以我们用CountDownLatch来同步多线程。CyclicBarrier也是可以同步多线程的,但因为其无法获取返回值,最后只能选择CountDownLatch.

因公司的代码不便共享,这里只提供一小部分代码。

CountDownLatch latch = new CountDownLatch(1);

<br data-filtered="filtered">ExecutorService pool;
    if (concurrentNum != null&&concurrentNum <= maxConcurrentNum && concurrentNum > 0) {
        pool = Executors.newFixedThreadPool(concurrentNum);
    } else {
      concurrentNum = defaultConcurrentNum;
      pool = Executors.newFixedThreadPool(defaultConcurrentNum);
    }
    for (int i = 0; i < concurrentNum; i++) {
      Future res = pool.submit(new Callable<Object>() {
        @Override
        public Object call() throws Exception {
          latch.await();
          Object retObj = executeApi();
          return retObj;
        }
      });
      resultList.add(res);
    }
    latch.countDown();
    for (int i = 0; i < concurrentNum; i++) {
      retList.add(resultList.get(i).get());
    }
    pool.shutdown();

这里顺便提一下,latch在中文中就是门栓的意思,这样就很好理解了,当有门栓时,latch.await()的线程都在等待,只有当门栓的个数为0时那些线程才能同时释放出来,所以能同步运行多线程。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持自学编程网。

编程技巧