1、背景
在使用锁的时候遇到了Semaphore,之前一直有个疑惑,就是Semaphore创建时要指定信号量数,例如Semaphore(1),那说明同时只能有一个线程拥有锁。但Semaphore(0)也可以让线程拥有锁,这个一直是我困惑的地方,它信号量都为0了,为什么还能具有锁的功能。
2、结论
Semaphore(0)之所以拥有锁的功能,在于其释放时会增加信号量证书。进而让等待线程被唤醒执行。如果没有释放操作,使用Semaphore(0)的所有线程都会卡住进入等待状态。而不像Semaphore(n),会有n个线程正常执行。
3、源码追踪
3.1 代码demo
下面是关于源码的追踪,首先编辑一个demo:
public static void main(String[] args) throws Exception{
Semaphore semaphore = new Semaphore(0);
Thread thread1 = new Thread(()->{
try {
semaphore.acquire();
System.out.println("threa1");
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
});
thread1.start();
Thread thread2 = new Thread(()->{
try {
semaphore.acquire();
System.out.println("threa2");
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
});
thread2.start();
Thread.sleep(3000);
System.out.println("start test semaphore");
semaphore.release();
semaphore.release();
}
打印结果如下:
可以看到,只有在执行semaphore.release()时才会唤醒等待线程。下面我们分别看下semaphore.acquire()和semaphore.release()的具体逻辑。
3.2 acquire()逻辑
首先进入acquire方法:
在这可以看到其回先判断线程是否中断,不中断再尝试获取锁,我们接着向下看:
注意Semaphone中的锁对象是自身的内部类,所以调用的方法也是NonfairSync类对象里的方法,这些在debug中都可以看到,具体的不细讲。我们接着向下看:
这里会获取可用的信号量证书和剩余的信号量证书(信号量证书在初始化的时候设置),如果剩余的信号量证书小于0或者成功获取到证书,则会返回,否则则会一直在for循环中等待获取信号量证书。因为我们初始设置的信号量证书为0,这里调用acquire方法分配一个证书后,剩余值为-1,所以会直接返回,我们接着向下看:
当信号量证书小于0时,会执行doAcquireSharedInterruptibly方法,我们到该方法中进一步查看:
可以看到acquire获取不到锁,会加入到AQS队列,并等待唤醒和竞争锁。
3.3 release()逻辑
首先进入release方法:
这里可以看到,首先是尝试释放锁信息,注意这里tryReleaseShared方法所属的类是semaphore中的sync,所以需要到semaphore中查看:
可以看到在tryReleaseShared中会尝试增加信号量证书。信号量证书增加成功后,我们再看下后续的逻辑:
可以看到,其会判断AQS中是否还有等待获取锁的线程,如果还有获取锁的线程,那么就唤醒等待的线程获取锁。
4、总结
semaphore底层基于AQS队列,初始化设置的信号量证书也即AQS最大的锁数量,当锁全分配出去后,后续的线程会进入AQS等待队列。semaphore(0)的使用,在于和release方法的配和。release方法被调用后会增加锁数量(信号量证书)。因此semaphore(0)可以用于一些需要初始阻塞所有线程,后续通过release唤醒线程使用的场景。
因为我比较感兴趣的是semaphore(0)的实现机制,所以这里对于AQS中线程的挂起和唤醒并没有做详细的介绍,感兴趣的小伙伴可以看我之前的AQS相关文章或者自行查阅。