Java高并发高性能编程(多线程,协程,Actor,RxJava、Akka、Reactor)

多线程-线程同步工具CountDownLatch

2017-04-19  本文已影响100人  发呆的小耳朵

背景:最近做测试需要准备大量的数据,用单线程去连续调几个接口耗时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();

}

}

}

}

上一篇下一篇

猜你喜欢

热点阅读