多线程-线程同步工具CountDownLatch

背景:最近做测试需要准备大量的数据,用单线程去连续调几个接口耗时N秒,而数据量大这个时间消耗就成倍数增长,单线程效率太低,决定用多线程跑数据。

使用多线程,首先实现一个线程,再开辟一个线程池,循环调用。但是线程什么时候结束是个问题,这里使用到了CountDownLatch。

Latch:英[lætʃ], 门闩。

CountDownLatch 是一个同步工具类,它允许一个或多个线程一直等待,直到其他线程的操作执行完成后再执行。

CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值达到0时,表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务了。

CountDownLatch的伪代码如下所示:

1. //Main thread start

2. //Create CountDownLatch for N threads

3. //Create and start N threads

4. //Main thread wait on latch

5. //N threads completes there tasks are returns

6. //Main thread resume execution

CountDownLatch如何使用?

CountDownLatch.java类中定义的构造函数:

1. //Constructs a CountDownLatch initialized with the given count.

2. public void CountDownLatch(int count) {...}

构造器中的计数值实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,且不能被修改。

主线程必须在启动其他线程后立即调用CountDownLatch.await()方法。这样主线程的操作就会在这个方法上阻塞,直到其他线程完成各自的任务。

其他N个线程必须引用闭锁对象,因为他们需要通知CountDownLatch对象,他们已经完成了各自的任务。这种通知通过CountDownLatch.countDown()方法来完成;每一次调用这个方法,在构造函数中初始化的count值就减1.所以当所有N个线程都调用了这个方法,count=0,主线程就能通过await()方法,恢复执行自己的任务。

代码Demo:

public void bizDealCode(){

Long startTime = System.currentTimeMillis();

intsize =10000;

ExecutorService threadPool = Executors.newFixedThreadPool(15);

CountDownLatch countDownLatch =newCountDownLatch(size);

for(intj =0; j < size; ++j) {

threadPool.submit(newMakeOrderThread(countDownLatch));

}

System.out.println("finished submit");

try{

countDownLatch.await();

}catch(Exception e) {

}

System.out.println("finished process");

threadPool.shutdown();

System.out.println("costTime is "+ (System.currentTimeMillis() - startTime) +"ms");

}

static classMakeOrderThreadextendsThread {

privateCountDownLatchcountDownLatch;

publicMakeOrderThread(CountDownLatch countDownLatch) {

this.countDownLatch= countDownLatch;

}

@Override

public voidrun() {

try{

bizCode();

}catch(Exception e) {

}finally{

countDownLatch.countDown();

}

}

}

}

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

推荐阅读更多精彩内容

  • 正如每个Java文档所描述的那样,CountDownLatch是一个同步工具类,它允许一个或多个线程一直等待,直到...
    霁暖阁阅读 1,099评论 0 7
  • 此篇博客所有源码均来自JDK 1.8 在上篇博客中介绍了Java四大并发工具之一的CyclicBarrier,今天...
    chenssy阅读 2,935评论 2 27
  • CountDownLatch是一个同步工具类,它允许一个或多个线程一直等待,直到其他线程的操作执行完后再执行。在J...
    时待吾阅读 791评论 0 2
  • 1. Java基础部分 基础部分的顺序:基本语法,类相关的语法,内部类的语法,继承相关的语法,异常的语法,线程的语...
    子非鱼_t_阅读 31,764评论 18 399
  • 今日移栽芦荟,分出6小盆,加上母盆共7。另有芦荟苗若干。在群中发消息后,有5位邻居分成3波来至家中,聊天,参观,顺...
    鹰王守仁阅读 149评论 0 0