AQS之独占锁源码分析

 AbstractQueuedSynchronizer是java concurrent包的基础,JUC的底层实现就是靠AQS和volatile来实现的。每个java coder都必须了解的一块内容,自己也是看了忘,忘了又看,今天打算好好总结下,留着后面复习之用。
AQS采用了模板设计模式,我们在自定义锁的时候,只需要实现其几个简单的方法即可。AQS内部采用了一个用volatile修饰的state来表示的同步状态,以及用FIFO双向队列来给等待获取锁的线程排队

AQS独占锁的大致执行流程:

  • 获取锁的过程:
      1、调用acquire尝试获取锁,获取成功就执行逻辑代码
      2、如果获取锁资源失败的就把当前线程加入到同步队列中 去,并且挂起当前线程,等待锁资源释放唤醒线程
      3、同步队列中的线程等待被唤醒重新获取锁资源,获取成功则继续执行逻辑代码,失败则继续挂起等待。
  • 释放锁的过程:
      1、当前线程调用release()方法释放锁,如果没有其他线程等待锁资源(即同步队列为空),则释放锁动作完成。
      2、若同步队列不为空时,则唤醒同步队列头结点

下面进行源码的具体分析:

获取锁的方法acquire()源码:

public final void acquire(int arg) {
        if (!tryAcquire(arg) &&
            acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            selfInterrupt();
}

这里是获取独占锁的逻辑,代码很短。
tryAcquire(arg):是获取锁资源,由用户自己实现定义的接口。如果成功则整个方法结束。
addWaiter(Node.EXCLUSIVE):以独占模式将当前线程添加到同步队列中,并且返回以改线程包装的节点Node
acquireQueued:同步队列中的线程以不间断的模式获取锁资源,也用在条件队列中调用(即Condition中)。

addWaiter(Node.EXCLUSIVE)源码分析:

static final class Node {
        //共享模式
        static final Node SHARED = new Node();
        //独占模式
        static final Node EXCLUSIVE = null;

        //线程对应的节点已被取消
        static final int CANCELLED =  1;
        //暗示要通知后继节点
        static final int SIGNAL    = -1;
        //等待状态,节点在等待队列中,节点中的线程在等待Condition,当其他线程对condition调用singnal()方法时,当前线程的节点从等待队列移到同步队列中,加入到同步状态的获取
        static final int CONDITION = -2;
        //共享状态,无条件将行为往下传播
        static final int PROPAGATE = -3;

        //节点状态
        volatile int waitStatus;

        //当前节点的前驱节点
        volatile Node prev;

        //当前节点的后继节点
        volatile Node next;

        //当前节点对应的线程
        volatile Thread thread;

        //链接在等待队列中的下一个等待节点
        Node nextWaiter;
}
private Node addWaiter(Node mode) {
        //以当前线程创建一个节点(独占锁是以独占模式创建节点,并且waitStatus为0)
        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;
            //如果当前尾节点为空,则初始化头节点和尾节点
            if (t == null) {
                if (compareAndSetHead(new Node()))
                    tail = head;
            } else {
                node.prev = t;
                //cas操作把当前节点加入到尾节点
                if (compareAndSetTail(t, node)) {
                    t.next = node;
                    return t;
                }
            }
      }
}

之前一直没有搞懂这里为什么要new一个空的头结点,可以这样来理解:初始化队列的触发条件是当前已有线程占有了锁资源,当前空的头节点可以被认为是此时已经占有锁资源的节点。
enq(node)就是死循环来把节点加入到尾节点,因为存在并发获取锁资源的线程也在这个时候加入到同步队列中,cas操作不一定会成功,所以这里就采用死循环来不停地进行重试插入到尾节点。

下面是刚刚加入到同步队列中的节点开始不间断地获取锁资源,并且来看线程是否需要挂起或中断

不间断地获取锁资源
final boolean acquireQueued(final Node node, int arg) {
        //获取锁资源的标记置为失败
        boolean failed = true;
        try {
            //线程中断的标记置为false
            boolean interrupted = false;
            for (;;) {
                //获取当前节点的前驱
                final Node p = node.predecessor();
                //如果当前节点的前驱节点为头节点,则去获取锁资源
                if (p == head && tryAcquire(arg)) {
                    //获取锁资源成功则将当前节点设为头结点,这里不需要用cas操作,因为这里已成功获取到了锁资源后不需要靠率并发问题
                    setHead(node);
                    //将当前节点的前驱节点的next置为null,帮助GC
                    p.next = null; // help GC
                    //获取锁资源标记置为成功
                    failed = false;
                    //返回终端标记
                    return interrupted;
                }
                //检查在获取锁资源后是否需要挂起并挂起检查中断
                if (shouldParkAfterFailedAcquire(p, node) &&
                    parkAndCheckInterrupt())
                    interrupted = true;
            }
        } finally {
            //最后获取锁资源失败的时候取消获取,这一般使我们的tryAcquire的实现出现了异常
            if (failed)
                //这里我暂且没有看懂等我后续梳理
                cancelAcquire(node);
        }
}

//判断当前节点是否需要挂起
private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
         //当前节点的前驱节点的状态
        int ws = pred.waitStatus;
        //如果当前节点的前驱节点状态是-1,那么当前节点可以安心的挂起,等待前驱节点释放锁资源的时候进行唤醒
        if (ws == Node.SIGNAL)
            return true;
        //当前节点的前驱节点为取消状态时,那么往前追溯到一个不是取消的节点,将它作为当前节点node的前驱节点,其实这个时候也删除了同步队列中状态为取消的节点
        if (ws > 0) {
            do {
                node.prev = pred = pred.prev;
            } while (pred.waitStatus > 0);
            //注,由于头结点head是通过new Node()创建,它的waitStatus为0,因此这里不会出现空指针问题,也就是说最多就是找到头节点上面的循环就退出了
            pred.next = node;
        } else {
            //这里状态只能是0或-3(因为状态为-2的节点在等待队列中),那么久把当前节点的前驱节点置为-1,让外层for循环的下一次循环再进入到这个方法,重新进行判断是否需要挂起
            compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
        }
        return false;
}

//挂起当前线程,并且返回当前线程的中断标志位
private final boolean parkAndCheckInterrupt() {
        //挂起当前线程
        LockSupport.park(this);
        //被唤醒之后,返回当前线程的中断标志位
        return Thread.interrupted();
}

这个不间断获取同步状态的代码流程是:

  • 声明两个标记变量:failed(获取同步状态的标记)和interrupted(当前线程是否中断标记)
  • 死循环进行获取同步状态,其实这里只有前驱节点是头节点的节点才会去尝试获取同步状态,其他节点进行了挂起,不去尝试获取同步状态,这样节约资源,不去浪费资源。获取到了同步状态时,把当前节点设置为头节点(头节点都是获取到了同步状态的节点),并且把前驱节点移出同步队列,因为当前节点能获取到同步状态是因为前驱节点释放了同步状态,并且唤醒了当前节点,因此当前节点才能获取到同步状态。
  • 判断当前节点是否需要挂起,当前节点的前驱节点状态是-1的时候就会被挂起,安心等前驱节点释放同步状态的时候唤醒。并且把状态为1(线程被取消或中断的节点)的节点移出同步队列。如果当前节点的前驱节点为0或-3就把前驱节点的状态设置为-1,最终线程挂起,返回中断标致。

下面来看下cancelAcquire:

//这里我暂且没有搞清楚,后续来继续整理下
private void cancelAcquire(Node node) {
        if (node == null)
            return;

        node.thread = null;

        Node pred = node.prev;
        while (pred.waitStatus > 0)
            node.prev = pred = pred.prev;

        Node predNext = pred.next;

        node.waitStatus = Node.CANCELLED;

        if (node == tail && compareAndSetTail(node, pred)) {
            compareAndSetNext(pred, predNext, null);
        } else {
            int ws;
            if (pred != head &&
                ((ws = pred.waitStatus) == Node.SIGNAL ||
                 (ws <= 0 && compareAndSetWaitStatus(pred, ws, Node.SIGNAL))) &&
                pred.thread != null) {
                Node next = node.next;
                if (next != null && next.waitStatus <= 0)
                    compareAndSetNext(pred, predNext, next);
            } else {
                unparkSuccessor(node);
            }
            node.next = node; // help GC
        }
    }

最后来看下释放同步状态的源码:

public final boolean release(int arg) {
        //释放同步状态
        if (tryRelease(arg)) {
            Node h = head;
            //如果头节点不为空并且头结点状态不为0时,唤醒头节点
            if (h != null && h.waitStatus != 0)
                unparkSuccessor(h);
            return true;
        }
        return false;
}

//唤醒节点,这里传过来的参数是头节点
private void unparkSuccessor(Node node) {
        int ws = node.waitStatus;
        //把当前节点的状态设置为0,表示唤醒动作已开始,提高并发的性能
        if (ws < 0)
            compareAndSetWaitStatus(node, ws, 0);
        //这里的s是释放同步状态节点的后继节点
        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);
    }

在头节点释放同步状态的时候,唤醒其后继节点。

你可能感兴趣的:(AQS之独占锁源码分析)