java并发编程CountDownLatch使用

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

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

主要方法

 public CountDownLatch(int count);

 public void countDown();

 public void await() throws InterruptedException

构造方法参数指定了计数的次数

countDown方法，当前线程调用此方法，则计数减一

awaint方法，调用此方法会一直阻塞当前线程，直到计时器的值为0

示例

1. public class CountDownLatchDemo {  
2.     final static SimpleDateFormat sdf=new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");  
3.     public static void main(String[] args) throws InterruptedException {  
4.         CountDownLatch latch=new CountDownLatch(2);//两个工人的协作  
5.         Worker worker1=new Worker("zhang san", 5000, latch);  
6.         Worker worker2=new Worker("li si", 8000, latch);  
7.         worker1.start();//  
8.         worker2.start();//  
9.         latch.await();//等待所有工人完成工作  
10.         System.out.println("all work done at "+sdf.format(new Date()));  
11.     }  
12.       
13.       
14.     static class Worker extends Thread{  
15.         String workerName;   
16.         int workTime;  
17.         CountDownLatch latch;  
18.         public Worker(String workerName ,int workTime ,CountDownLatch latch){  
19.              this.workerName=workerName;  
20.              this.workTime=workTime;  
21.              this.latch=latch;  
22.         }  
23.         public void run(){  
24.             System.out.println("Worker "+workerName+" do work begin at "+sdf.format(new Date()));  
25.             doWork();//工作了  
26.             System.out.println("Worker "+workerName+" do work complete at "+sdf.format(new Date()));  
27.             latch.countDown();//工人完成工作，计数器减一  
28.   
29.         }  
30.           
31.         private void doWork(){  
32.             try {  
33.                 Thread.sleep(workTime);  
34.             } catch (InterruptedException e) {  
35.                 e.printStackTrace();  
36.             }  
37.         }  
38.     }  
39.       
40.        
41. }  

 

输出：

Worker zhang san do work begin at 2011-04-14 11:05:11
Worker li si do work begin at 2011-04-14 11:05:11
Worker zhang san do work complete at 2011-04-14 11:05:16
Worker li si do work complete at 2011-04-14 11:05:19
all work done at 2011-04-14 11:05:19

例2

 1 import java.util.concurrent.CountDownLatch;
 2 
 3 public class Sample {
 4     /**
 5      * 计数器，用来控制线程
 6      * 传入参数2，表示计数器计数为2
 7      */
 8     private final static CountDownLatch mCountDownLatch = new CountDownLatch(2);
 9 
10     /**
11      * 示例工作线程类
12      */
13     private static class WorkingThread extends Thread {
14         private final String mThreadName;
15         private final int mSleepTime;
16         public WorkingThread(String name, int sleepTime) {
17             mThreadName = name;
18             mSleepTime = sleepTime;
19         }
20         
21         @Override
22         public void run() {
23             System.out.println("[" + mThreadName + "] started!");
24             try {  
25                     Thread.sleep(mSleepTime);  
26             } catch (InterruptedException e) {  
27                     e.printStackTrace();  
28             }
29             mCountDownLatch.countDown();
30             System.out.println("[" + mThreadName + "] end!"); 
31         }
32     }
33     
34     /**
35      * 示例线程类
36      */
37     private static class SampleThread extends Thread {
38         
39         @Override
40         public void run() {
41             System.out.println("[SampleThread] started!");
42             try {
43                 // 会阻塞在这里等待 mCountDownLatch 里的count变为0；
44                 // 也就是等待另外的WorkingThread调用countDown()
45                 mCountDownLatch.await();
46             } catch (InterruptedException e) {
47                 
48             }
49             System.out.println("[SampleThread] end!");
50         }
51     }
52     
53     public static void main(String[] args) throws Exception {
54         // 最先run SampleThread
55         new SampleThread().start();
56         // 运行两个工作线程
57         // 工作线程1运行5秒
58         new WorkingThread("WorkingThread1", 5000).start();
59         // 工作线程2运行2秒
60         new WorkingThread("WorkingThread2", 2000).start();
61     }
62 }

 

运行结果：

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

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