使用 CountDownLatch 控制多个线程执行顺序
阅读原文时间:2024年07月05日阅读:1

已同步更新至:http://dxjia.cn/2015/08/countdownlatch-use/

有时候会有这样的需求,多个线程同时工作,然后其中几个可以随意并发执行,但有一个线程需要等其他线程工作结束后,才能开始。举个例子,开启多个线程分块下载一个大文件,每个线程只下载固定的一截,最后由另外一个线程来拼接所有的分段,那么这时候我们可以考虑使用CountDownLatch来控制并发。

CountDownLatch是JAVA提供在java.util.concurrent包下的一个辅助类,可以把它看成是一个计数器,其内部维护着一个count计数,只不过对这个计数器的操作都是原子操作,同时只能有一个线程去操作这个计数器,CountDownLatch通过构造函数传入一个初始计数值,调用者可以通过调用CounDownLatch对象的cutDown()方法,来使计数减1;如果调用对象上的await()方法,那么调用者就会一直阻塞在这里,直到别人通过cutDown方法,将计数减到0,才可以继续执行。

示例

import java.util.concurrent.CountDownLatch;

public class Sample {
/**
* 计数器,用来控制线程
* 传入参数2,表示计数器计数为2
*/
private final static CountDownLatch mCountDownLatch = new CountDownLatch(2);

 /\*\*  
  \* 示例工作线程类  
  \*/  
 private static class WorkingThread extends Thread {  
     private final String mThreadName;  
     private final int mSleepTime;  
     public WorkingThread(String name, int sleepTime) {  
         mThreadName = name;  
         mSleepTime = sleepTime;  
     }

     @Override  
     public void run() {  
         System.out.println("\[" + mThreadName + "\] started!");  
         try {  
                 Thread.sleep(mSleepTime);  
         } catch (InterruptedException e) {  
                 e.printStackTrace();  
         }  
         mCountDownLatch.countDown();  
         System.out.println("\[" + mThreadName + "\] end!");  
     }  
 }

 /\*\*  
  \* 示例线程类  
  \*/  
 private static class SampleThread extends Thread {

     @Override  
     public void run() {  
         System.out.println("\[SampleThread\] started!");  
         try {  
             // 会阻塞在这里等待 mCountDownLatch 里的count变为0;  
             // 也就是等待另外的WorkingThread调用countDown()  
             mCountDownLatch.await();  
         } catch (InterruptedException e) {

         }  
         System.out.println("\[SampleThread\] end!");  
     }  
 }

 public static void main(String\[\] args) throws Exception {  
     // 最先run SampleThread  
     new SampleThread().start();  
     // 运行两个工作线程  
     // 工作线程1运行5秒  
     new WorkingThread("WorkingThread1", 5000).start();  
     // 工作线程2运行2秒  
     new WorkingThread("WorkingThread2", 2000).start();  
 }  

}

运行结果:

[SampleThread] started!
[WorkingThread1] started!
[WorkingThread2] started!
[WorkingThread2] end!
[WorkingThread1] end!
[SampleThread] end!

达到了目的。当然还有其他方式可以做到这样的效果,本文仅仅是介绍了一种使用CountDownLatch的方式。

手机扫一扫

移动阅读更方便

阿里云服务器
腾讯云服务器
七牛云服务器

你可能感兴趣的文章