最近写了一个并发幂等测试,用线程池加入多个线程,同时启动,领导觉得这样有一定的风险,要求更严格一点,把所有的线程加入池中,然后同时启动。
本来有多种方法,因为我们需要从多个线程中获取返回值,所以我们用CountDownLatch来同步多线程。CyclicBarrier也是可以同步多线程的,但因为其无法获取返回值,最后只能选择CountDownLatch.
因公司的代码不便共享,这里只提供一小部分代码。
CountDownLatch latch = new CountDownLatch(1);
java" id="highlighter_119456">
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