写一个并发请求的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()) {
    }

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

结语

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,294评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,780评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,001评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,593评论 1 289
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,687评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,679评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,667评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,426评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,872评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,180评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,346评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,019评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,658评论 3 323
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,268评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,495评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,275评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,207评论 2 352

推荐阅读更多精彩内容

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