006-ReentrantReadWriteLock和StampLock详解

SunriseYin

已于 2023-04-28 22:53:21 修改

阅读量74

点赞数

分类专栏：并发文章标签： java jvm 分布式

于 2023-04-26 23:52:05 首次发布

本文链接：https://blog.csdn.net/yin10484039/article/details/130394559

版权

并发专栏收录该内容

10 篇文章 0 订阅

订阅专栏

ReentrantReadWriteLock

场景

我们直到独占锁每次访问资源只允许一个线程
但是我们程序往往是读多写少
只允许一个线程的话性能会由很大限制
所以就有了读写锁
我在读的时候，另外一个读的操作也可以获取锁
我在读的时候，另外一个写的操作不能获取锁，反之我在写的时候，读线程也不能获取锁
我在写的时候，另外一个写的线程不能获取锁

读读共享
读写互斥
写写互斥
ReentrantReadWriteLock就是这样一把可重入读写锁

示例：
在这里插入图片描述

模拟读读场景

模拟读写场景

写线程再读线程结束后才能拿到锁进行操作

如果写线程放第一个申请锁，那就只有写完释放锁了读操作才能进行

模拟写写场景

分析

目前用下来可能由如下几个问题

AQS只有一个状态，怎么能发两把锁，也就是读写互斥怎么实现？
读锁的可重入是怎么实现的？

可以先说一下结论，后边分析代码验证

一、读写互斥是根据 state 字段的高低位来判断的
int类型有4个字节，也就是4*8=32位
高16位给读锁用
低16位给写锁用
在判断的时候只需要进行移位操作后判断是否 >0 即可
二、读锁的可重入是使用ThreadLocal实现的
我们直到独占锁的可重入是判断是否是独占线程后state+1这个实在实现类中tryLock实现的
所以是否可重入完全就可以由实现类来控制

源码验证分析

我们先看写锁
因为写锁肯定是独占锁，逻辑比较熟悉
在这里插入图片描述
具体逻辑在tryAcquire(1)中

代码大致分了3块

获取锁标志state
有其他锁操作
没有锁操作

我们分别来看

写锁-tryAcquire()获取锁标志

exclusiveCount(state)
在这里插入图片描述
这个涉及到位运算，但是这个位运算很简单 SHARED_SHIFT = 16
$2^{16}-1$
数学上：获取16位二进制的最大值
物理上：获取16位都是1的二进制数
这是一个范式很多地方会用，我们记住就行
而 c & EXCLUSIVE_MASK （11得1，有0得0）
计算下来 c的高位全部抹零了，低位全部保留了
所以写锁就是用 state 的低16位做锁的标记

写锁-tryAcquire()没有其他锁分支

writerShouldBlock()
这个直接返回了false：在重入判断的时候设置不需要阻塞，也就是可重入用的
然后就 CAS state 从原值到原值+1,
这里也是重入的逻辑
然后返回true取反为false
设置独占锁线程，返回true

写锁-tryAcquire()有其他锁

在这里插入图片描述
因为这个是所锁分支
所以这里判断没有写锁或者不是当前线程独占，就返回false

这里是判断一下因为只有16位长度了加多了就加到读锁那里了，所以就直接异常了
如果校验通过，那就是重入了
直接设置state，返回true

写锁的加锁逻辑ok了
下边看下读锁

读锁-加锁

因为是一把共享锁
在这里插入图片描述
所以实现的是 tryAcquireShard(1)
我们在AQS那边其实已经分析过了
规范是如果tryAcquireShard(1)<0就是加锁失败
那我们就进入tryAcquireShared(1)看一下如何尝试加锁

加锁失败场景
exclusiveCount（1）就是获取写锁标记，非0则存在写锁
且当前独占线程不是当前线程则直接加锁失败

加锁成功场景-第一个线程
判断是否应该Block*

这个Block里边有解释，大致说的就是如果一个写线程在申请了那后续的读线程都先block，避免写线程很长时间或者长时间无法获取锁

不需要Block 并且写锁小于 2^16-1 并且 CAS state 从 0 到高16位 + 1成功
然后：
判断原读锁为0，那就是第一个读线程进入了
设置 firstReader 和 firstReaderHoldCountfirstReader = 1

判断原读锁不为0，那就是2种可能

第一个读线程重入
后续其他读线程进入了

第一个读线程重入，firstReaderHoldCountfirstReader ++
其他读线程进入了
如果当前线程HoldCounter为空或者当前缓存的HoldCounter属于当前线程：那就去创建一个ThreadLocal<HoldCounter>
如果已经有HoldCounter但是值为0 则把HoldCounter塞入当前线程
然后就是HoldCounter种的count++

总结来说
如果时读锁的第一个线程，则重入次数维护在lock对象中的firstReaderHoldCount字段中
如果非第一个线程则维护在ThreadLocal中

最后的return fullTryAcquireShared(current);
则是CAS失败或者进入Block
只要来说就是设置一个自旋，并且每种失败场景分开处理
CAS失败、block则自选
超限则抛出异常