有时候需要对某个业务方法进行多线程并发测试,可以考虑如下代码段:
@Test
public void testParse() {
final int THREAD_COUNT = 10;
ExecutorService executorService = Executors.newFixedThreadPool(THREAD_COUNT);
CountDownLatch latch = new CountDownLatch(THREAD_COUNT);
CyclicBarrier cyclicBarrier = new CyclicBarrier(THREAD_COUNT);
for (int i = 0; i < THREAD_COUNT; i++) {
executorService.execute(() -> {
try {
cyclicBarrier.await();
// 业务代码
} catch (Throwable e) {
e.printStackTrace();
} finally {
latch.countDown();
}
});
}
try {
latch.await();
executorService.shutdownNow();
} catch (InterruptedException e) {
e.printStackTrace();
}
}