Java AQS信号量Semaphore的使用
作者：飞奔的小付 发布时间：2021-06-11 17:10:44
标签：Java,信号量,Semaphore
一.什么是Semaphore
Semaphore，俗称信号量，它是操作系统中PV操作的原语在java的实现，它也是基于AbstractQueuedSynchronizer实现的。
Semaphore的功能非常强大，大小为1的信号量就类似于互斥锁，通过同时只能有一个线程获取信号量实现。大小为n（n>0）的信号量可以实现限流的功能，它可以实现只能有n个线程同时获取信号量。
PV操作是操作系统一种实现进程互斥与同步的有效方法。PV操作与信号量（S）的处理相关，P表示通过的意思，V表示释放的意思。用PV操作来管理共享资源时，首先要确保PV操作自身执行的正确性。
P操作的主要动作是：
①S减1；
②若S减1后仍大于或等于0，则进程继续执行；
③若S减1后小于0，则该进程被阻塞后放入等待该信号量的等待队列中，然后转进程调度。
V操作的主要动作是：
①S加1；
②若相加后结果大于0，则进程继续执行；
③若相加后结果小于或等于0，则从该信号的等待队列中释放一个等待进程，然后再返回原进程继续执行或转进程调度。
二.Semaphore的使用
构造器
public Semaphore(int permits) {
sync = new NonfairSync(permits);
}
public Semaphore(int permits, boolean fair) {
sync = fair ? new FairSync(permits) : new NonfairSync(permits);
}
permits 表示许可证的数量（资源数）
fair 表示公平性，如果这个设为 true 的话，下次执行的线程会是等待最久的线程
常用方法
public void acquire() throws InterruptedException
public boolean tryAcquire()
public void release()
public int availablePermits()
public final int getQueueLength()
public final boolean hasQueuedThreads()
protected void reducePermits(int reduction)
acquire() 表示阻塞并获取许可
tryAcquire() 方法在没有许可的情况下会立即返回 false，要获取许可的线程不会阻塞
release() 表示释放许可
int availablePermits()：返回此信号量中当前可用的许可证数。
int getQueueLength()：返回正在等待获取许可证的线程数。
boolean hasQueuedThreads()：是否有线程正在等待获取许可证。
void reducePermit（int reduction）：减少 reduction 个许可证
Collection getQueuedThreads()：返回所有等待获取许可证的线程集合
应用场景
可以用于做流量控制，特别是公用资源有限的应用场景
限流
/**
* 实现一个同时只能处理5个请求的限流器
*/
private static Semaphore semaphore = new Semaphore(5);
/**
* 定义一个线程池
*/
private static ThreadPoolExecutor executor = new ThreadPoolExecutor(10, 50, 60, TimeUnit.SECONDS, new LinkedBlockingDeque<>(200));
public static void exec() {
try {
semaphore.acquire(1);
// 模拟真实方法执行
System.out.println("执行exec方法" );
Thread.sleep(2000);
} catch (Exception e) {
e.printStackTrace();
} finally {
semaphore.release(1);
}
}
public static void main(String[] args) throws InterruptedException {
{
for (; ; ) {
Thread.sleep(100);
// 模拟请求以10个/s的速度
executor.execute(() -> exec());
}
}
}
三.Semaphore源码分析
主要关注 Semaphore的加锁解锁（共享锁）逻辑实现，线程竞争锁失败入队阻塞逻辑和获取锁的线程释放锁唤醒阻塞线程竞争锁的逻辑实现
public void acquire() throws InterruptedException {
sync.acquireSharedInterruptibly(1);
}
public final void acquireSharedInterruptibly(int arg)
throws InterruptedException {
if (Thread.interrupted())
throw new InterruptedException();
//尝试获取共享锁，大于等于0则直接获取锁成功，小于0时则共享锁阻塞
if (tryAcquireShared(arg) < 0)
doAcquireSharedInterruptibly(arg);
}
tryAcquireShared的实现
final int nonfairTryAcquireShared(int acquires) {
for (;;) {
int available = getState();
int remaining = available - acquires;
// 当减一之后的值小于0 或者
// compareAndSetState成功，把state变为remaining，即将状态减一
if (remaining < 0 ||
compareAndSetState(available, remaining))
return remaining;
}
}
入队阻塞
private void doAcquireSharedInterruptibly(int arg)
throws InterruptedException {
//入队，创建节点 使用共享模式
final Node node = addWaiter(Node.SHARED);
boolean failed = true;
try {
for (;;) {
//获取当前节点的前躯节点
final Node p = node.predecessor();
//如果节点为head节点
if (p == head) {
//阻塞动作比较重，通常会再尝试获取资源，没有获取到返回负数
int r = tryAcquireShared(arg);
if (r >= 0) {
setHeadAndPropagate(node, r);
p.next = null; // help GC
failed = false;
return;
}
}
//判断是否可以阻塞
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
throw new InterruptedException();
}
} finally {
if (failed)
cancelAcquire(node);
}
}
入队操作
private Node addWaiter(Node mode) {
//构建节点，模式是共享模式
Node node = new Node(Thread.currentThread(), mode);
Node pred = tail;
if (pred != null) {
//设置前一节点为tail
node.prev = pred;
//设置当前节点为尾节点
if (compareAndSetTail(pred, node)) {
// 前一节点的next为当前节点
pred.next = node;
return node;
}
}
//创建队列
enq(node);
return node;
}
创建队列，经典的for循环创建双向链表
private Node enq(final Node node) {
for (;;) {
Node t = tail;
if (t == null) {
//节点为空 则new一个节点 设置头节点
if (compareAndSetHead(new Node()))
//把这个节点的头节点赋值给尾节点
tail = head;
} else {
// 如果尾节点存在 就将该节点的前节点指向tail
node.prev = t;
//设置当前节点为tail
if (compareAndSetTail(t, node)) {
//前一个节点的next指向当前节点，入队操作就完成了
t.next = node;
return t;
}
}
}
}
设置waitStatus状态及获取waitStatus状态，waitStatus为-1时可以唤醒后续节点
private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
int ws = pred.waitStatus;
if (ws == Node.SIGNAL)
/*
* 状态是-1 就可以唤醒后续节点
*
*/
return true;
if (ws > 0) {
/*
* 前置任务已取消，删掉节点
*
*/
do {
node.prev = pred = pred.prev;
} while (pred.waitStatus > 0);
pred.next = node;
} else {
/*
* cas设置waitstatus状态，设置为-1
*
*/
compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
}
return false;
}
阻塞 调用LockSupport.park
private final boolean parkAndCheckInterrupt() {
LockSupport.park(this);
return Thread.interrupted();
}
释放锁的逻辑
public void release() {
sync.releaseShared(1);
}
public final boolean releaseShared(int arg) {
//cas成功则进行释放共享锁
if (tryReleaseShared(arg)) {
doReleaseShared();
return true;
}
return false;
}
state状态+1操作，cas成功，返回true
protected final boolean tryReleaseShared(int releases) {
for (;;) {
int current = getState();
int next = current + releases;
if (next < current) // overflow
throw new Error("Maximum permit count exceeded");
if (compareAndSetState(current, next))
return true;
}
}
private void doReleaseShared() {
for (;;) {
Node h = head;
//头节点不为空并且不是尾节点
if (h != null && h != tail) {
int ws = h.waitStatus;
//waitstatus为-1
if (ws == Node.SIGNAL) {
//将SIGNAL置为0
if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))
continue;
//唤醒
unparkSuccessor(h);
}
else if (ws == 0 &&
!compareAndSetWaitStatus(h, 0, Node.PROPAGATE))
continue;
}
if (h == head)
break;
}
}
唤醒操作
private void unparkSuccessor(Node node) {
int ws = node.waitStatus;
//ws小于0就将其设置为0
if (ws < 0)
compareAndSetWaitStatus(node, ws, 0);
Node s = node.next;
//当前节点的下一个节点为空或者ws大于0
if (s == null || s.waitStatus > 0) {
s = null;
for (Node t = tail; t != null && t != node; t = t.prev)
if (t.waitStatus <= 0)
s = t;
}
//s不为空 则进行唤醒
if (s != null)
LockSupport.unpark(s.thread);
}
唤醒下一个线程之后，要把上一个节点移除队列
private void doAcquireSharedInterruptibly(int arg)
throws InterruptedException {
final Node node = addWaiter(Node.SHARED);
boolean failed = true;
try {
for (;;) {
final Node p = node.predecessor();
//下一个线程进来，如果前置节点是头节点，则将前置节点出队
if (p == head) {
int r = tryAcquireShared(arg);
//cas获取资源成功
if (r >= 0) {
//出队操作
setHeadAndPropagate(node, r);
//将p.next移除
p.next = null;
failed = false;
return;
}
}
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
throw new InterruptedException();
}
} finally {
if (failed)
cancelAcquire(node);
}
}
出队操作
private void setHeadAndPropagate(Node node, int propagate) {
//设置当前节点为head节点，前一节点的head属性被删除
Node h = head;
setHead(node);
//如果是传播属性
if (propagate > 0 || h == null || h.waitStatus < 0 ||
(h = head) == null || h.waitStatus < 0) {
//并且是共享模式，可以持续唤醒下一个，
//只要资源数充足 就可以一直往下唤醒，提高并发量
Node s = node.next;
if (s == null || s.isShared())
doReleaseShared();
}
}
private void setHead(Node node) {
head = node;
node.thread = null;
node.prev = null;
}
至此，线程的阻塞唤醒核心逻辑就这么多，共享锁与独占锁的区别是可以唤醒后续的线程，如果资源数充足的话，可以一直往下唤醒，提高了并发量。
来源：https://blog.csdn.net/feibendexiaoma/article/details/127716489
0
投稿猜你喜欢
场景File与FileStream的区别举例：将读取文件比作是从A桶往B桶运水。使用File就是整个用桶倒进去，使用FileStream就是- 在以往的 Tomcat 项目中，一直习惯用 Ant 打包，使用 build.xml 配置，通过 ant -buildfile 的方式在机器上
- springboot：整合sa-token一、简介Sa-Token 是一个轻量级 Java 权限认证框架，主要解决：登录认证、权限认证、Se
- 以下是测试代码：新建一个classlibrary，包含两个类class1和class2，这两个类中分别有一个方法，都是返回一个字符串，代码如
- 本文实例讲述了java实现的n*n矩阵求值及求逆矩阵算法。分享给大家供大家参考，具体如下：先来看看运行结果：java版的写出来了，用的跟c语
- 本文源码：GitHub·点这里 || GitEE·点这里一、Ehcache缓存简介1、基础简介EhCache是一个纯Java的进程内缓存框架
- JAVA反射机制JAVA反射机制是在运行状态中，对于任意一个类，都能够知道这个类的所有属性和方法；对于任意一个对象，都能够调用它的任意一个方
- 前言直接使用项目或直接复制libs中的so库到项目中即可（当前只构建了armeabi），需要其他ABI可检下项目另外使用CMake构建即可。
- 1.新建springBoot项目在前面有两种方式2.加入thymeleaf模板引擎SpringBoot推荐使用thymeleaf模板引擎语法
- 前言当系统的并发比较高的时候，日志的处理输出也是一种性能的开销负担，所以，选择一个中间件来处理消费日志必不可少！下面是spring boot
- 幂等概述幂等性原本是数学上的概念，即使公式：f(x)=f(f(x)) 能够成立的数学性质。用在编程领域，则意为对同一个系统，使用同样的条件，
- 修改readme.txt文件如下：Git is a distributed version control system.Git is fr
- 如果一个对象不为空null时，把它赋给另外一个对象：像下面这个样子，需要把str的值赋给result，前提条件是在不为空null的前提之下：
- 如图所示为程序效果动画图地图滚动的原理在本人之前博客的文章中介绍过人物在屏幕中的移动方式，因为之前拼的游戏地图是完全填充整个手机屏幕的，所以
- 判断是否为飞行模式: boolean isAirplaneMode = Settings.System.getInt(mConte
- WPF 实现面包屑控件框架使用.NET4 至 .NET6Visual Studio 2022创建 BreadCrumbBar.xa
- 前言前面我们已经分析Dubbo SPI相关的源码，看过的小伙伴相信已经知晓整个加载过程，我们也留下两个问题，今天我们先来处理下其中关于注解A
- 一个Java程序的执行要经过编译和执行（解释）这两个步骤，同时Java又是面向对象的编程语言。当子类和父类存在同一个方法，子类重写了父类的方
- 引言 批量下载思路，首先前端发送下载请求，携带必要参数，
- 在AS中进行NDK开发之前，我们先来简单的介绍几个大家都容易搞懵的概念： &
