Semaphore(信号量)是JUC包中比较常用到的一个类,它是AQS共享模式的一个应用,可以允许多个线程同时对共享资源进行操作,并且可以有效的控制并发数,利用它可以很好的实现流量控制。Semaphore提供了一个许可证的概念,可以把这个许可证看作公共汽车车票,只有成功获取车票的人才能够上车,并且车票是有一定数量的,不可能毫无限制的发下去,这样就会导致公交车超载。所以当车票发完的时候(公交车以满载),其他人就只能等下一趟车了。如果中途有人下车,那么他的位置将会空闲出来,因此如果这时其他人想要上车的话就又可以获得车票了。利用Semaphore可以实现各种池,我们在本篇末尾将会动手写一个简易的数据库连接池。首先我们来看一下Semaphore的构造器。
1 //构造器1
2 public Semaphore(int permits) {
3 sync = new NonfairSync(permits);
4 }
5
6 //构造器2
7 public Semaphore(int permits, boolean fair) {
8 sync = fair ? new FairSync(permits) : new NonfairSync(permits);
9 }
Semaphore提供了两个带参构造器,没有提供无参构造器。这两个构造器都必须传入一个初始的许可证数量,使用构造器1构造出来的信号量在获取许可证时会采用非公平方式获取,使用构造器2可以通过参数指定获取许可证的方式(公平or非公平)。Semaphore主要对外提供了两类API,获取许可证和释放许可证,默认的是获取和释放一个许可证,也可以传入参数来同时获取和释放多个许可证。在本篇中我们只讲每次获取和释放一个许可证的情况。
1.获取许可证
1 //获取一个许可证(响应中断)
2 public void acquire() throws InterruptedException {
3 sync.acquireSharedInterruptibly(1);
4 }
5
6 //获取一个许可证(不响应中断)
7 public void acquireUninterruptibly() {
8 sync.acquireShared(1);
9 }
10
11 //尝试获取许可证(非公平获取)
12 public boolean tryAcquire() {
13 return sync.nonfairTryAcquireShared(1) >= 0;
14 }
15
16 //尝试获取许可证(定时获取)
17 public boolean tryAcquire(long timeout, TimeUnit unit) throws InterruptedException {
18 return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));
19 }
上面的API是Semaphore提供的默认获取许可证操作。每次只获取一个许可证,这也是现实生活中较常遇到的情况。除了直接获取还提供了尝试获取,直接获取操作在失败之后可能会阻塞线程,而尝试获取则不会。另外还需注意的是tryAcquire方法是使用非公平方式尝试获取的。在平时我们比较常用到的是acquire方法去获取许可证。下面我们就来看看它是怎样获取的。可以看到acquire方法里面直接就是调用sync.acquireSharedInterruptibly(1),这个方法是AQS里面的方法,我们在讲AQS源码系列文章的时候曾经讲过,现在我们再来回顾一下。
1 //以可中断模式获取锁(共享模式)
2 public final void acquireSharedInterruptibly(int arg) throws InterruptedException {
3 //首先判断线程是否中断, 如果是则抛出异常
4 if (Thread.interrupted()) {
5 throw new InterruptedException();
6 }
7 //1.尝试去获取锁
8 if (tryAcquireShared(arg) < 0) {
9 //2. 如果获取失败则进人该方法
10 doAcquireSharedInterruptibly(arg);
11 }
12 }
acquireSharedInterruptibly方法首先就是去调用tryAcquireShared方法去尝试获取,tryAcquireShared在AQS里面是抽象方法,FairSync和NonfairSync这两个派生类实现了该方法的逻辑。FairSync实现的是公平获取的逻辑,而NonfairSync实现的非公平获取的逻辑。
1 abstract static class Sync extends AbstractQueuedSynchronizer {
2 //非公平方式尝试获取
3 final int nonfairTryAcquireShared(int acquires) {
4 for (;;) {
5 //获取可用许可证
6 int available = getState();
7 //获取剩余许可证
8 int remaining = available - acquires;
9 //1.如果remaining小于0则直接返回remaining
10 //2.如果remaining大于0则先更新同步状态再返回remaining
11 if (remaining < 0 || compareAndSetState(available, remaining)) {
12 return remaining;
13 }
14 }
15 }
16 }
17
18 //非公平同步器
19 static final class NonfairSync extends Sync {
20 private static final long serialVersionUID = -2694183684443567898L;
21
22 NonfairSync(int permits) {
23 super(permits);
24 }
25
26 //尝试获取许可证
27 protected int tryAcquireShared(int acquires) {
28 return nonfairTryAcquireShared(acquires);
29 }
30 }
31
32 //公平同步器
33 static final class FairSync extends Sync {
34 private static final long serialVersionUID = 2014338818796000944L;
35
36 FairSync(int permits) {
37 super(permits);
38 }
39
40 //尝试获取许可证
41 protected int tryAcquireShared(int acquires) {
42 for (;;) {
43 //判断同步队列前面有没有人排队
44 if (hasQueuedPredecessors()) {
45 //如果有的话就直接返回-1,表示尝试获取失败
46 return -1;
47 }
48 //获取可用许可证
49 int available = getState();
50 //获取剩余许可证
51 int remaining = available - acquires;
52 //1.如果remaining小于0则直接返回remaining
53 //2.如果remaining大于0则先更新同步状态再返回remaining
54 if (remaining < 0 || compareAndSetState(available, remaining)) {
55 return remaining;
56 }
57 }
58 }
59 }
这里需要注意的是NonfairSync的tryAcquireShared方法直接调用的是nonfairTryAcquireShared方法,这个方法是在父类Sync里面的。非公平获取锁的逻辑是先取出当前同步状态(同步状态表示许可证个数),将当前同步状态减去参入的参数,如果结果不小于0的话证明还有可用的许可证,那么就直接使用CAS操作更新同步状态的值,最后不管结果是否小于0都会返回该结果值。这里我们要了解tryAcquireShared方法返回值的含义,返回负数表示获取失败,零表示当前线程获取成功但后续线程不能再获取,正数表示当前线程获取成功并且后续线程也能够获取。我们再来看acquireSharedInterruptibly方法的代码。
1 //以可中断模式获取锁(共享模式)
2 public final void acquireSharedInterruptibly(int arg) throws InterruptedException {
3 //首先判断线程是否中断, 如果是则抛出异常
4 if (Thread.interrupted()) {
5 throw new InterruptedException();
6 }
7 //1.尝试去获取锁
8 //负数:表示获取失败
9 //零值:表示当前线程获取成功, 但是后继线程不能再获取了
10 //正数:表示当前线程获取成功, 并且后继线程同样可以获取成功
11 if (tryAcquireShared(arg) < 0) {
12 //2. 如果获取失败则进人该方法
13 doAcquireSharedInterruptibly(arg);
14 }
15 }
如果返回的remaining小于0的话就代表获取失败,因此tryAcquireShared(arg) < 0就为true,所以接下来就会调用doAcquireSharedInterruptibly方法,这个方法我们在讲AQS的时候讲过,它会将当前线程包装成结点放入同步队列尾部,并且有可能挂起线程。这也是当remaining小于0时线程会排队阻塞的原因。而如果返回的remaining>=0的话就代表当前线程获取成功,因此tryAcquireShared(arg) < 0就为flase,所以就不会再去调用doAcquireSharedInterruptibly方法阻塞当前线程了。以上是非公平获取的整个逻辑,而公平获取时仅仅是在此之前先去调用hasQueuedPredecessors方法判断同步队列是否有人在排队,如果有的话就直接return -1表示获取失败,否则才继续执行下面和非公平获取一样的步骤。
2.释放许可证
1 //释放一个许可证
2 public void release() {
3 sync.releaseShared(1);
4 }
调用release方法是释放一个许可证,它的操作很简单,就调用了AQS的releaseShared方法,我们来看看这个方法。
1 //释放锁的操作(共享模式)
2 public final boolean releaseShared(int arg) {
3 //1.尝试去释放锁
4 if (tryReleaseShared(arg)) {
5 //2.如果释放成功就唤醒其他线程
6 doReleaseShared();
7 return true;
8 }
9 return false;
10 }
AQS的releaseShared方法首先调用tryReleaseShared方法尝试释放锁,这个方法的实现逻辑在子类Sync里面。
1 abstract static class Sync extends AbstractQueuedSynchronizer {
2 ...
3 //尝试释放操作
4 protected final boolean tryReleaseShared(int releases) {
5 for (;;) {
6 //获取当前同步状态
7 int current = getState();
8 //将当前同步状态加上传入的参数
9 int next = current + releases;
10 //如果相加结果小于当前同步状态的话就报错
11 if (next < current) {
12 throw new Error("Maximum permit count exceeded");
13 }
14 //以CAS方式更新同步状态的值, 更新成功则返回true, 否则继续循环
15 if (compareAndSetState(current, next)) {
16 return true;
17 }
18 }
19 }
20 ...
21 }
可以看到tryReleaseShared方法里面采用for循环进行自旋,首先获取同步状态,将同步状态加上传入的参数,然后以CAS方式更新同步状态,更新成功就返回true并跳出方法,否则就继续循环直到成功为止,这就是Semaphore释放许可证的流程。
这里是引用