Java中用来协调多个线程之间的同步,或者说起到线程之间的通信。这里介绍三种工具类:CountDownLatch(线程计数器)、CyclicBarrier(回环栅栏), Semaphore(信号量)。
- CountDownLatch(线程计数器)
通过计数器做减法,以此来达到效果。比如一个任务A,它要等待其他4个任务执行完毕之后才能执行,此时就可以利用CountDownLatch 来实现这种功能了。比如有一个任务A,它要等待其他4个任务执行完毕之后才能执行。
public class CountDownLatchDemo {
public static void main(String[] args) throws InterruptedException {
final CountDownLatch latch = new CountDownLatch(2);
new Thread(){
@SneakyThrows
@Override
public void run() {
System.out.println("子线程"+Thread.currentThread().getName() + "正在执行!");
Thread.sleep(3000);
System.out.println("子线程"+Thread.currentThread().getName() + "执行完毕!");
latch.countDown();
}
}.start();
new Thread(){
@SneakyThrows
public void run(){
System.out.println("子线程"+Thread.currentThread().getName() + "正在执行!");
Thread.sleep(3000);
System.out.println("子线程"+Thread.currentThread().getName() + "执行完毕!");
latch.countDown();
}
}.start();
System.out.println("等待2个子线程执行完毕....");
latch.await();
Thread.sleep(3000);
System.out.println("继续执行主线程");
}
}
主线程只有等待线程1和线程2执行完毕之后才会继续执行。
- CyclicBarrier(回环栅栏)
CyclicBarrier 可以实现让一组线程等待至某个状态之后再全部同时执行。当所有等待线程都被释放以后,CyclicBarrier 可以被重用。 当调用await()方法之后,线程就处于barrier了。
CyclicBarrier中最重要的方法就是await方法,它有2个重载版本:
1、public int await() 用来挂起当前线程,直至所有线程都到达barrier状态再同时执行后续任务。
2、public int await(long timeout, TimeUnit unit): 让这些线程等待至一定的时间,如果还有线程没有到大barrier状态,就直接让到大barrier的线程执行后续任务。
public class CyclicBarrierDemo {
static Integer count = 0;
/***
* CyclicBarrier 回环栅栏
* @param args
*/
public static void main(String[] args) {
int N = 4;
CyclicBarrier cyclicBarrier = new CyclicBarrier(N);
for (int i = 0; i < N; i++){
new Writer(cyclicBarrier).start();
}
}
static class Writer extends Thread{
private CyclicBarrier cyclicBarrier;
public Writer(CyclicBarrier cyclicBarrier){
this.cyclicBarrier = cyclicBarrier;
}
@SneakyThrows
@Override
public void run() {
Thread.sleep(2000);
System.out.println("线程"+Thread.currentThread().getName()+"写入完毕,等待其他线程写入完毕!");
cyclicBarrier.await(); //将当前线程设置为barrier状态,等待其他线程都到达barrier状态之后才执行后续操作
Thread.sleep(1000);
System.out.println("全部执行完毕!");
}
}
}
- Semaphore(信号量)
Semaphore可以控制同时访问的线程个数,通过acquire()获取一个许可,如果没有就等待,而release()释放一个许可。
public class SemaphoreDemo {
public static void main(String[] args) {
int N = 8;
Semaphore semaphore = new Semaphore(4); //信号量, 只有4个机器
for (int i = 0; i < N; i++){
new Worker(i, semaphore).start();
}
}
// 消费者
static class Worker extends Thread{
private int number; //当前线程的序号
private Semaphore semaphore; //信号量
public Worker(int number, Semaphore semaphore){
this.number = number;
this.semaphore = semaphore;
}
@SneakyThrows
@Override
public void run() {
semaphore.acquire(); //尝试去使用信号量,用来获取一个许可,若无许可就一直等待,直到获得许可
System.out.println("工人"+this.number+"占用一个机器在生产");
Thread.sleep(2000);
System.out.println("工人"+this.number+"释放机器");
semaphore.release();
}
}
}
总结:CountDownLatch和CyclicBarrier都能够实现线程间的等待,只不过侧重点不一样。CountDownLatch一般用于某个线程A等待若干个其他线程执行完任务之后,它才执行,而CyclicBarrier则用于一组线程互相等待至某个状态之后,然后一起执行。
Semaphore用于同步互斥的锁。
2022新年第一篇blog,新年加油!