JUC的四个并发工具类
本篇文章主要记录了JUC的四个并发工具类,闭锁CountDownlatch、栅栏CyclicBarrier、信号量Semaphore、交换器Exchanger。CountDownlatch通常用于主线程等待其他任务线程执行完毕的场景;CyclicBarrier主要阻塞当前线程,等待其他线程(大家无论谁先跑到A点,必须要等其他线程也到达了A点,大家才能继续)。信号量Semaphore可以用来控制同时访问特定资源的线程数量(比如100个线程只能有10个线程可以获得MySQL连接)。交换器Exchanger很少用,只适用于两个线程在同步点交换数据的场景(如下图)。
闭锁CountDownlatch
CountDownLatch也叫闭锁,使得一(多)个主线程必须等待其他线程完成操作后再执行。 CountDownLatch内部维护一个计数器(父类的int state),主线程先执行await方法,如果此时计数器大于0,则阻塞等待。当一个线程完成任务后,计数器值减1。直到计数器为0时,表示所有的线程已经完成任务,等待的主线程被唤醒继续执行。
CountDownLatch实现主要基于Java同步器AQS,关于AQS的分析可以看我这篇文章: 《ReentrantLock与AQS》 ,countDown方法核心实现如下:
1public void countDown() {
2 sync.releaseShared(1);
3}
4
5public final boolean releaseShared(int arg) {
6 // countDownLatch -> tryReleaseShared
7 if (tryReleaseShared(arg)) {
8 doReleaseShared();
9 return true;
10 }
11 return false;
12}
13
14private void doReleaseShared() {
15 for (;;) {
16 Node h = head;
17 if (h != null && h != tail) {
18 int ws = h.waitStatus;
19 if (ws == Node.SIGNAL) {
20 if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))
21 continue; // loop to recheck cases
22 unparkSuccessor(h);
23 }
24 else if (ws == 0 &&
25 !compareAndSetWaitStatus(h, 0, Node.PROPAGATE))
26 continue; // loop on failed CAS
27 }
28 if (h == head) // loop if head changed
29 break;
30 }
31}
CountDownlatch的使用案例,下面使用三个线程来打印三个List,三个线程任务都完成得时候才输出Print Task Finish!
1import java.util.Arrays;
2import java.util.List;
3import java.util.concurrent.CountDownLatch;
4
5public class CountDownLatchDemo {
6 public static void main(String[] args) {
7 List<String> list1 = Arrays.asList("AAA", "BBB", "CCC");
8 List<String> list2 = Arrays.asList("DDD", "EEE", "FFF");
9 List<String> list3 = Arrays.asList("GGG", "HHH", "III");
10
11 CountDownLatch countDownLatch = new CountDownLatch(3);
12 new Thread(()->{
13 for (String string: list1) {
14 System.out.println(Thread.currentThread().getName() + ":" + string);
15 }
16 countDownLatch.countDown();
17 }).start();
18 new Thread(()->{
19 for (String string: list2) {
20 System.out.println(Thread.currentThread().getName() + ":" + string);
21 }
22 countDownLatch.countDown();
23 }).start();
24 new Thread(()->{
25 for (String string: list3) {
26 System.out.println(Thread.currentThread().getName() + ":" + string);
27 }
28 countDownLatch.countDown();
29 }).start();
30 try {
31 countDownLatch.await();
32 } catch (InterruptedException e) {
33 e.printStackTrace();
34 }
35 System.out.println("Print Task Finish!");
36 }
37}
栅栏CyclicBarrier
CyclicBarrier:阻塞当前线程,等待其他线程。等待其它线程,且会阻塞自己当前线程,所有线程必须同时到达栅栏位置后才能继续执行;所有线程到达栅栏处,可以触发执行另外一个预先设置的线程。
1import java.util.concurrent.BrokenBarrierException;
2import java.util.concurrent.CyclicBarrier;
3
4public class CyclicBarrierDemo {
5 public static void main(String[] args) throws InterruptedException {
6 new CyclicBarrierDemo().go();
7 }
8
9 private void go() throws InterruptedException {
10 //初始化栅栏得参与者数为3
11 CyclicBarrier cyclicBarrier = new CyclicBarrier(3);
12 new Thread(new Task(cyclicBarrier), "Thread1").start();
13 Thread.sleep(1000);
14 new Thread(new Task(cyclicBarrier), "Thread2").start();
15 Thread.sleep(1000);
16 new Thread(new Task(cyclicBarrier), "Thread3").start();
17 }
18
19 class Task implements Runnable{
20 private CyclicBarrier cyclicBarrier;
21
22 public Task(CyclicBarrier cyclicBarrier) {
23 this.cyclicBarrier = cyclicBarrier;
24 }
25
26 @Override
27 public void run() {
28 System.out.println("线程" + Thread.currentThread().getName()
29 + "已经送达"
30 + System.currentTimeMillis());
31 try {
32 cyclicBarrier.await();
33 } catch (BrokenBarrierException e) {
34 e.printStackTrace();
35 } catch (InterruptedException e) {
36 e.printStackTrace();
37 }
38 System.out.println("线程" + Thread.currentThread().getName()
39 + "开始处理"
40 + System.currentTimeMillis());
41 }
42 }
43}
信号量Semaphore
Semaphore也叫信号量,在JDK1.5被引入,可以用来控制同时访问特定资源的线程数量,通过协调各个线程,以保证合理的使用资源。Semaphore内部维护了一组虚拟的许可,许可的数量可以通过构造函数的参数指定。
访问特定资源前,必须使用acquire方法获得许可,如果许可数量为0,该线程则一直阻塞,直到有可用许可。访问资源后,使用release释放许可。Semaphore和ReentrantLock类似,获取许可有公平策略和非公平许可策略,默认情况下使用非公平策略。
信号量Semaphore得应用场景:Semaphore可以用来做流量分流,特别是对公共资源有限的场景,比如数据库连接。假设有这个的需求,读取几万个文件的数据到数据库中,由于文件读取是IO密集型任务,可以启动几十个线程并发读取,但是数据库连接数只有10个,这时就必须控制最多只有10个线程能够拿到数据库连接进行操作。这个时候,就可以使用Semaphore做流量控制。
1package thread_study;
2
3import java.util.concurrent.ExecutorService;
4import java.util.concurrent.Executors;
5import java.util.concurrent.Semaphore;
6
7public class SemaphoreDemo {
8 public static void main(String[] args) {
9 ExecutorService pool = Executors.newCachedThreadPool();
10 //只能5个线程同时访问
11 Semaphore semaphore = new Semaphore(5);
12 //模拟20个客户端访问
13 for (int i = 0; i < 20; i++) {
14 final int NO = i;
15 pool.execute(()->{
16 try {
17 //获取许可
18 semaphore.acquire();
19 System.out.println("Accessing: " + NO);
20 Thread.sleep((long)(Math.random() * 10000));
21 //访问完毕后释放
22 semaphore.release();
23 } catch (InterruptedException e) {
24 e.printStackTrace();
25 }
26 });
27 }
28 //退出线程池
29 pool.shutdown();
30 }
31}
交换器Exchanger
Exchanger(交换者)是一个用于线程间数据交换协作的工具类。它提供一个同步点,在这个同步点多个线程间两两之间线程可以交换彼此的数据。这两个线程通过exchange方法交换数据, 如果第一个线程先执行exchange方法,它会一直等待第二个线程也执行exchange方法,当两个线程都到达同步点时,这两个线程就可以交换数据,将本线程生产出来的数据传递给对方。
1import java.util.concurrent.Exchanger;
2import java.util.concurrent.ExecutorService;
3import java.util.concurrent.Executors;
4import java.util.concurrent.TimeUnit;
5
6public class ExchangerDemo {
7 public static void main(String[] args) {
8 Exchanger<String> exchanger = new Exchanger<>();
9 ExecutorService threadPool = Executors.newFixedThreadPool(2);
10 threadPool.execute(()->{
11 try {
12 //男生对女生说的话
13 String girl = exchanger.exchange("我其实暗恋你很久了....");
14 System.out.println("女生说: " + girl);
15 } catch (InterruptedException e) {
16 e.printStackTrace();
17 }
18 });
19
20 threadPool.execute(()->{
21 try {
22 System.out.println("女生慢慢的从教室走出来.... ");
23 TimeUnit.SECONDS.sleep(3);
24 //男生对女生说的话
25 String boy = exchanger.exchange("我很喜欢你....");
26 System.out.println("男生说:" + boy);
27 } catch (InterruptedException e) {
28 e.printStackTrace();
29 }
30 });
31
32 threadPool.shutdown();
33 }
34}