并发测试工具类 封装，方便使用

并发测试工具类 封装，方便使用

很多时候我们都想做一些并发测试，又不想用 JMeter ,ab等工具所以直接用代码写一个工具类

@Slf4j
public class MultiThreadConcurrentTestUtils {

	/**
	 * 多线程并发测试
	 * @param threadNum  线程数量
	 * @param consumer 函数
	 * @param param 消费参数
	 * @throws InterruptedException
	 */
	@SneakyThrows
	public static void multiThreadTest(int threadNum, Object param,Consumer consumer) {
		CountDownLatch countDownLatch = new CountDownLatch(threadNum);
		ExecutorService executorService = Executors.newFixedThreadPool(threadNum);
		CyclicBarrier cyclicBarrier = new CyclicBarrier(threadNum);
		for (int i = 0; i < threadNum; i++) {
			executorService.execute(() -> {
				try {
					Thread.sleep(new Random().nextInt(60)*1000);
					cyclicBarrier.await();
					// 业务方法
					consumer.accept(param);
					log.info("线程:{} 开始执行",Thread.currentThread().getName());
					countDownLatch.countDown();
				} catch (InterruptedException e) {
					e.printStackTrace();
				} catch (BrokenBarrierException e) {
					e.printStackTrace();
				}
			});
		}
		countDownLatch.await();
		executorService.shutdown();
	}


	/**
	 * 多线程测试
	 * @param threadNum
	 * @param runnable
	 */

	@SneakyThrows
	public static void multiThreadTest(int threadNum, Runnable runnable) {
		CountDownLatch countDownLatch = new CountDownLatch(threadNum);
		ExecutorService executorService = Executors.newFixedThreadPool(threadNum);
		CyclicBarrier cyclicBarrier = new CyclicBarrier(threadNum);
		for (int i = 0; i < threadNum; i++) {
			executorService.execute(() -> {
				try {
					Thread.sleep(new Random().nextInt(60)*1000);
					cyclicBarrier.await();
					// 业务方法
					runnable.run();
					log.info("线程:{} 开始执行",Thread.currentThread().getName());
					countDownLatch.countDown();
				} catch (Exception e) {
					e.printStackTrace();
				}
			});
		}
		countDownLatch.await();
		executorService.shutdown();
	}
}

使用

  List<Integer> list = new ArrayList<>();

    List<Integer> copyList = new CopyOnWriteArrayList<>();
    @Test
    public void testDistributedLock() {
        MultiThreadConcurrentTestUtils.multiThreadTest(10,"test",(e) ->{
            for (int i = 0; i < 10; i++) {
                copyList.add(i);
            }
        });

        System.out.println("copyList "+copyList.size());
        MultiThreadConcurrentTestUtils.multiThreadTest(10,() -> {
            for (int i = 0; i < 10; i++) {
                list.add(i);
            }
        });

        System.out.println("list "+ list.size());
    }

可见 ArrayList 线程不安全，CopyOnWriteArrayList 线程安全