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

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

很多时候我们都想做一些并发测试，又不想用 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());
    }

image.png

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