读写锁原理解读

r囧r小猫 2023-10-12 21:04 86阅读 0赞

**目录**

回顾什么是读写锁

t1 w.lock，t2 r.lock

t3 r.lock，t4 w.lock

t1 w.unlock

t2 r.unlock，t3 r.unlock

写锁上锁流程

写锁释放流程

读锁上锁流程

读锁释放流程

--------------------

#### 回顾什么是读写锁 ####

读写锁是一对互斥锁，分为读锁和写锁。读锁和写锁互斥，让一个线程在进行读操作时，不允许其他线程的写操作，但是不影响其他线程的读操作；当一个线程在进行写操作时，不允许任何线程进行读操作或者写操作。读写锁用的是同一个 Sycn 同步器，因此等待队列、state 等也是同一个

#### t1 w.lock，t2 r.lock ####

1） t1 成功上锁，流程与 ReentrantLock 加锁相比没有特殊之处，**不同是写锁状态占了 state 的低 16 位，而读锁 使用的是 state 的高 16 位**

![d6ef8c2e28c147cbacb9fb9bbbeb88e7.png][]

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

> tryAcquireShared 返回值表示
> 
> \-1 表示失败
> 
> 0 表示成功，但后继节点不会继续唤醒
> 
> 正数表示成功，而且数值是还有几个后继节点需要唤醒，读写锁返回 1

![0d4eb70796a441828a5b216b007b965a.png][]

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

![ee41f9db09074c6cbb01dc64d5309b24.png][]

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

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

![b9cb9d7662384ac7b26afbfff63efeea.png][]

#### t3 r.lock，t4 w.lock ####

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

![441da96292cc443d91f113e238c9d9cb.png][]

#### t1 w.unlock ####

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

![d981e22d746541178ca418a09ed0906d.png][]

接下来执行唤醒流程 sync.unparkSuccessor，即让老二恢复运行，这时 t2 在 doAcquireShared 内parkAndCheckInterrupt() 处恢复运行 这回再来一次 for (;;) 执行 tryAcquireShared 成功则让读锁计数加一

![b60d0b9e144a4756b6813c688d9140ae.png][]

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

![dbd909d67ffd4895a2b5febcfc218b4f.png][]

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

![c0918b914b044c6686c91f568443de66.png][]

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

![9fb0f7c37c2a4e9780a636ffcac97e55.png][]

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

![ae2f94b6d6de477daae04a49ea2287c5.png][]

下一个节点不是 shared 了，因此不会继续唤醒 t4 所在节点

#### t2 r.unlock，t3 r.unlock ####

t2 进入 sync.releaseShared(1) 中，调用 tryReleaseShared(1) 让计数减一，但由于计数还不为零

![654e18774d5949abbd0d1db94b4713c5.png][]

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

![ecd4b4ece6884903baebba2af012f88f.png][]

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

![00a1f54e95864b43b556c0531b1e574e.png][]

#### 写锁上锁流程 ####

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;
            }
        }
    }

[d6ef8c2e28c147cbacb9fb9bbbeb88e7.png]: https://img-blog.csdnimg.cn/d6ef8c2e28c147cbacb9fb9bbbeb88e7.png
[0d4eb70796a441828a5b216b007b965a.png]: https://img-blog.csdnimg.cn/0d4eb70796a441828a5b216b007b965a.png
[ee41f9db09074c6cbb01dc64d5309b24.png]: https://img-blog.csdnimg.cn/ee41f9db09074c6cbb01dc64d5309b24.png
[b9cb9d7662384ac7b26afbfff63efeea.png]: https://img-blog.csdnimg.cn/b9cb9d7662384ac7b26afbfff63efeea.png
[441da96292cc443d91f113e238c9d9cb.png]: https://img-blog.csdnimg.cn/441da96292cc443d91f113e238c9d9cb.png
[d981e22d746541178ca418a09ed0906d.png]: https://img-blog.csdnimg.cn/d981e22d746541178ca418a09ed0906d.png
[b60d0b9e144a4756b6813c688d9140ae.png]: https://img-blog.csdnimg.cn/b60d0b9e144a4756b6813c688d9140ae.png
[dbd909d67ffd4895a2b5febcfc218b4f.png]: https://img-blog.csdnimg.cn/dbd909d67ffd4895a2b5febcfc218b4f.png
[c0918b914b044c6686c91f568443de66.png]: https://img-blog.csdnimg.cn/c0918b914b044c6686c91f568443de66.png
[9fb0f7c37c2a4e9780a636ffcac97e55.png]: https://img-blog.csdnimg.cn/9fb0f7c37c2a4e9780a636ffcac97e55.png
[ae2f94b6d6de477daae04a49ea2287c5.png]: https://img-blog.csdnimg.cn/ae2f94b6d6de477daae04a49ea2287c5.png
[654e18774d5949abbd0d1db94b4713c5.png]: https://img-blog.csdnimg.cn/654e18774d5949abbd0d1db94b4713c5.png
[ecd4b4ece6884903baebba2af012f88f.png]: https://img-blog.csdnimg.cn/ecd4b4ece6884903baebba2af012f88f.png
[00a1f54e95864b43b556c0531b1e574e.png]: https://img-blog.csdnimg.cn/00a1f54e95864b43b556c0531b1e574e.png