【Java多线程】JUC包下的工具类CountDownLatch、CyclicBarrier和Semaphore

前言

JUC中为了满足在并发编程中不同的需求，提供了几个工具类供我们使用，分别是CountDownLatch、CyclicBarrier和Semaphore，其原理都是使用了AQS来实现，下面分别进行介绍。

CountDownLatch

CountDownLatch的主要作用是利用计数来保证线程的执行顺序（我自己的理解），有点像倒计时，当计数为0时某个线程才能开始执行。

CountDownLatch的主要方法很简单易用，包括：

• CountDownLatch(int count) ： 构造方法，需要传入计数的初始值
• void await() ： 调用者线程会被挂起，直到计数为0时才能执行
• boolean await(long timeout, TimeUnit unit) ： 同上，但是加入了超时参数，如果超时了计数还不为0，也会照样执行，避免了一直阻塞
• void countDown() ： 计数减一

示例代码：CountDownLatchTest.java

package juc.util;

import java.util.concurrent.CountDownLatch;

/**
 * Created by puyangsky on 2017/1/7.
 */
public class CountDownLatchTest {
    public static void main(String[] args) {
        final CountDownLatch latch = new CountDownLatch(2);
		
		//用了匿名类创建线程
        new Thread() {
            public void run() {
                try {
                    System.out.println("子线程"+Thread.currentThread().getName()+"正在执行");
                    Thread.sleep(2000);
                    System.out.println("子线程"+Thread.currentThread().getName()+"执行完毕");
                    latch.countDown();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }.start();

        new Thread() {
            public void run() {
                try {
                    System.out.println("子线程"+Thread.currentThread().getName()+"正在执行");
                    Thread.sleep(2000);
                    System.out.println("子线程"+Thread.currentThread().getName()+"执行完毕");
                    latch.countDown();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }.start();

        System.out.println("主线程"+Thread.currentThread().getName()+"正在执行");
        try {
            latch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println("主线程"+Thread.currentThread().getName()+"执行完毕");
    }
}

结果：

主线程main正在执行
子线程Thread-0正在执行
子线程Thread-1正在执行
子线程Thread-0执行完毕
子线程Thread-1执行完毕
主线程main执行完毕

可以从结果中看到主线程本来是最先执行完，结果需要等两个子线程执行完才结束，就是因为调用了await方法。

CyclicBarrier

CyclicBarrier的意思是回环栅栏，作用是让一组线程同时到达某个时间节点。提供的重要方法如下：

• CyclicBarrier(int parties) ： 构造方法，传入线程组的数量
• CyclicBarrier(int parties, Runnable barrierAction) ： 构造方法，传入线程组的数量和当线程达到时间节点后要做的操作（由其中的某一个线程去执行）
• int await() : 挂起当前线程，直到所有线程组中的线程都完成后继续执行，返回当前线程到达的次序
• int await(long timeout, TimeUnit unit) ： 加了一个超时参数

示例代码 CyclicBarrierTest.java：

package juc.util;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;

/**
 * Created by puyangsky on 2017/1/7.
 */
public class CyclicBarrierTest {
    public static void main(String[] args) {
        CyclicBarrier barrier = new CyclicBarrier(5, new Runnable() {
            @Override
            public void run() {
                System.out.println("当前线程"+Thread.currentThread().getName());
            }
        });
        for (int i=0;i<5;i++) {
            new Task(barrier).start();
        }
    }

    static class Task extends Thread {
        private CyclicBarrier barrier;

        public Task(CyclicBarrier barrier) {
            this.barrier = barrier;
        }

        public void run() {
            System.out.println("子线程"+Thread.currentThread().getName()+"正在执行");
            try {
                Thread.sleep(3000);
                System.out.println("子线程"+Thread.currentThread().getName()+"执行完成");
                barrier.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
            System.out.println("全部线程执行完成");
        }
    }
}

结果：

子线程Thread-0正在执行
子线程Thread-3正在执行
子线程Thread-2正在执行
子线程Thread-1正在执行
子线程Thread-4正在执行
子线程Thread-3执行完成
子线程Thread-2执行完成
子线程Thread-0执行完成
子线程Thread-1执行完成
子线程Thread-4执行完成
当前线程Thread-4
全部线程执行完成
全部线程执行完成
全部线程执行完成
全部线程执行完成
全部线程执行完成

可以看到这一组线程是同步的去执行的。

Semaphore

Semaphore的意思是信号量，其作用是提供一个许可范围，只有获得了许可才能继续执行。

• Semaphore(int permits)：构造方法，需要传入许可数
• Semaphore(int permits, boolean fair)：fair为true时使用公平锁，false时使用非公平锁，具体在介绍AQS时讲解
• void acquire()：获得许可，可中断
• void acquireUninterruptibly()：不可中断的获取
• void release()：释放许可

示例 SemaphoreTest.java

package juc.util;

import java.util.concurrent.Semaphore;

/**
 * Created by puyangsky on 2017/1/7.
 */
public class SemaphoreTest {

    public static void main(String[] args) {
        Semaphore semaphore = new Semaphore(5);
        for (int i = 0;i < 10;i++) {
            new Worker(i, semaphore).start();
        }
    }

    static class Worker extends Thread {
        private Semaphore semaphore;
        private int num;

        public Worker(int num, Semaphore semaphore) {
            this.semaphore = semaphore;
            this.num = num;
        }

        public void run() {
            try {

                semaphore.acquire();
                System.out.println("工人" + (this.num + 1) + "占用一个机器在生产...");
                Thread.sleep(2000);

                System.out.println("工人" + (this.num + 1) + "释放一个机器...");
                semaphore.release();

            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

    }
}

结果

工人1占用一个机器在生产...
工人3占用一个机器在生产...
工人4占用一个机器在生产...
工人5占用一个机器在生产...
工人7占用一个机器在生产...
工人1释放一个机器...
工人5释放一个机器...
工人4释放一个机器...
工人7释放一个机器...
工人3释放一个机器...
工人8占用一个机器在生产...
工人9占用一个机器在生产...
工人10占用一个机器在生产...
工人6占用一个机器在生产...
工人2占用一个机器在生产...
工人9释放一个机器...
工人6释放一个机器...
工人10释放一个机器...
工人8释放一个机器...
工人2释放一个机器...

可以看到每次只有五个线程能运行，其他线程只有等待这五个线程释放许可后才能运行。

参考文章：

• Java并发编程：CountDownLatch、CyclicBarrier和Semaphore
• jdk1.7