最近写了一个并发幂等测试,用线程池加入多个线程,同时启动,领导觉得这样有一定的风险,要求更严格一点,把所有的线程加入池中,然后同时启动。
本来有多种方法,因为我们需要从多个线程中获取返回值,所以我们用CountDownLatch来同步多线程。CyclicBarrier也是可以同步多线程的,但因为其无法获取返回值,最后只能选择CountDownLatch.
因公司的代码不便共享,这里只提供一小部分代码。
CountDownLatch latch = new CountDownLatch(1);
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
<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时那些线程才能同时释放出来,所以能同步运行多线程。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://www.cnblogs.com/linwenbin/p/12700983.html