天天看点

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

一、AQS 框架简介

AQS即

AbstractQueuedSynchronizer

是JUC(

java.util .concurrent

工具包的简称)里是一个用于构建锁和同步器的工具框架。

Ok,它定性为一种框架,且是工具框架。

那就是说,通过这个工具框架我们可以更方便的更便捷的去实现我们想要的功能,这里当然指的是多线程相关的功能。可以想想其他的框架Spring,Mybatis等是什么作用,类比一下。

它能降低构建锁和同步器的工作量,还可以避免处理多个位置上发生的竞争问题

。在基于AQS构建的同步器中,只可能在一个时刻发生阻塞,从而降低上下文切换的开销,并提高吞吐量。

同时,定性了是工具框架,那就意味着,如何使用这个工具是重点,在会使用的基础上弄清原理。

AQS诞生于Jdk1.5,在当时低效且功能单一的synchronized的年代,某种意义上讲,她拯救了Java。

其实Doug Lea也意识到问题的复杂性,不可能出一个超级工具来解决所有问题,所以他把AQS设计为一个abstract类,并提供一系列子类去解决不同场景的问题,例如

ReentrantLock

Semaphore

等;当我们发现这些子类也不能满足我们加锁需求时,我们可以定义自己的子类,通过重写两三个方法,就能实现自己的同步器,寥寥几行代码,实现强大的功能。再次说明了AQS就是个工具框架!

所以这个定性很重要!

一、AQS 核心思想

AQS核心思想是:

如果被请求的共享资源空闲,则将当前请求资源的线程设置为有效的工作线程,并且将共享资源设置为锁定状态

。如果被请求的共享资源被占用,那么就需要一套线程阻塞等待以及被唤醒时锁分配的机制,这个机制AQS是用

CLH队列锁

实现的,即将暂时获取不到锁的线程加入到队列中。

二、AQS内部的数据结构与原理

AQS支持

独占锁(exclusive)

共享锁(share)

两种模式

  • 独占锁:只能被一个线程获取到(Reentrantlock)
  • 共享锁:可以被多个线程同时获取(CountDownLatch,ReadWriteLock)

无论是独占锁还是共享锁,本质上都是对AQS内部的一个变量state的获取。state是一个原子的int变量,用来表示锁状态、资源数等。如下:

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

state

属性来表示资源的状态(

分独占模式

共享模式

),子类需要定义如何维护这个状态,控制如何获取锁和释放锁。

  • getState - 获取 state 状态
  • setState - 设置 state 状态
  • compareAndSetState - cas 机制设置 state 状态

这个

sync queue同步队列

只是其中一个对列,AQS内部实现了两个队列,一个同步队列,一个条件队列:

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
  • 同步队列的作用是:当线程获取资源失败之后,就进入同步队列的尾部保持自旋等待,不断判断自己是否是链表的头节点,如果是头节点,就不断参试获取资源,获取成功后则退出同步队列。
  • 条件队列的作用是:为Lock实现的一个基础同步器,并且一个线程可能会有多个条件队列,只有在使用了Condition才会存在条件队列。

同步队列和条件队列都是由一个个Node组成的。AQS内部有一个静态内部类Node:

static final class Node {
        static final Node EXCLUSIVE = null;

        //当前节点由于超时或中断被取消
        static final int CANCELLED =  1;
     
        //表示当前节点的前节点被阻塞
        static final int SIGNAL    = -1;
        
        //当前节点在等待condition
        static final int CONDITION = -2;
      
        //状态需要向后传播
        static final int PROPAGATE = -3;
        
        volatile int waitStatus;
        
        volatile Node prev;
        volatile Node next;
        volatile Thread thread;

        Node nextWaiter;

        final boolean isShared() {
            return nextWaiter == SHARED;
        }

        final Node predecessor() throws NullPointerException {
            Node p = prev;
            if (p == null)
                throw new NullPointerException();
            else
                return p;
        }

        Node() {    // Used to establish initial head or SHARED marker
        }

        Node(Thread thread, Node mode) {     // Used by addWaiter
            this.nextWaiter = mode;
            this.thread = thread;
        }

        Node(Thread thread, int waitStatus) { // Used by Condition
            this.waitStatus = waitStatus;
            this.thread = thread;
        }
    }
           

三、子类要实现的方法

前面说了AOS是用于构建锁和同步器的工具框架,当我们发现它的子类(例如

ReentrantLock

Semaphore

等)不能满足我们加锁需求时,我们可以定义自己的子类实现锁的功能,方式就是通过重写AOS的方法,具体有哪些主要的方法呢?如下:

  • tryAcquire:独占方式。arg为

    获取锁

    的次数,尝试获取资源,成功则返回True,失败则返回False。
  • tryRelease:独占方式。arg为

    释放锁

    的次数,尝试释放资源,成功则返回True,失败则返回False。
  • tryAcquireShared:共享方式。arg为获取锁的次数,尝试获取资源。负数表示失败;0表示成功,但没有剩余可用资源;正数表示成功,且有剩余资源。
  • tryReleaseShared:共享方式。arg为释放锁的次数,尝试释放资源,如果释放后允许唤醒后续等待结点返回True,否则返回False。
  • isHeldExclusively:该线程是否正在独占资源。只有用到

    Condition

    才需要去实现它。

举个例子:

使用AQS实现一个自定义的

同步器

来实现不可重入锁。

public class MyNonReentrantLock implements Lock {
    
    // 独占锁  同步器类
    class MySync extends AbstractQueuedSynchronizer{

        @Override
        protected boolean tryAcquire(int arg) {
            // 确保原子性
            if (compareAndSetState(0, 1)) {
                // 加上了锁,并设置 owner 为当前线程
                setExclusiveOwnerThread(Thread.currentThread());
                return true;
            }
            return false;
        }

        @Override
        protected boolean tryRelease(int arg) {
            // 这里不需要确定原子性, 因为是持锁者自己进行释放
            // 在setState(0)上面设置Owner为null, 防止指令重排序带来的问题
            setExclusiveOwnerThread(null);
            setState(0); // state是volatile修饰的, 在setState(0)前面的属性修改, 对于其他线程也是可见的, 具体见volatile原理(写屏障)
            return true;
        }

        @Override
        protected boolean isHeldExclusively() {
            return getState() == 1;
        }

        public Condition newCondition() {
            return new ConditionObject();
        }
    }

    private MySync sync = new MySync();

    @Override
    public void lock() {
        sync.acquire(1);
    }

    @Override
    public void lockInterruptibly() throws InterruptedException {
        sync.acquireInterruptibly(1);
    }

    @Override
    public boolean tryLock() {
        return sync.tryAcquire(1);
    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        return sync.tryAcquireNanos(1, unit.toNanos(time));
    }

    @Override
    public void unlock() {
        sync.release(1);
    }

    @Override
    public Condition newCondition() {
        return sync.newCondition();
    }
}
           
public class MyNonReentrantLockTest {
    public static void main(String[] args) {
        MyNonReentrantLock lock = new MyNonReentrantLock();
        new Thread(() -> {
            lock.lock();
            ConsoleUtil.print(Thread.currentThread() + "locking...");
            try {
                Thread.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                ConsoleUtil.print(Thread.currentThread() + "unlocking...");
                lock.unlock();
            }
        }, "t1").start();

        new Thread(() -> {
            lock.lock();
            try {
                ConsoleUtil.print(Thread.currentThread() + "locking...");
            } finally {
                ConsoleUtil.print(Thread.currentThread() + "unlocking...");
                lock.unlock();
            }
        }, "t2").start();
    }
}
           
2021-08-03 20:42:04 Thread[t1,5,main]locking...
2021-08-03 20:42:04 Thread[t1,5,main]unlocking...
2021-08-03 20:42:04 Thread[t2,5,main]locking...
2021-08-03 20:42:04 Thread[t2,5,main]unlocking...
           

可以看到有了AQS框架,自己写一个锁就变得简单了,框架已经提供了很多现成的方法。

四、ReentrantLock 原理

ReentrantLock

提供了两个同步器,实现公平锁和非公平锁,默认是非公平锁!

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

非公平锁实现原理

先从构造器开始看,默认为非公平锁实现:

public ReentrantLock() {
 	sync = new NonfairSync();
}
           

NonfairSync extends Sync extends AbstractQueuedSynchronizer

【没有竞争时】

  • Thread-0成为锁的持有者
    【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

    【第一个竞争出现时】

    查看源码的NonfairSync的lock方法

    【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
// 非公平锁的lock
final void lock() {
    if (compareAndSetState(0, 1))
        setExclusiveOwnerThread(Thread.currentThread());
    else
        acquire(1);
}

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

Thread-1 执行了

1.lock方法中CAS 尝试将 state 由 0 改为 1,结果失败 (因为此时CAS操作, 已经state已经为1了)

2.lock方法中进一步调用acquire方法,进入 tryAcquire 逻辑,这里我们认为这时 state 已经是1,结果仍然失败

3.接下来进入 acquire方法的addWaiter 逻辑,构造 Node 队列 (双向链表实现)

①下图中黄色三角表示该 Node 的waitStatus状态,其中 0 为默认正常状态

② Node 的创建是懒惰的

③ 其中第一个 Node 称为 Dummy(哑元)或哨兵,用来占位,并不关联线程

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

当前线程进入 acquire方法的 acquireQueued 逻辑

① acquireQueued 会在一个死循环中不断尝试获得锁,失败后进入 park 阻塞

② 如果自己是紧邻着 head(排第二位),那么再次 tryAcquire 尝试获取锁,我们这里设置这时 state 仍为 1,失败

③ 进入 shouldParkAfterFailedAcquire 逻辑,将前驱 node,即 head 的 waitStatus 改为 -1,这次返回 false

④ shouldParkAfterFailedAcquire 执行完毕回到 acquireQueued ,再次 tryAcquire 尝试获取锁,当然这时 state 仍为 1,失败

⑤ 当再次进入 shouldParkAfterFailedAcquire 时,这时因为其前驱 node 的 waitStatus 已经是 -1,这次返回 true

⑥ 进入parkAndCheckInterrupt, Thread-1 park(灰色表示已经阻塞)

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

再次有多个线程经历上述过程竞争失败,变成这个样子

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

Thread-0 调用unlock方法里的release方法释放锁,进入tryRelease(使用ctrl+alt+b查看tryRelease方法的具体ReentrantLock实现)流程,如果成功,设置 exclusiveOwnerThread 为 null,state = 0

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

unlock方法里的release方法方法中,如果当前队列不为 null,并且 head 的 waitStatus = -1,进入 unparkSuccessor 流程: unparkSuccessor中会找到队列中离 head 最近的一个 Node(没取消的),unpark 唤醒Thread-1 恢复其运行,本例中即为 Thread-1 回到 Thread-1 阻塞的位置继续执行, 会继续执行 acquireQueued 流程

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

如果加锁成功(没有竞争),会设置 (acquireQueued 方法中)exclusiveOwnerThread 为 Thread-1,state = 1;head 指向刚刚 Thread-1 所在的 Node,该 Node 清空 Thread;原本的 head 因为从链表断开,而可被垃圾回收

如果这时候有其它线程来竞争(非公平的体现),例如这时有 Thread-4 来了

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

如果不巧又被 Thread-4 占了先

Thread-4 被设置为 exclusiveOwnerThread,state = 1

Thread-1 再次进入 acquireQueued 流程,获取锁失败,重新进入 park 阻塞

加锁源码

// Sync 继承自 AQS
static final class NonfairSync extends Sync {
    private static final long serialVersionUID = 7316153563782823691L;

     // 加锁实现
    final void lock() {
        // 首先用 cas 尝试(仅尝试一次)将 state 从 0 改为 1, 如果成功表示获得了独占锁
        if (compareAndSetState(0, 1))
            setExclusiveOwnerThread(Thread.currentThread());
        else
            // 如果尝试失败,进入 ㈠
            acquire(1);
    }

    // ㈠ AQS 继承过来的方法, 方便阅读, 放在此处
    public final void acquire(int arg) {
        // ㈡ tryAcquire
        if (
                !tryAcquire(arg) &&
            	// 当 tryAcquire 返回为 false 时, 先调用 addWaiter ㈣, 接着 acquireQueued ㈤
                 acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
        ) {
            selfInterrupt();
        }
    }

    // ㈡ 进入 ㈢
    protected final boolean tryAcquire(int acquires) {
        return nonfairTryAcquire(acquires);
    }

    // ㈢ Sync 继承过来的方法, 方便阅读, 放在此处
    final boolean nonfairTryAcquire(int acquires) {
        final Thread current = Thread.currentThread();
        int c = getState();
        // 如果还没有获得锁
        if (c == 0) {
            // 尝试用 cas 获得, 这里体现了非公平性: 不去检查 AQS 队列
            if (compareAndSetState(0, acquires)) {
                setExclusiveOwnerThread(current);
                return true;
            }
        }
        // 如果已经获得了锁, 线程还是当前线程, 表示发生了锁重入
        else if (current == getExclusiveOwnerThread()) {
            // state++
            int nextc = c + acquires;
            if (nextc < 0) // overflow
                throw new Error("Maximum lock count exceeded");
            setState(nextc);
            return true;
        }
        // 获取失败, 回到调用处
        return false;
    }

    // ㈣ AQS 继承过来的方法, 方便阅读, 放在此处
    private Node addWaiter(Node mode) {
// 将当前线程关联到一个 Node 对象上, 模式为独占模式,新建的Node的waitstatus默认为0,因为waitstatus是成员变量,默认被初始化为0
        Node node = new Node(Thread.currentThread(), mode);
        // 如果 tail 不为 null, cas 尝试将 Node 对象加入 AQS 队列尾部
        Node pred = tail;
        if (pred != null) {
            node.prev = pred;
            if (compareAndSetTail(pred, node)) {
                // 双向链表
                pred.next = node;
                return node;
            }
        }
        //如果tail为null,尝试将 Node 加入 AQS, 进入 ㈥
        enq(node);
        return node;
    }

    // ㈥ AQS 继承过来的方法, 方便阅读, 放在此处
    private Node enq(final Node node) {
        for (;;) {
            Node t = tail;
            if (t == null) {
                // 还没有, 设置 head 为哨兵节点(不对应线程,状态为 0)
                if (compareAndSetHead(new Node())) {
                    tail = head;
                }
            } else {
                // cas 尝试将 Node 对象加入 AQS 队列尾部
                node.prev = t;
                if (compareAndSetTail(t, node)) {
                    t.next = node;
                    return t;
                }
            }
        }
    }

    // ㈤ AQS 继承过来的方法, 方便阅读, 放在此处
    final boolean acquireQueued(final Node node, int arg) {
        boolean failed = true;
        try {
            boolean interrupted = false;
            for (;;) {
                final Node p = node.predecessor();
                // 上一个节点是 head, 表示轮到自己(当前线程对应的 node)了, 尝试获取
                if (p == head && tryAcquire(arg)) {
                    // 获取成功, 设置自己(当前线程对应的 node)为 head
                    setHead(node);
                    // 上一个节点 help GC
                    p.next = null;
                    failed = false;
                    // 返回中断标记 false
                    return interrupted;
                }
                if (
                    // 判断是否应当 park, 进入 ㈦
                    shouldParkAfterFailedAcquire(p, node) &&
                    // park 等待, 此时 Node 的状态被置为 Node.SIGNAL ㈧
                    parkAndCheckInterrupt()
                ) {
                    interrupted = true;
                }
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }

    // ㈦ AQS 继承过来的方法, 方便阅读, 放在此处
    private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
        // 获取上一个节点的状态
        int ws = pred.waitStatus;
        if (ws == Node.SIGNAL) {
            // 上一个节点都在阻塞, 那么自己也阻塞好了
            return true;
        }
        // > 0 表示取消状态
        if (ws > 0) {
            // 上一个节点取消, 那么重构删除前面所有取消的节点, 返回到外层循环重试
            do {
                node.prev = pred = pred.prev;
            } while (pred.waitStatus > 0);
            pred.next = node;
        } else {
            // 这次还没有阻塞
            // 但下次如果重试不成功, 则需要阻塞,这时需要设置上一个节点状态为 Node.SIGNAL
            compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
        }
        return false;
    }

    // ㈧ 阻塞当前线程
    private final boolean parkAndCheckInterrupt() {
        LockSupport.park(this);
        return Thread.interrupted();
    }
}
           

解锁源码

// Sync 继承自 AQS
static final class NonfairSync extends Sync {
    // 解锁实现
    public void unlock() {
        sync.release(1);
    }

    // AQS 继承过来的方法, 方便阅读, 放在此处
    public final boolean release(int arg) {
        // 尝试释放锁, 进入 ㈠
        if (tryRelease(arg)) {
            // 队列头节点 unpark
            Node h = head;
            if (
                // 队列不为 null
                h != null &&
                // waitStatus == Node.SIGNAL 才需要 unpark
                h.waitStatus != 0
            ) {
                // unpark AQS 中等待的线程, 进入 ㈡
                unparkSuccessor(h);
            }
            return true;
        }
        return false;
    }

    // ㈠ Sync 继承过来的方法, 方便阅读, 放在此处
    protected final boolean tryRelease(int releases) {
        // state--
        int c = getState() - releases;
        if (Thread.currentThread() != getExclusiveOwnerThread())
            throw new IllegalMonitorStateException();
        boolean free = false;
        // 支持锁重入, 只有 state 减为 0, 才释放成功
        if (c == 0) {
            free = true;
            setExclusiveOwnerThread(null);
        }
        setState(c);
        return free;
    }

    // ㈡ AQS 继承过来的方法, 方便阅读, 放在此处
    private void unparkSuccessor(Node node) {
        // 如果状态为 Node.SIGNAL 尝试重置状态为 0, 如果线程获取到了锁那么后来头结点会被抛弃掉
        // 不成功也可以
        int ws = node.waitStatus;
        if (ws < 0) {
            compareAndSetWaitStatus(node, ws, 0);
        }
        // 找到需要 unpark 的节点, 但本节点从 AQS 队列中脱离, 是由唤醒节点完成的
        Node s = node.next;
        // 不考虑已取消的节点, 从 AQS 队列从后至前找到队列最前面需要 unpark 的节点
        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队列中, 自己(线程) 有没有前驱节点(线程), (该节点不是占位的哨兵节点); 如果有就不去竞争锁. 如果没有, 才会去CAS操作

if (!hasQueuedPredecessors() &&
	compareAndSetState(0, acquires)) {
	setExclusiveOwnerThread(current);
	return true;
}
           
static final class FairSync extends Sync {
    private static final long serialVersionUID = -3000897897090466540L;
    final void lock() {
        acquire(1);
    }

    // AQS 继承过来的方法, 方便阅读, 放在此处
    public final void acquire(int arg) {
        if (
                !tryAcquire(arg) &&
                        acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
        ) {
            selfInterrupt();
        }
    }
    // 与非公平锁主要区别在于 tryAcquire 方法的实现
    protected final boolean tryAcquire(int acquires) {
        final Thread current = Thread.currentThread();
        int c = getState();
        if (c == 0) {
            // 先检查 AQS 队列中是否有前驱节点, 没有才去竞争
            if (!hasQueuedPredecessors() &&
                    compareAndSetState(0, acquires)) {
                setExclusiveOwnerThread(current);
                return true;
            }
        }
        else if (current == getExclusiveOwnerThread()) {
            int nextc = c + acquires;
            if (nextc < 0)
                throw new Error("Maximum lock count exceeded");
            setState(nextc);
            return true;
        }
        return false;
    }

    // ㈠ AQS 继承过来的方法, 方便阅读, 放在此处
    public final boolean hasQueuedPredecessors() {
        Node t = tail;
        Node h = head;
        Node s;
        // h != t 时表示队列中有 Node
        return h != t &&
                (
                        // (s = h.next) == null 表示队列中还有没有老二
                        (s = h.next) == null || // 或者队列中老二线程不是此线程
                                s.thread != Thread.currentThread()
                );
    }
}
           

可重入原理

同一个线程, 锁重入, 会对state进行自增. 释放锁的时候, state进行自减; 当state自减为0的时候. 此时线程才会将锁释放成功, 才会进一步去唤醒其他线程来竞争锁

static final class NonfairSync extends Sync {
    // ...

    // Sync 继承过来的方法, 方便阅读, 放在此处
    final boolean nonfairTryAcquire(int acquires) {
        final Thread current = Thread.currentThread();
        int c = getState();
        if (c == 0) {
            if (compareAndSetState(0, acquires)) {
                setExclusiveOwnerThread(current);
                return true;
            }
        }
        // 如果已经获得了锁, 线程还是当前线程, 表示发生了锁重入
        else if (current == getExclusiveOwnerThread()) {
            // state++
            int nextc = c + acquires;
            if (nextc < 0) // overflow
                throw new Error("Maximum lock count exceeded");
            setState(nextc);
            return true;
        }
        return false;
    }

    // Sync 继承过来的方法, 方便阅读, 放在此处
    protected final boolean tryRelease(int releases) {
        // state--
        int c = getState() - releases;
        if (Thread.currentThread() != getExclusiveOwnerThread())
            throw new IllegalMonitorStateException();
        boolean free = false;
        // 支持锁重入, 只有 state 减为 0, 才释放成功
        if (c == 0) {
            free = true;
            setExclusiveOwnerThread(null);
        }
        setState(c);
        return free;
    }
}
           

可打断原理

不可打断模式:在此模式下,即使它被打断,仍会驻留在 AQS 队列中,一直要等到获得锁后方能得知自己被打断了

// Sync 继承自 AQS
static final class NonfairSync extends Sync {
    // ...

    private final boolean parkAndCheckInterrupt() {
        // 如果打断标记已经是 true, 则 park 会失效
        // 被park阻塞的线程, 可以被别的线程调用它的interrupt方法打断该park阻塞
        LockSupport.park(this);
        // interrupted 会清除打断标记; 下次park仍然可以阻塞
        return Thread.interrupted();
    }

    final boolean acquireQueued(final Node node, int arg) {
        boolean failed = true;
        try {
            boolean interrupted = false;
            for (;;) {
                final Node p = node.predecessor();
                if (p == head && tryAcquire(arg)) {
                    setHead(node);
                    p.next = null;
                    failed = false;
                    // 还是需要获得锁后, 才能返回打断状态
                    return interrupted;
                }
                if (
                        shouldParkAfterFailedAcquire(p, node) &&
                                parkAndCheckInterrupt()
                ) {
                    // 如果是因为 interrupt 被唤醒, 返回打断状态为 true
                    interrupted = true;
                }
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }

    public final void acquire(int arg) {
        if (
                !tryAcquire(arg) &&
                        acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
        ) {
            // 如果打断状态为 true
            selfInterrupt();
        }
    }

    static void selfInterrupt() {
        // 重新产生一次中断,这时候线程是如果正常运行的状态,那么不是出于sleep等状态,interrupt方法就不会报错
        Thread.currentThread().interrupt();
    }
}
}
           

可打断模式

static final class NonfairSync extends Sync {
    public final void acquireInterruptibly(int arg) throws InterruptedException {
        if (Thread.interrupted())
            throw new InterruptedException();
        // 如果没有获得到锁, 进入 ㈠
        if (!tryAcquire(arg))
            doAcquireInterruptibly(arg);
    }

    // ㈠ 可打断的获取锁流程
    private void doAcquireInterruptibly(int arg) throws InterruptedException {
        final Node node = addWaiter(Node.EXCLUSIVE);
        boolean failed = true;
        try {
            for (;;) {
                final Node p = node.predecessor();
                if (p == head && tryAcquire(arg)) {
                    setHead(node);
                    p.next = null; // help GC
                    failed = false;
                    return;
                }
                if (shouldParkAfterFailedAcquire(p, node) &&
                        parkAndCheckInterrupt()) {
                    // 在 park 过程中如果被 interrupt 会进入此
                    // 这时候抛出异常, 而不会再次进入 for (;;)
                    throw new InterruptedException();
                }
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }
}
           

条件变量实现原理

每个条件变量其实就对应着一个等待队列,其实现类是 ConditionObject。

await 流程

开始 Thread-0 持有锁,conditionObject对象调用 await,进入 ConditionObject 的 addConditionWaiter 流程 创建新的 Node 状态为 -2(Node.CONDITION),关联 Thread-0,加入等待队列尾部

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

接下来进入 AQS 的 fullyRelease 流程,释放同步器上的所有的锁 (因为可能线程发生可重入, 锁有很多层)

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

unparkSuccessor(h); —> unpark唤醒 AQS 队列中的下一个节点,竞争锁,假设没有其他竞争线程,那么 Thread-1 竞争成功

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

LockSupport.park(this); —> park 阻塞 Thread-0

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

signal 流程

假设 Thread-1 要来唤醒 Thread-0

// 如果没有持有锁,会抛出异常 --> 这里表示Thread-1要持有锁, 
	//才可以去条件变量中去唤醒等待的线程
        if (!isHeldExclusively()) 
           
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

进入 ConditionObject 的 doSignal 流程,取得等待队列中第一个 Node,即 Thread-0 所在 Node

private void doSignal(Node first) {
    do {
    	// 去firstWaiter条件变量中将等待的线程拿出来.
        if ( (firstWaiter = first.nextWaiter) == null)
            lastWaiter = null;
        first.nextWaiter = null;
        // 转移到AQS的队列中, 等待竞争锁
    } while (!transferForSignal(first) &&
             (first = firstWaiter) != null);
}
           
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

执行 transferForSignal 流程,将该 Node 加入 AQS 队列尾部,将 Thread-0 的 waitStatus 改为 0,Thread-3 的waitStatus 改为 -1 , 改为-1就有责任去唤醒自己的后继节点

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

Thread-1 释放锁,进入 unlock 流程,略

源码分析

public class ConditionObject implements Condition, java.io.Serializable {
    private static final long serialVersionUID = 1173984872572414699L;

    // 第一个等待节点
    private transient Node firstWaiter;

    // 最后一个等待节点
    private transient Node lastWaiter;
    public ConditionObject() { }
    // ㈠ 添加一个 Node 至等待队列
    private Node addConditionWaiter() {
        Node t = lastWaiter;
        // 所有已取消的 Node 从队列链表删除, 见 ㈡
        if (t != null && t.waitStatus != Node.CONDITION) {
            unlinkCancelledWaiters();
            t = lastWaiter;
        }
        // 创建一个关联当前线程的新 Node, 添加至队列尾部
        Node node = new Node(Thread.currentThread(), Node.CONDITION);
        if (t == null)
            firstWaiter = node;
        else
            t.nextWaiter = node;
        lastWaiter = node;
        return node;
    }
    // 唤醒 - 将没取消的第一个节点转移至 AQS 队列
    private void doSignal(Node first) {
        do {
            // 已经是尾节点了
            if ( (firstWaiter = first.nextWaiter) == null) {
                lastWaiter = null;
            }
            first.nextWaiter = null;
        } while (
            // 将等待队列中的 Node 转移至 AQS 队列, 不成功且还有节点则继续循环 ㈢
                !transferForSignal(first) &&
                        // 队列还有节点
                        (first = firstWaiter) != null
        );
    }

    // 外部类方法, 方便阅读, 放在此处
    // ㈢ 如果节点状态是取消, 返回 false 表示转移失败, 否则转移成功
    final boolean transferForSignal(Node node) {
        // 设置当前node状态为0(因为处在队列末尾),如果状态已经不是 Node.CONDITION, 说明被取消了
        if (!compareAndSetWaitStatus(node, Node.CONDITION, 0))
            return false;
        // 加入 AQS 队列尾部
        Node p = enq(node);
        int ws = p.waitStatus;
        if (
            // 插入节点的上一个节点被取消
                ws > 0 ||
                        // 插入节点的上一个节点不能设置状态为 Node.SIGNAL
                        !compareAndSetWaitStatus(p, ws, Node.SIGNAL)
        ) {
            // unpark 取消阻塞, 让线程重新同步状态
            LockSupport.unpark(node.thread);
        }
        return true;
    }
// 全部唤醒 - 等待队列的所有节点转移至 AQS 队列
private void doSignalAll(Node first) {
    lastWaiter = firstWaiter = null;
    do {
        Node next = first.nextWaiter;
        first.nextWaiter = null;
        transferForSignal(first);
        first = next;
    } while (first != null);
}

    // ㈡
    private void unlinkCancelledWaiters() {
        // ...
    }
    // 唤醒 - 必须持有锁才能唤醒, 因此 doSignal 内无需考虑加锁
    public final void signal() {
        // 如果没有持有锁,会抛出异常
        if (!isHeldExclusively())
            throw new IllegalMonitorStateException();
        Node first = firstWaiter;
        if (first != null)
            doSignal(first);
    }
    // 全部唤醒 - 必须持有锁才能唤醒, 因此 doSignalAll 内无需考虑加锁
    public final void signalAll() {
        if (!isHeldExclusively())
            throw new IllegalMonitorStateException();
        Node first = firstWaiter;
        if (first != null)
            doSignalAll(first);
    }
    // 不可打断等待 - 直到被唤醒
    public final void awaitUninterruptibly() {
        // 添加一个 Node 至等待队列, 见 ㈠
        Node node = addConditionWaiter();
        // 释放节点持有的锁, 见 ㈣
        int savedState = fullyRelease(node);
        boolean interrupted = false;
        // 如果该节点还没有转移至 AQS 队列, 阻塞
        while (!isOnSyncQueue(node)) {
            // park 阻塞
            LockSupport.park(this);
            // 如果被打断, 仅设置打断状态
            if (Thread.interrupted())
                interrupted = true;
        }
        // 唤醒后, 尝试竞争锁, 如果失败进入 AQS 队列
        if (acquireQueued(node, savedState) || interrupted)
            selfInterrupt();
    }
    // 外部类方法, 方便阅读, 放在此处
    // ㈣ 因为某线程可能重入,需要将 state 全部释放,获取state,然后把它全部减掉,以全部释放
    final int fullyRelease(Node node) {
        boolean failed = true;
        try {
            int savedState = getState();
            // 唤醒等待队列队列中的下一个节点
            if (release(savedState)) {
                failed = false;
                return savedState;
            } else {
                throw new IllegalMonitorStateException();
            }
        } finally {
            if (failed)
                node.waitStatus = Node.CANCELLED;
        }
    }
    // 打断模式 - 在退出等待时重新设置打断状态
    private static final int REINTERRUPT = 1;
    // 打断模式 - 在退出等待时抛出异常
    private static final int THROW_IE = -1;
    // 判断打断模式
    private int checkInterruptWhileWaiting(Node node) {
        return Thread.interrupted() ?
                (transferAfterCancelledWait(node) ? THROW_IE : REINTERRUPT) :
                0;
    }
    // ㈤ 应用打断模式
    private void reportInterruptAfterWait(int interruptMode)
            throws InterruptedException {
        if (interruptMode == THROW_IE)
            throw new InterruptedException();
        else if (interruptMode == REINTERRUPT)
            selfInterrupt();
    }
    // 等待 - 直到被唤醒或打断
    public final void await() throws InterruptedException {
        if (Thread.interrupted()) {
            throw new InterruptedException();
        }
        // 添加一个 Node 至等待队列, 见 ㈠
        Node node = addConditionWaiter();
        // 释放节点持有的锁
        int savedState = fullyRelease(node);
        int interruptMode = 0;
        // 如果该节点还没有转移至 AQS 队列, 阻塞
        while (!isOnSyncQueue(node)) {
            // park 阻塞              
            LockSupport.park(this);
            // 如果被打断, 退出等待队列
            if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
                break;
        }
        // 退出等待队列后, 还需要获得 AQS 队列的锁
        if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
            interruptMode = REINTERRUPT;
        // 所有已取消的 Node 从队列链表删除, 见 ㈡
        if (node.nextWaiter != null)
            unlinkCancelledWaiters();
        // 应用打断模式, 见 ㈤
        if (interruptMode != 0)
            reportInterruptAfterWait(interruptMode);
    }
    // 等待 - 直到被唤醒或打断或超时
    public final long awaitNanos(long nanosTimeout) throws InterruptedException {
        if (Thread.interrupted()) {
            throw new InterruptedException();
        }
        // 添加一个 Node 至等待队列, 见 ㈠
        Node node = addConditionWaiter();
        // 释放节点持有的锁
        int savedState = fullyRelease(node);
        // 获得最后期限
        final long deadline = System.nanoTime() + nanosTimeout;
        int interruptMode = 0;
        // 如果该节点还没有转移至 AQS 队列, 阻塞
        while (!isOnSyncQueue(node)) {
            // 已超时, 退出等待队列
            if (nanosTimeout <= 0L) {
                transferAfterCancelledWait(node);
                break;
            }
            // park 阻塞一定时间, spinForTimeoutThreshold 为 1000 ns
            if (nanosTimeout >= spinForTimeoutThreshold)
                LockSupport.parkNanos(this, nanosTimeout);
            // 如果被打断, 退出等待队列
            if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
                break;
            nanosTimeout = deadline - System.nanoTime();
        }
        // 退出等待队列后, 还需要获得 AQS 队列的锁
        if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
            interruptMode = REINTERRUPT;
        // 所有已取消的 Node 从队列链表删除, 见 ㈡
        if (node.nextWaiter != null)
            unlinkCancelledWaiters();
        // 应用打断模式, 见 ㈤
        if (interruptMode != 0)
            reportInterruptAfterWait(interruptMode);
        return deadline - System.nanoTime();
    }
    // 等待 - 直到被唤醒或打断或超时, 逻辑类似于 awaitNanos
    public final boolean awaitUntil(Date deadline) throws InterruptedException {
        // ...
    }
    // 等待 - 直到被唤醒或打断或超时, 逻辑类似于 awaitNanos
    public final boolean await(long time, TimeUnit unit) throws InterruptedException {
        // ...
    }
    // 工具方法 省略 ...
}
           

五、ReentrantReadWriteLock 读写锁

当读操作远远高于写操作时,这时候使用 读写锁 让 读-读 可以并发,提高性能。

类似于数据库中的 select …from … lock in share mode

提供一个 数据容器类 内部分别使用读锁保护数据的 read() 方法,写锁保护数据的 write() 方法

class DataContainer {
    private Object data;
    private ReentrantReadWriteLock rw = new ReentrantReadWriteLock();
    private ReentrantReadWriteLock.ReadLock r = rw.readLock();
    private ReentrantReadWriteLock.WriteLock w = rw.writeLock();
 
    public Object read() {
        log.debug("获取读锁...");
        r.lock();
        try {
            log.debug("读取");
            sleep(1);
            return data;
        } finally {
            log.debug("释放读锁...");
            r.unlock();
        }
    }
 
    public void write() {
        log.debug("获取写锁...");
        w.lock();
        try {
            log.debug("写入");
            sleep(1);
        } finally {
            log.debug("释放写锁...");
            w.unlock();
        }
    }
}
           

先看【读——读】

@Slf4j(topic = "c.TestReadWriteLock")
public class TestReadWriteLock {
    public static void main(String[] args) throws InterruptedException {
        DataContainer dataContainer = new DataContainer();
        new Thread(() -> {
            dataContainer.read();
        }, "t1").start();
 
        new Thread(() -> {
            dataContainer.read();
        }, "t2").start();
    }
}
           
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

输出结果,从这里可以看到 Thread-0 锁定期间,Thread-1 的读操作不受影响。

再看【读——写】

public static void main(String[] args) throws InterruptedException {
        DataContainer dataContainer = new DataContainer();
        new Thread(() -> {
            dataContainer.read();
        }, "t1").start();
 
        new Thread(() -> {
            dataContainer.write();
        }, "t2").start();
    }
           
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

可以看到读写是互斥的。

再看【写——写】

public static void main(String[] args) throws InterruptedException {
        DataContainer dataContainer = new DataContainer();
        new Thread(() -> {
            dataContainer.write();
        }, "t1").start();
 
        new Thread(() -> {
            dataContainer.write();
        }, "t2").start();
    }
           
2021-08-03 22:14:01 获取写锁...
2021-08-03 22:14:01 获取写锁...
2021-08-03 22:14:01 写入
2021-08-03 22:14:02 释放写锁...
2021-08-03 22:14:02 写入
2021-08-03 22:14:03 释放写锁...
           

写锁-写锁 也是相互阻塞的。

锁的降级

public void write_read() {
        log.debug("获取锁...");
        w.lock();
        try {
            log.debug("准备降级成读锁");
            sleep(1);
            r.lock();
        } finally {
            log.debug("释放读写锁...");
            r.unlock();
            w.unlock();
        } 
}           
           
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

锁的升级

public void read_write() {
        log.debug("获取锁...");
        r.lock();
        try {
            log.debug("准备升级成读锁");
            sleep(1);
            w.lock();
        } finally {
            log.debug("释放读写锁...");
            w.unlock();
            r.unlock();
        }
    }
           
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

为什么需要锁的降级?

如果我们在刚才的方法中,一直使用写锁,最后才释放写锁的话,虽然确实是线程安全的,但是也是没有必要的,因为我们只有一处修改数据的代码:data = new Object();后面我们对于 data 仅仅是读取。如果还一直使用写锁的话,就不能让多个线程同时来读取了,持有写锁是浪费资源的,降低了整体的效率,所以这个时候利用锁的降级是很好的办法,可以提高整体性能。

为什么不支持锁的升级?

读写锁的特点是如果线程都申请读锁,是可以多个线程同时持有的,可是如果是写锁,只能有一个线程持有,并且不可能存在读锁和写锁同时持有的情况。正是因为不可能有读锁和写锁同时持有的情况,所以升级写锁的过程中,需要等到所有的读锁都释放,此时才能进行升级。

假设有 A,B 和 C 三个线程,它们都已持有读锁。假设线程 A 尝试从读锁升级到写锁。那么它必须等待 B 和 C 释放掉已经获取到的读锁。如果随着时间推移,B 和 C 逐渐释放了它们的读锁,此时线程 A 确实是可以成功升级并获取写锁。

但是我们考虑一种特殊情况。假设线程 A 和 B 都想升级到写锁,那么对于线程 A 而言,它需要等待其他所有线程,包括线程 B 在内释放读锁。而线程 B 也需要等待所有的线程,包括线程 A 释放读锁。这就是一种非常典型的死锁的情况。谁都愿不愿意率先释放掉自己手中的锁。

但是读写锁的升级并不是不可能的,也有可以实现的方案,如果我们保证每次只有一个线程可以升级,那么就可以保证线程安全。只不过最常见的

ReentrantReadWriteLock

对此并不支持。

读写锁原理

t1线程是进行写,t2是读,t3是读,t4是写,因此把所有的案例基本都考虑到了,写-读,读-读,读-写。

读写锁用的是同一个 Sycn 同步器,因此等待队列、state 等也是同一个t1 w.lock,t2 r.lock,

(1)t1 成功上锁,流程与

ReentrantLock

加锁相比没有特殊之处,不同是写锁状态占了 state 的低 16 位,而读锁使用的是 state 的高 16 位。

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

(2) t2 执行 r.lock,这时进入读锁的 sync.acquireShared(1) 流程,首先会进入 tryAcquireShared 流程。如果有写锁占据,那么 tryAcquireShared 返回 -1 表示失败.

-1 表示失败

0 表示成功,但后继节点不会继续唤醒

正数表示成功,而且数值是还有几个后继节点需要唤醒,读写锁返回 1

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

(3) 这时会进入 sync.doAcquireShared(1) 流程,首先也是调用 addWaiter 添加节点,不同之处在于节点被设置为Node.SHARED 模式而非 Node.EXCLUSIVE 模式,注意此时 t2 仍处于活跃状态

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

(4) t2 会看看自己的节点是不是老二,如果是,还会再次调用 tryAcquireShared(1) 来尝试获取锁

5)如果没有成功,在 doAcquireShared 内 for ( 循环一次,把前驱节点的 waitStatus 改为 -1,再 for ( 循环一次尝试 tryAcquireShared(1) 如果还不成功,那么在 parkAndCheckInterrupt() 处 park

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

t3 r.lock,t4 w.lock

这种状态下,假设又有 t3 加读锁和 t4 加写锁,这期间 t1 仍然持有锁,就变成了下面的样子

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

t1 w.unlock

这时会走到写锁的 sync.release(1) 流程,调用 sync.tryRelease(1) 成功,变成下面的样子

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

接下来执行唤醒流程 sync.unparkSuccessor,即让老二恢复运行,这时 t2 在 doAcquireShared 内parkAndCheckInterrupt() 处恢复运行

这回再来一次 for (执行 tryAcquireShared 成功则让读锁计数加一

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

这时 t2 已经恢复运行,接下来 t2 调用 setHeadAndPropagate(node, 1),它原本所在节点被置为头节点

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

事情还没完,在 setHeadAndPropagate 方法内还会检查下一个节点是否是 shared,如果是则调用doReleaseShared() 将 head 的状态从 -1 改为 0 并唤醒老二,这时 t3 在 doAcquireShared 内parkAndCheckInterrupt() 处恢复运行

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

这回再来一次 for ( 执行 tryAcquireShared 成功则让读锁计数加一

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

这时 t3 已经恢复运行,接下来 t3 调用 setHeadAndPropagate(node, 1),它原本所在节点被置为头节点

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

下一个节点不是 shared 了,因此不会继续唤醒 t4 所在节点t2 r.unlock,t3 r.unlock,t2 进入 sync.releaseShared(1) 中,调用 tryReleaseShared(1) 让计数减一,但由于计数还不为零

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

t3 进入 sync.releaseShared(1) 中,调用 tryReleaseShared(1) 让计数减一,这回计数为零了,进入doReleaseShared() 将头节点从 -1 改为 0 并唤醒老二,即

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

之后 t4 在 acquireQueued 中 parkAndCheckInterrupt 处恢复运行,再次 for (;😉 这次自己是老二,并且没有其他竞争,tryAcquire(1) 成功,修改头结点,流程结束

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

源码分析:

写锁上锁流程

static final class NonfairSync extends Sync {
    // ... 省略无关代码
// 外部类 WriteLock 方法, 方便阅读, 放在此处
    public void lock() {
        sync.acquire(1);
    }
    // AQS 继承过来的方法, 方便阅读, 放在此处
    public final void acquire(int arg) {
        if (
// 尝试获得写锁失败
                !tryAcquire(arg) &&
// 将当前线程关联到一个 Node 对象上, 模式为独占模式
// 进入 AQS 队列阻塞
                        acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
        ) {
            selfInterrupt();
        }
    }
    // Sync 继承过来的方法, 方便阅读, 放在此处
    protected final boolean tryAcquire(int acquires) {
// 获得低 16 位, 代表写锁的 state 计数
        Thread current = Thread.currentThread();
        int c = getState();
        int w = exclusiveCount(c);
        if (c != 0) {
            if (
// c != 0 and w == 0 表示有读锁, 或者
                    w == 0 ||
// 如果 exclusiveOwnerThread 不是自己
                            current != getExclusiveOwnerThread()
            ) {
// 获得锁失败
                return false;
            }
// 写锁计数超过低 16 位, 报异常
            if (w + exclusiveCount(acquires) > MAX_COUNT)
                throw new Error("Maximum lock count exceeded");
// 写锁重入, 获得锁成功
            setState(c + acquires);
            return true;
        }
        if (
// 判断写锁是否该阻塞, 或者
                writerShouldBlock() ||
// 尝试更改计数失败
                        !compareAndSetState(c, c + acquires)
        ) {
// 获得锁失败
            return false;
        }
// 获得锁成功
        setExclusiveOwnerThread(current);
        return true;
    }
    // 非公平锁 writerShouldBlock 总是返回 false, 无需阻塞
    final boolean writerShouldBlock() {
        return false;
    }
}
           

写锁释放流程

static final class NonfairSync extends Sync {
    // ... 省略无关代码
// WriteLock 方法, 方便阅读, 放在此处
    public void unlock() {
        sync.release(1);
    }
    // AQS 继承过来的方法, 方便阅读, 放在此处
    public final boolean release(int arg) {
// 尝试释放写锁成功
        if (tryRelease(arg)) {
// unpark AQS 中等待的线程
            Node h = head;
            if (h != null && h.waitStatus != 0)
                unparkSuccessor(h);
            return true;
        }
        return false;
    }
    // Sync 继承过来的方法, 方便阅读, 放在此处
    protected final boolean tryRelease(int releases) {
        if (!isHeldExclusively())
            throw new IllegalMonitorStateException();
        int nextc = getState() - releases;
// 因为可重入的原因, 写锁计数为 0, 才算释放成功
        boolean free = exclusiveCount(nextc) == 0;
        if (free) {
            setExclusiveOwnerThread(null);
        }
        setState(nextc);
        return free;
    }
}
           

读锁上锁流程

static final class NonfairSync extends Sync {
    // ReadLock 方法, 方便阅读, 放在此处
    public void lock() {
        sync.acquireShared(1);
    }
    // AQS 继承过来的方法, 方便阅读, 放在此处
    public final void acquireShared(int arg) {
// tryAcquireShared 返回负数, 表示获取读锁失败
        if (tryAcquireShared(arg) < 0) {
            doAcquireShared(arg);
        }
    }
    // Sync 继承过来的方法, 方便阅读, 放在此处
    protected final int tryAcquireShared(int unused) {
        Thread current = Thread.currentThread();
        int c = getState();
// 如果是其它线程持有写锁, 获取读锁失败
        if (
                exclusiveCount(c) != 0 &&
                        getExclusiveOwnerThread() != current
        ) {
            return -1;
        }
        int r = sharedCount(c);
        if (
// 读锁不该阻塞(如果老二是写锁,读锁该阻塞), 并且
                !readerShouldBlock() &&
// 小于读锁计数, 并且
                        r < MAX_COUNT &&
// 尝试增加计数成功
                        compareAndSetState(c, c + SHARED_UNIT)
        ) {
// ... 省略不重要的代码
            return 1;
        }
        return fullTryAcquireShared(current);
    }
    // 非公平锁 readerShouldBlock 看 AQS 队列中第一个节点是否是写锁
// true 则该阻塞, false 则不阻塞
    final boolean readerShouldBlock() {
        return apparentlyFirstQueuedIsExclusive();
    }
    // AQS 继承过来的方法, 方便阅读, 放在此处
// 与 tryAcquireShared 功能类似, 但会不断尝试 for (;;) 获取读锁, 执行过程中无阻塞
    final int fullTryAcquireShared(Thread current) {
        HoldCounter rh = null;
        for (;;) {
            int c = getState();
            if (exclusiveCount(c) != 0) {
                if (getExclusiveOwnerThread() != current)
                    return -1;
            } else if (readerShouldBlock()) {
// ... 省略不重要的代码
            }
            if (sharedCount(c) == MAX_COUNT)
                throw new Error("Maximum lock count exceeded");
            if (compareAndSetState(c, c + SHARED_UNIT)) {
// ... 省略不重要的代码
                return 1;
            }
        }
    }
    // AQS 继承过来的方法, 方便阅读, 放在此处
    private void doAcquireShared(int arg) {
// 将当前线程关联到一个 Node 对象上, 模式为共享模式
        final Node node = addWaiter(Node.SHARED);
        boolean failed = true;
        try {
            boolean interrupted = false;
            for (;;) {
                final Node p = node.predecessor();
                if (p == head) {
// 再一次尝试获取读锁
                    int r = tryAcquireShared(arg);
// 成功
                    if (r >= 0) {
// ㈠
// r 表示可用资源数, 在这里总是 1 允许传播
//(唤醒 AQS 中下一个 Share 节点)
                        setHeadAndPropagate(node, r);
                        p.next = null; // help GC
                        if (interrupted)
                            selfInterrupt();
                        failed = false;
                        return;
                    }
                }
                if (
// 是否在获取读锁失败时阻塞(前一个阶段 waitStatus == Node.SIGNAL)
                        shouldParkAfterFailedAcquire(p, node) &&
// park 当前线程
                                parkAndCheckInterrupt()
                ) {
                    interrupted = true;
                }
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }
    // ㈠ AQS 继承过来的方法, 方便阅读, 放在此处
    private void setHeadAndPropagate(Node node, int propagate) {
        Node h = head; // Record old head for check below
// 设置自己为 head
        setHead(node);
// propagate 表示有共享资源(例如共享读锁或信号量)
// 原 head waitStatus == Node.SIGNAL 或 Node.PROPAGATE
// 现在 head waitStatus == Node.SIGNAL 或 Node.PROPAGATE
        if (propagate > 0 || h == null || h.waitStatus < 0 ||
                (h = head) == null || h.waitStatus < 0) {
            Node s = node.next;
// 如果是最后一个节点或者是等待共享读锁的节点
            if (s == null || s.isShared()) {
// 进入 ㈡
                doReleaseShared();
            }
        }
    }
    // ㈡ AQS 继承过来的方法, 方便阅读, 放在此处
    private void doReleaseShared() {
// 如果 head.waitStatus == Node.SIGNAL ==> 0 成功, 下一个节点 unpark
// 如果 head.waitStatus == 0 ==> Node.PROPAGATE, 为了解决 bug, 见后面分析
        for (;;) {
            Node h = head;
// 队列还有节点
            if (h != null && h != tail) {
                int ws = h.waitStatus;
                if (ws == Node.SIGNAL) {
                    if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))
                        continue; // loop to recheck cases
// 下一个节点 unpark 如果成功获取读锁
// 并且下下个节点还是 shared, 继续 doReleaseShared
                    unparkSuccessor(h);
                }
                else if (ws == 0 &&
                        !compareAndSetWaitStatus(h, 0, Node.PROPAGATE))
                    continue; // loop on failed CAS
            }
            if (h == head) // loop if head changed
                break;
        }
    }
}
           

读锁释放流程

static final class NonfairSync extends Sync {
    // ReadLock 方法, 方便阅读, 放在此处
    public void unlock() {
        sync.releaseShared(1);
    }
    // AQS 继承过来的方法, 方便阅读, 放在此处
    public final boolean releaseShared(int arg) {
        if (tryReleaseShared(arg)) {
            doReleaseShared();
            return true;
        }
        return false;
    }
    // Sync 继承过来的方法, 方便阅读, 放在此处
    protected final boolean tryReleaseShared(int unused) {
// ... 省略不重要的代码
        for (;;) {
            int c = getState();
            int nextc = c - SHARED_UNIT;
            if (compareAndSetState(c, nextc)) {
// 读锁的计数不会影响其它获取读锁线程, 但会影响其它获取写锁线程
// 计数为 0 才是真正释放
                return nextc == 0;
            }
        }
    }
    // AQS 继承过来的方法, 方便阅读, 放在此处
    private void doReleaseShared() {
// 如果 head.waitStatus == Node.SIGNAL ==> 0 成功, 下一个节点 unpark
// 如果 head.waitStatus == 0 ==> Node.PROPAGATE
        for (;;) {
            Node h = head;
            if (h != null && h != tail) {
                int ws = h.waitStatus;
// 如果有其它线程也在释放读锁,那么需要将 waitStatus 先改为 0
// 防止 unparkSuccessor 被多次执行
                if (ws == Node.SIGNAL) {
                    if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))
                        continue; // loop to recheck cases
                    unparkSuccessor(h);
                }
// 如果已经是 0 了,改为 -3,用来解决传播性,见后文信号量 bug 分析
                else if (ws == 0 &&
                        !compareAndSetWaitStatus(h, 0, Node.PROPAGATE))
                    continue; // loop on failed CAS
            }
            if (h == head) // loop if head changed
                break;
        }
    }
}
           

六、StampedLock

该类自JDK8加入,是为了进一步优化读性能,它的特点是在使用读锁、写锁时都必须配合【戳】使用。

加解读锁

stamp = lock.readLock();
    lock.unlockRead(stamp);
           

加解写锁

long stamp = lock.writeLock();
    lock.unlockWrite(stamp);
           

乐观读,StampedLock支持tryOptimisticRead()方法(乐观读),读取完毕后需要做一次“戳校验”如果校验通过,表示这期间确实没有写操作,数据可以安全使用,如果校验没有通过,需要重新获取读锁(进行锁的升级),保证数据安全。、

//获取读锁的戳
long stamp = lock.tryOptimisticRead();
//判断戳是否改变了
if(!lock.validate(stamp)){
    //锁升级
}
           

StampedLock的使用:

public class StampLockTest {
    public static void main(String[] args) {
        DataContainerStamped dataContainerStamped = new DataContainerStamped(111);
        new Thread(()->{
            int read = dataContainerStamped.read(1000);
            log.info("get data:{}",read);
        },"t1").start();
        try {
            Thread.sleep(500);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        new Thread(()->{
            dataContainerStamped.write(1);
        },"t2").start();

    }
}

@Slf4j(topic = "c.com.jl.juc.stamplock.DataContainerStamped")
class DataContainerStamped{
    private int data;
    private final StampedLock lock = new StampedLock();

    public DataContainerStamped(int data) {
        this.data = data;
    }

    public int read(int readTime){
        //获取读锁的戳
        long stamp = lock.tryOptimisticRead();
        log.info("optimistic read locking...{}",stamp);
        try {
            Thread.sleep(readTime);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        //判断戳是否改变了
        if(lock.validate(stamp)){
            //没有改变则内容未被其他线程修改,则正常获取数据,正常返回
            log.info("read finish...{}",stamp);
            return data;
        }
        //戳被改变了,进行读锁的升级
        log.info("updating to read lock...{}",stamp);
        try {
            //获取读锁
            stamp = lock.readLock();
            log.info("read lock {}",stamp);
            try {
                Thread.sleep(readTime);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            log.info("read finish...{}",stamp);
            return data;
        } finally {
            lock.unlockRead(stamp);
        }

    }

    public void write(int newData){
        long stamp = lock.writeLock();
        log.info("write lock {}",stamp);
        try {
            try {
                Thread.sleep(2000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            this.data = newData;
        } finally {
            log.info("write unlock {}",stamp);
            lock.unlockWrite(stamp);
        }

    }
}
           
13:40:33.293 [t1] INFO c.com.jl.juc.stamplock.DataContainerStamped - optimistic read locking…256
13:40:33.794 [t2] INFO c.com.jl.juc.stamplock.DataContainerStamped - write lock 384
13:40:34.310 [t1] INFO c.com.jl.juc.stamplock.DataContainerStamped - updating to read lock…256
13:40:35.807 [t2] INFO c.com.jl.juc.stamplock.DataContainerStamped - write unlock 384
13:40:35.807 [t1] INFO c.com.jl.juc.stamplock.DataContainerStamped - read lock 513
13:40:36.816 [t1] INFO c.com.jl.juc.stamplock.DataContainerStamped - read finish…513
13:40:36.816 [t1] INFO jl.com.jl.juc.stamplock.StampLockTest - get data:1
           

步骤解析:

1.首先t1线程启动,调用read()方法

2.read()方法中调用lock.tryOptimisticRead()获取乐观锁,获取到戳stamp,打印optimistic read locking…256

3.然后进入Thread.sleep(readTime),线程进入阻塞

4.经过0.5秒后,t2线程启动,t2线程调用write()方法。

5.write()方法中,调用lock.writeLock()获取写锁(乐观读和写锁不互斥),获取成功并返回戳stamp,打印write lock 384

6.然后t2线程进入Thread.sleep(2000)阻塞

7.经过1秒后t1线程被唤醒,继续执行,read()方法中通过调用lock.validate(stamp)进行校验戳(如果此时锁没有被其他线程获取写锁则返回成功)。此时由于t2已经获取了写锁,所以此时llock.validate(stamp)返回失败

8.t1线程进入获取读锁流程(锁升级),t1线程调用lock.readLock()方法获取写锁,由于t2线程写锁还没有释放,此时t1线程获取写锁失败进入阻塞。

9.经过2秒后t2线程被唤醒,执行后续代码,将属性data改为newData,然后释放写锁,释放锁的过程会唤醒等待队列里的阻塞线程(t1由于获取读锁失败进入等待队列,这里将t1唤醒)。

10.t1被唤醒后,获得到了读锁,执行后续流程,返回data值,最后释放读锁

注意:

  • StampedLock不支持条件变量
  • StampedLock不支持可重入

七、StampedLock

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

上面10个线程同时执行的,下面来限制线程个数:

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

semaphore的应用

  • 使用 Semaphore 限流,在访问高峰期时,让请求线程阻塞,高峰期过去再释放许可,当然它只适合限制单机线程数量,并且仅是限制线程数,而不是限制资源数(例如连接数,请对比 Tomcat LimitLatch 的实现)
  • 用 Semaphore 实现简单连接池,对比『享元模式』下的实现(用wait notify),性能和可读性显然更好, 注意下面的实现中线程数和数据库连接数是相等的

semaphore原理

1. 加锁解锁流程

Semaphore 有点像一个停车场,permits 就好像停车位数量,当线程获得了 permits 就像是获得了停车位,然后停车场显示空余车位减一

刚开始,permits(state)为 3,这时 5 个线程来获取资源

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

假设其中 Thread-1,Thread-2,Thread-4 cas 竞争成功,而 Thread-0 和 Thread-3 竞争失败,进入 AQS 队列 park 阻塞。

【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock
【八】多线程 —— AQS原理、ReentrantLock原理一、AQS 框架简介一、AQS 核心思想二、AQS内部的数据结构与原理三、子类要实现的方法四、ReentrantLock 原理五、ReentrantReadWriteLock 读写锁六、StampedLock七、StampedLock

继续阅读