写一个并发请求的Case

有些场景下,你可能要检查代码有没有并发问题,验证幂等或者做一下简单的压力测试, 这时候需要写一个并发请求的程序,那么怎么做呢?
先上代码

        final long time_start = System.currentTimeMillis();
        ExecutorService service = Executors.newFixedThreadPool(10);

        int client_num = 10;

        final CountDownLatch countDownLatch = new CountDownLatch(1);

        for (int i = 0; i < client_num; i++) {
            Runnable job = new Runnable() {
                @Override
                public void run()  {
                    Worker worker = new Worker();
                    try {
                        countDownLatch.await();
                        worker.doWork();
                    }catch(Exception e){
                        e.printStackTrace();
                    }

                }
            };

            service.submit(job);
        }

        countDownLatch.countDown();

        service.shutdown();

        while (!service.isTerminated()) {
        }
        System.out.println("time consuming is " + (System.currentTimeMillis() - time_start));

我们来拆解下这个Case,看看都做了些什么:

1. 容器准备

首先,你需要一个线程池来调用任务处理,虽然你可以每次new一个线程来做,但是无论是在生产环境还是测试,都不太建议这样做。本示例用的是
ExecutorService service = Executors.newFixedThreadPool(10) 其效果是构造一个固定线程数的线程池,这个线程池的coreSize和maxSize都是10, 也就是说,最多可以同时处理10个任务,多了的就要排队。
线程池装载完任务后,执行submit , 但是此时任务并未执行,具体原因看下面。

2. 并发执行

这里是比较tricky的一个点,因为要模拟完全并发的情况,就要保证所有任务同一时间触发,这里我们用了一个java.util.concurrent的工具类 CountDownLatch, 从名字大概可以猜测到,这个是一个像栅栏一样的工具,先把羊圈在里面,等count数到0的时候,打开栅栏,放出群羊。。
使用步骤: 首先,初始化countDownLatch计数为1。 然后, 在业务代码里在执行业务逻辑之前我们嵌入了一行 countDownLatch.await() 。最后, 当所有任务都装载完毕提交后,执行countDownLatch.countDown(),如同发令枪一般,所有的任务同时触发,并发量达到client_num的数量10。

3. 任务结束

任务结束的时候调用的是线程池的shutdown方法, 看一下这个方法的注释

Attempts to stop all actively executing tasks, halts the processing of waiting tasks, and returns a list of the tasks that were awaiting execution.

意思是这个方法只是尝试去停止所有任务,但是并不保证都能结束,并且没有结束的任务是以list返回的。
所以后面又跟了一个

 while (!service.isTerminated()) {
    }

循环一直等待所有的任务结束完毕,最后再统计所有任务的执行时间。

结语

这是一段比较简单的并发测试代码,单元测试足够用,但是如果用做规模化的压力测试,还需要考虑压力机与测试机的连接以及数据热点等一系列问题,这又是个很大的话题了。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 第1章 并发编程的挑战 1.1 上下文切换 即便是单核CPU也支持多线程并发,CPU通过给每个线程分配时间片(几十...
    卑鄙的鹿尤菌阅读 4,822评论 1 22
  • 一.线程安全性 线程安全是建立在对于对象状态访问操作进行管理,特别是对共享的与可变的状态的访问 解释下上面的话: ...
    黄大大吃不胖阅读 865评论 0 3
  • 《逻辑思维》2015跨年演讲里,罗胖讲了一个他小时候的故事,他的一个同学常常欺负他。罗胖说并不害怕他上课的时候对他...
    Lynette_C阅读 636评论 0 8
  • 我迷失了一段记忆, 记忆里是我不懂的虚无。 我把所有的猜想加铸, 仍是沧海桑田的茫茫。 若有若无的片段, 纠葛不清...
    糜情阅读 210评论 0 1
  • 最近看了《太太万岁》,里面的叶舒心接到任务要去刚离婚的前夫公司开除一位高管。上司兼好友担心她会情绪失控难以胜任,叶...
    家有果宝阅读 622评论 0 0