CountDownLatch、CyclicBarrier和 Semaphore
CountDownLatch类位于java.util.concurrent包下,利用它可以实现类似计数器的功能。比如有一个任务A,它要等待其他4个任务执行完毕之后才能执行,此时就可以利用CountDownLatch来实现这种功能了。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
public class Test { 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( 3000 ); 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( 3000 ); System.out.println( "子线程" +Thread.currentThread().getName()+ "执行完毕" ); latch.countDown(); } catch (InterruptedException e) { e.printStackTrace(); } }; }.start(); try { System.out.println( "等待2个子线程执行完毕..." ); latch.await(); System.out.println( "2个子线程已经执行完毕" ); System.out.println( "继续执行主线程" ); } catch (InterruptedException e) { e.printStackTrace(); } } } |
执行结果:
1
2
3
4
5
6
7
|
线程Thread- 0 正在执行 线程Thread- 1 正在执行 等待 2 个子线程执行完毕... 线程Thread- 0 执行完毕 线程Thread- 1 执行完毕 2 个子线程已经执行完毕 继续执行主线程 |
CyclicBarrier字面意思回环栅栏,通过它可以实现让一组线程等待至某个状态之后再全部同时执行。叫做回环是因为当所有等待线程都被释放以后,CyclicBarrier可以被重用。我们暂且把这个状态就叫做barrier,当调用await()方法之后,线程就处于barrier了。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
public class Test { public static void main(String[] args) { int N = 4 ; CyclicBarrier barrier = new CyclicBarrier(N); for ( int i= 0 ;i<N;i++) new Writer(barrier).start(); } static class Writer extends Thread{ private CyclicBarrier cyclicBarrier; public Writer(CyclicBarrier cyclicBarrier) { this .cyclicBarrier = cyclicBarrier; } @Override public void run() { System.out.println( "线程" +Thread.currentThread().getName()+ "正在写入数据..." ); try { Thread.sleep( 5000 ); //以睡眠来模拟写入数据操作 System.out.println( "线程" +Thread.currentThread().getName()+ "写入数据完毕,等待其他线程写入完毕" ); cyclicBarrier.await(); } catch (InterruptedException e) { e.printStackTrace(); } catch (BrokenBarrierException e){ e.printStackTrace(); } System.out.println( "所有线程写入完毕,继续处理其他任务..." ); } } } |
执行结果:
1
2
3
4
5
6
7
8
9
10
11
12
|
线程Thread- 0 正在写入数据... 线程Thread- 3 正在写入数据... 线程Thread- 2 正在写入数据... 线程Thread- 1 正在写入数据... 线程Thread- 2 写入数据完毕,等待其他线程写入完毕 线程Thread- 0 写入数据完毕,等待其他线程写入完毕 线程Thread- 3 写入数据完毕,等待其他线程写入完毕 线程Thread- 1 写入数据完毕,等待其他线程写入完毕 所有线程写入完毕,继续处理其他任务... 所有线程写入完毕,继续处理其他任务... 所有线程写入完毕,继续处理其他任务... 所有线程写入完毕,继续处理其他任务... |
Semaphore翻译成字面意思为 信号量,Semaphore可以控同时访问的线程个数,通过 acquire() 获取一个许可,如果没有就等待,而 release() 释放一个许可。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
public class Test { public static void main(String[] args) { int N = 8 ; //工人数 Semaphore semaphore = new Semaphore( 5 ); //机器数目 for ( int i= 0 ;i<N;i++) new Worker(i,semaphore).start(); } static class Worker extends Thread{ private int num; private Semaphore semaphore; public Worker( int num,Semaphore semaphore){ this .num = num; this .semaphore = semaphore; } @Override public void run() { try { semaphore.acquire(); System.out.println( "工人" + this .num+ "占用一个机器在生产..." ); Thread.sleep( 2000 ); System.out.println( "工人" + this .num+ "释放出机器" ); semaphore.release(); } catch (InterruptedException e) { e.printStackTrace(); } } } } |
执行结果:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
工人 0 占用一个机器在生产... 工人 1 占用一个机器在生产... 工人 2 占用一个机器在生产... 工人 4 占用一个机器在生产... 工人 5 占用一个机器在生产... 工人 0 释放出机器 工人 2 释放出机器 工人 3 占用一个机器在生产... 工人 7 占用一个机器在生产... 工人 4 释放出机器 工人 5 释放出机器 工人 1 释放出机器 工人 6 占用一个机器在生产... 工人 3 释放出机器 工人 7 释放出机器 工人 6 释放出机器 |
下面对上面说的三个辅助类进行一个总结:
1)CountDownLatch和CyclicBarrier都能够实现线程之间的等待,只不过它们侧重点不同:
CountDownLatch一般用于某个线程A等待若干个其他线程执行完任务之后,它才执行;
而CyclicBarrier一般用于一组线程互相等待至某个状态,然后这一组线程再同时执行;
另外,CountDownLatch是不能够重用的,而CyclicBarrier是可以重用的。
2)Semaphore其实和锁有点类似,它一般用于控制对某组资源的访问权限。