【Java】AQS底层实现

本文详细介绍了Java AQS(AbstractQueuedSynchronizer)的内部实现,包括基于FIFO双向队列的同步机制,节点状态waitStatus的五种取值,节点的添加以及同步状态的获取和释放过程。AQS使用CAS操作维护同步队列,线程获取锁失败会变为节点加入队列并自旋等待,释放锁时会唤醒后续节点。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、AQS内部实现
AQS是依赖内部的同步队列实现,也就是FIFO双向队列,如果当前线程竞争锁失败,那么AQS会把当前线程以及等待状态封装成一个Node节点加入到同步队列中,同时阻塞该线程,当同步状态释放时,会把首节点唤醒,使其再次尝试获取同步状态。

AQS队列内部维护的是一个双向链表,这种结构每个数据都有两个指针,分别指向直接的的前驱节点和后继节点,当线程抢占锁失败时候,会封装成Node加入到AQS中去。
在同步队列中,一个节点表示一个线程,他保存这线程的引用ThreadId,状态(watiStatus),前驱结点(pre),后继节点(next)。

2、节点状态waitStatus
每个节点包含了线程的的等待状态,是否被阻塞,是否等待唤醒,是否被取消。变量waitStatus则表示当前Node节点的等待状态,共有5中取值,cancelled,signal,condition,propagate,0
cancelled(1):表示当前节点已取消调度。当timeout或者中断情况下,会触发变更为此状态,进入该状态后的节点不再变化
signal(-1),表示当前节点释放锁的时候,需要唤醒下一个节点。或者说后继节点在等待当前节点唤醒,后继节点入队时候,会将前驱节点更新给signal
condition(-2),当其他线程调用了condition的signal方法后,condition状态的节点会从等待队列转移到同步队列中,等待获取同步锁。
propagate(-3),共享模式下,前驱节点不仅会唤醒其后继节点,同时也可能唤醒后继的后继节点。
0 ,新节点入队时候的默认状态。

3、添加节点addWaiter

先通过快速尝试设置尾节点,如果失败,则调用enq(Node node)方法设置尾节点,设置尾节点使用的CAS自旋。假如有两个线程t1,t2,同时进入enq方法 ,t==null表示队列是首次使用,需要先初始化,另一个线程cas失败,则进入下次循环,通过cas操作将node添加到队尾。

每个节点都必须设置前置节点的 ws 状态为 SIGNAL(-1),因为每个节点在休眠前,都需要将前置节点的 ws 设置成 SIGNAL。否则自己永远无法被唤醒,所以必须要一个前置节点,而这个前置节点,实际上就是当前持有锁的节点。
由于第一个节点他是没有前置节点的,就创建一个假的。
总结下来就是:每个节点都需要设置前置节点的 ws 状态(这个状态为是为了保证数据一致性),而第一个节点是没有前置节点的,所以需要创建一个虚拟节点。

private Node addWaiter(Node mode) {
        //根据给定的模式(独占或者共享)新建Node
        Node node = new Node(Thread.currentThread(), mode);
        //快速尝试添加尾节点
        Node pred = tail;
        if (pred != null) {
            node.prev = pred;
            //CAS设置尾节点
            if (compareAndSetTail(pred, node)) {
                pred.next = node;
                return node;
            }
        }
        //多次尝试
        enq(node);
        return node;
    }

private Node enq(final Node node) {
        //多次尝试,直到成功为止
        for (;;) {
            Node t = tail;
            //tail不存在,设置为首节点
            if (t == null) {
                if (compareAndSetHead(new Node()))
                    tail = head;
            } else {
                //设置为尾节点
                node.prev = t;
                if (compareAndSetTail(t, node)) {
                    t.next = node;
                    return t;
                }
            }
        }
    }

4、同步状态的获取和释放

独占式同步状态获取
acquire(int arg): 独占式的获取锁,此方法不响应中断,在这个过程中中断,线程不会从同步队列中移除,也不会立马中断,在整个过程结束后再自我中断。

public final void acquire(int arg){
	if(!tryAcquire(arg) && acquireQueued(addWaiter(Node.EXCLUSIVE))){
		selfInterrupt();//如果这个过程中出现中断,在整个过程结束后再自我中
	}
 }

tryAcquire:去尝试获取锁,获取成功后设置锁的状态并返回true,否则返回false。该方法由自定的同步组件实现,该方法的实现必须保证线程安全的获取同步状态。
addWaiter:如果tryAcquire返回false(获取锁失败),则调用该方法将当前线程封装成节点加入到同步队列尾部,并标记为独占模式。addWaiter(Node.EXCLUSIVE)),Node.EXCLUSIV既为独占模式。
acquiredQueued:当前线程获取锁失败后,就进入了一个自旋的状态,如果在等待过程中被中断过(如timeout),就返回true(接着自我中断),否则返回false。也就是说当前线程进入同步队列后进入自旋状态,每个节点会自我观察,当条件满足时,获取到同步状态后,就可以从这个自旋过程中退出,否则一直执行下去。
如果线程在等待过程中被中断过,它是不响应的。只是获取资源后才再进行自我中断selfInterrupt(),将中断补上。因为在等待队列中自旋状态的线程是不会响应中断的,它会把中断记录下来,如果在自旋时发生过中断,就返回true。然后就会执行selfInterrupt()方法,而这个方法就是简单的中断当前线程Thread.currentThread().interrupt();其作用就是补上在自旋时没有响应的中断。
tryAcquire在NonfairSync的实现

final boolean nonfairTryAcquire(int acquires) {
	final Thread current = Thread.currentThread();
	// 获取锁同步状态
	int c = getState();
	if (c == 0) {// 0表示无锁状态
	// cas竞争锁,替换state的值改为1
	if (compareAndSetState(0, acquires)) {
		// 保存当前获得锁的线程
		setExclusiveOwnerThread(current);
		return true;
        }
	}
	// 如果是同一个线程来获得锁,则直接增加冲入次数
	else if (current == getExclusiveOwnerThread()) {
		int nextc = c + acquires;// 增加重入次数
		if (nextc < 0) // overflow
			throw new Error("Maximum lock count exceeded");
			setState(nextc);
			return true;
        }
		return false;
	}

acquiredQueued的实现
前线程进入同步队列后进入自旋状态,每个节点会自我观察,当条件满足时,获取到同步状态后,就可以从这个自旋过程中退出(返回interrupted为false),否则一直执行下去。也不是每个节点都有获取锁的资格,因为是FIFO的先进先出队列,acquireQueued方法保证了只有头部节点的后继节点才有资格去获取同步状态

shouldParkAfterFailedAcquire(Node node,Node node):判断一个线程是否阻塞

private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
    int ws = pred.waitStatus;//拿到前驱节点的状态
    if (ws == Node.SIGNAL)
        //状态为SIGNAL,如果前驱节点处于等待状态,直接返回true
        return true;
    if (ws > 0) {
        /*
         * 如果前驱节点放弃了,那就一直往前找,直到找到最近一个正常等待的状态,并排在它的后边。
         * 注意:那些放弃的结点,由于被自己“加塞”到它们前边,它们相当于形成一个无引用链,稍后就会被GC回收,这个操作实际是把队列中的cancelled节点剔除掉
         */
        do {
            node.prev = pred = pred.prev;
        } while (pred.waitStatus > 0);
        pred.next = node;
    } else {
         //如果前驱节点正常,那就把前驱的状态通过CAS的方式设置成SIGNAL
        compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
    }
    return false;
}

1.如果当前线程的前驱节点状态为SINNAL,则表明当前线程需要被阻塞,调用unpark()方法唤醒,直接返回true,当前线程阻塞
2.如果当前线程的前驱节点状态为CANCELLED(ws > 0),则表明该线程的前驱节点已经等待超时或者被中断了,则需要从CLH队列中将该前驱节点删除掉,直到回溯到前驱节点状态 <= 0 ,返回false
3.如果前驱节点非SINNAL,非CANCELLED,则通过CAS的方式将其前驱节点设置为SINNAL,返回false

整个流程中,如果前驱结点的状态不是SIGNAL,那么自己就不能安心去休息,需要去找个安心的休息点(前驱节点状态 <= 0 ),也就是只有当前驱节点为SIGNAL时这个线程才可以进入等待状态。shouldParkAfterFailedAcquire —> 检查上一个节点的状态,如果是 SIGNAL 就阻塞,否则就改成 SIGNAL。

上面如果 shouldParkAfterFailedAcquire(Node pred, Node node) 方法返回true,则调用parkAndCheckInterrupt()方法阻塞当前线程:

private final boolean parkAndCheckInterrupt() {
        //调用park()使线程进入waiting状态
          LockSupport.park(this); 
          //如果被唤醒,查看自己是不是被中断的
          return Thread.interrupted();
 }

release(int arg):独占式的释放锁

public final boolean release(int arg){
	if(tryRelease(arg)){//子类自定义实现
		Node h = head;
        	if(h != null && h.waitStatus != 0){
            	unparkSuccessor(h);//唤醒下一个节点
          	}
        	return true;
  		}
    	return false;
	}

释放锁的流程很简单,首先子类自定义的方法如果释放了同步状态,如果头节点不为空并且头节点的等待状态不为0就唤醒其后继节点。主要依赖的就是子类自定义实现的释放操作。

unparkSuccessor(Node node):唤醒后继节点获取同步状态

private void unparkSuccessor(Node node){
           //获取头节点的状态
           int ws = node.waitStatus;
           if(ws < 0){
              compareAndSetWaitStatus(node,ws,0);//通过CAS将头节点的状态设置为初始状态
           }
           Node s = node.next;//后继节点
           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;
                 }
              }
           }
           if(s != null){
              LockSupport.unpark(s.thread);//真正的唤醒操作
           }
        }

唤醒操作,通过判断后继节点是否存在,如果不存在就寻找等待时间最长的适合的节点将其唤醒唤醒操作通过LockSupport中的unpark方法唤醒底层也就是unsafe类的操作。

锁释放移除节点
head节点表示获取锁成功的节点,当头结点在释放同步状态时,会唤醒后继节点,如果后继节点获得锁成功,会把自己设置为头结点,节点变化过程如下:

这个过程涉及到2个变化

设置头结点指向下一个获得锁的节点
新的获得锁的节点,将pre指针指向null
将原来head节点的next设置为null,即断开原head节点的next引用

acquire方法流程总结

首先通过子类判断是否获取了锁,如果获取了就什么也不干。tryAcquire
如果没有获取锁、通过线程创建节点加入同步队列的队尾。addWaiter
当线程在同步队列中不断的通过自旋去获取同步状态,如果获取了锁,就把其设为同步队列中的头节点,否则在同步队列中不停的自旋等待获取同步状态 acquireQueued,shouldParkAfterFailedAcquire(Node pre,Node node),parkAndCheckInterrupt()
如果在获取同步状态的过程中被中断过最后自行调用interrupted方法进行中断操作

总结的来说:线程获取锁,如果获取了锁就 保存当前获得锁的线程,如果没获取就创造一个节点通过compareAndSetTail(CAS操作)操作的方式将创建的节点加入同步队列的尾部,在同步队列中的节点通过自旋的操作不断去获取同步状态【当然由于FIFO先进先出的特性】等待时间越长就越先被唤醒。当头节点释放同步状态的时候,首先查看是否存在后继节点,如果存在就唤醒自己的后继节点,如果不存在就获取等待时间最长的符合条件的线程。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值