多线程执行时，如果一个逻辑需要等若干个线程执行完成后再执行，怎么实现？

实际开发过程中，尤其是大型的业务系统中或批量处理系统中，我们通常会用到多线程来进行任务拆分，来提高系统的处理能力。当对业务不断拆分时，多线程各自任务之间也形成了错综复杂的关系。

我们常常遇到这样的情况，业务模块A 拆分成了 A1 、A2.....An个多线程，来提高处理速度，可是 Ak（1<k<n）这个线程执行时，它有依赖关系，它必须等 Ak-1 和Ak-2这两个进程执行完，才能执行，以免造成业务混乱，那怎么实现？

我们先定义两个实现多线程任务：

任务1：

public class WorkA1 extends Thread {
   @Override
   public void run() {
      System.out.println("WorkA1 开始执行");
      try {
         Thread.sleep(2000);
      } catch (InterruptedException e) {
         e.printStackTrace();
      }
      System.out.println("WorkA1 结束执行");
   }
}

任务2：

public class WorkA2 extends Thread {
   @Override
   public void run() {
      System.out.println("WorkA2 开始执行");
      try {
         Thread.sleep(1800);
      } catch (InterruptedException e) {
         e.printStackTrace();
      }
      System.out.println("WorkA2 结束执行");
   }
}

主线程开始实现是这样的：

public class ThreedClient {
   public static void main(String[] args) {
      System.out.println("准备开始执行任务");
      WorkA1 workA1 = new WorkA1();
      WorkA2 workA2 = new WorkA2();
      workA1.start();
      workA2.start();
      System.out.println("执行完成了。");
   }
}

执行结果是：

准备开始执行任务
执行完成了。
WorkA1 开始执行
WorkA2 开始执行
WorkA2 结束执行
WorkA1 结束执行

如果我们想最后打印 “执行完成了”在所有子线程完成后，怎么办呢？

我们对主线程进行改造：

改造方法1：

public class ThreedClient {
   public static void main(String[] args) {
      System.out.println("准备开始执行任务");
      WorkA1 workA1 = new WorkA1();
      WorkA2 workA2 = new WorkA2();
      workA1.start();
      workA2.start();
      try {
         workA1.join();
         workA2.join();
      }catch (InterruptedException e){
         e.printStackTrace();
      }
      System.out.println("执行完成了。");
   }
}

改造完成后，在看执行结果：

准备开始执行任务
      WorkA2 开始执行
      WorkA1 开始执行
      WorkA2 结束执行
      WorkA1 结束执行
      执行完成了。

OK，达到我们的预期了。

改造方法2：

CountDownLatch允许一个或多个线程等待其他线程完成操作。CountDownLatch的构造函数接收一个int类型的参数作为计数器，如果你想等待N个点完成，这里就传入N。当我们调用countDown方法时，N就会减1，await方法会阻塞当前线程，直到N变成0。

它的原理很简单，就是你需要等待几个任务时，就先定义好，然后任务执行完成一个，计数器减1。看看具体实现：

public class WorkA1 extends Thread {
   private CountDownLatch countDownLatch;
   public WorkA1(CountDownLatch countDownLatch) {
      this.countDownLatch = countDownLatch;
   }
   @Override
   public void run() {
      System.out.println("WorkA1 开始执行");
      try {
         Thread.sleep(2000);
      } catch (InterruptedException e) {
         e.printStackTrace();
      }finally {
         countDownLatch.countDown();
      }
      System.out.println("WorkA1 结束执行");
   }
}

Work2的改造跟Work1完全一样，就不再单独贴代码了。

看下主线程中的改造:

public class ThreedClient {
   public static void main(String[] args) {
      System.out.println("准备开始执行任务");
      CountDownLatch countDownLatch = new CountDownLatch(2);
      WorkA1 workA1 = new WorkA1(countDownLatch);
      WorkA2 workA2 = new WorkA2(countDownLatch);
      workA1.start();
      workA2.start();
      try {
         countDownLatch.await();
      }catch (InterruptedException e){
         e.printStackTrace();
      }
      System.out.println("执行完成了。");
   }
}

我们看下改造后的执行结果：

准备开始执行任务
WorkA1 开始执行
WorkA2 开始执行
WorkA2 结束执行
WorkA1 结束执行
执行完成了。

跟改造1中的效果一模一样。

多线程执行时，如果一个逻辑需要等若干个线程执行完成后再执行，怎么实现？

猜你喜欢