对本地程序进行多线程并发压测的单元测试

在分布式系统开发的过程中， 我们写的方法要充分的考虑并发条件下会不会有问题，所以会有对方法进行多线程并发测试的要求，这里分享一个简单可行的多线程并发单元测试的模板。

假设我们有一个方法generateOrderId()要进行单元测试， 方法本身就是生成订单号的方法。

public static String generateOrderId(String prefix){
   return "";
}

单元测试：
定义内部类，执行generateOrderId（） 方法:

class MyCallable implements Callable<Object> {
    private Integer taskNum;
    MyCallable(Integer taskNum) {
        this.taskNum = taskNum;
    }
    public Object call(){
        String orderId = OrderIdGenerateUtil.generateOrderId("532", new HashMap<>());
        System.out.println("线程" + taskNum + "获取订单号任务启动: 获取订单号:" + orderId);
        return orderId;
    }
}

单元测试方法：

 @Test
    public void multithreadingTest() throws Exception{
        int taskSize = 100;
        ExecutorService pool = Executors.newFixedThreadPool(taskSize);
        List<Future> list = new ArrayList<Future>();
        for (int i = 0; i < taskSize; i++) {
            Callable c = new MyCallable(i);
            // 执行任务并获取Future对象
            Future f = pool.submit(c);
            list.add(f);
        }
        // 关闭线程池
        pool.shutdown();

        HashSet<String> hashSet = new HashSet<>();
        // 获取运行结果
        for (Future f : list) {
            // 从Future对象上获取任务的返回值，并输出到控制台
            hashSet.add(f.get().toString());
            Assert.assertEquals(f.get().toString().length(), 18);
        }
        Assert.assertEquals(hashSet.size(), taskSize);
    }

使用Callable定义线程的原因就是我们需要用方法返回的String，做断言使用。