CountDownLatch简单使用

CountDownLatch介绍

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

使用场景

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

示例

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