多线程如何等待所有子线程一起完成任务后再执行主线程
1、使用线程类自带的join方法,将子线程加入到主线程,在子线程执行完之后,在执行主线程逻辑。
public static void joinDemo()
throws InterruptedException
{
System.out.println("=========Test with join=====");
JoinWorker worker1 = new JoinWorker("worker1");
JoinWorker worker2 = new JoinWorker("worker2");
worker1.start();
worker2.start();
worker1.join();
worker2.join();
doSuperWork();
}
2、使用JDK的并发包中的CountDownLatch类, 使用CountDownLatch,每个线程调用其countDown方法使计数器-1,主线程调用await方法阻塞等待,直到CountDownLatch计数器为0时继续执行,例如
首先,定义子线程
static class CountDownLatchWorker extends Thread
{
String workerName;
CountDownLatch latch;
public CountDownLatchWorker(String workerName, CountDownLatch latch)
{
this.workerName = workerName;
this.latch = latch;
}
public void run()
{
System.out.println("Sub Worker " + workerName + " do work begin at "
+ sdf.format(new Date()));
new ThreadWaitDemo().doSomeWork();// 做实际工作
System.out.println("Sub Worker " + workerName + " do work complete at "
+ sdf.format(new Date()));
latch.countDown();// 完成之后,计数器减一
}
}
主线程中调研await方法阻塞等待,直到所有线程完成
public static void countDownLatchDemo()
throws InterruptedException
{
System.out.println("=========Test with CountDownLatch=====");
CountDownLatch latch = new CountDownLatch(2);
CountDownLatchWorker worker1 = new CountDownLatchWorker("worker1", latch);
CountDownLatchWorker worker2 = new CountDownLatchWorker("worker2", latch);
worker1.start();
worker2.start();
//主线程阻塞等待
latch.await();
doSuperWork();
}
3、使用JDK并发包CyclicBarrier,CyclicBarrier类似于CountDownLatch也是个计数器, 不同的是CyclicBarrier的await()方法没被调用一次,计数便会减少1,并阻塞住当前线程。当计数减至0时,阻塞解除,所有在此 CyclicBarrier 上面阻塞的线程开始运行。 在这之后,如果再次调用 await()方法,计数就又会变成 N-1,新一轮重新开始CyclicBarrier初始时还可带一个Runnable的参数,此Runnable任务在CyclicBarrier的数目达到后,所有其它线程被唤醒前被执行。
在这里插入图片描述
总结线程同步计数器俩者区别:
CountDownLatch:一个或者多个线程,等待其他多个线程完成各自的任务之后才能接下来往下执行主线程;
CyclicBarrier:多个线程互相等待,直到到达同一个同步点,再继续一起执行各自的任务,完成最后一个任务之后再执行构造方法第二个参数定义的主线程任务。
对于CountDownLatch来说,重点是“一个线程(多个线程)等待”,而其他的N个线程在完成“某件事情”之后,可以终止,也可以等待。而对于CyclicBarrier,重点是多个线程,在任意一个线程没有完成,所有的线程都必须互相等待,然后继续一起执行。
CountDownLatch是计数器,线程完成一个记录一个,只不过计数不是递增而是递减,而CyclicBarrier更像是一个阀门,需要所有线程都到达,阀门才能打开,然后继续执行。