要点解说
ReentrantLock在并发情况下只允许单个线程执行受保护的代码，而在大部分应用中都是读多写少，所以，如果使用ReentrantLock实现这种对共享数据的并发访问控制，将严重影响整体的性能。ReentrantReadWriteLock中提供的读取锁(ReadLock)可以实现并发访问下的多读，写入锁(WriteLock)可以实现每次只允许一个写操作。但是，在它的实现中，读、写是互斥的，即允许多读，但是不允许读写和写读同时发生。
实例演示
public class ReentrantReadWriteLockExample { private final ReentrantReadWriteLock readWriteLock = new ReentrantReadWriteLock(); private final Lock readLock = readWriteLock.readLock(); private final Lock writeLock = readWriteLock.writeLock(); public String get(String key) {
readLock.lock(); try { //省略读取代码
return "xxx";
} finally {
readLock.unlock();
}
} public void put(String key, String value) {
writeLock.lock(); try { //省略写入操作
} finally {
writeLock.unlock();
}
}
}方法解析
ReentrantReadWriteLock类中重要的还是在它内部实现的ReadLock类和WriteLock类的方法，本篇将重点分析它们俩提供的常用方法。
ReentrantReadWriteLock类
ReentrantReadWriteLock()：使用非公平策略创建一个ReentrantReadWriteLock对象；
ReentrantReadWriteLock(boolean fair)：根据指定的策略参数fair创建一个ReentrantReadWriteLock对象；
readLock()：返回用于读取操作的锁；
writeLock()： 返回用于写入操作的锁。
ReadLock类
ReentrantReadWriteLock.ReadLock(ReentrantReadWriteLock lock)：使用ReentrantReadWriteLock创建ReadLock对象；
lock()：获取读操作锁，如果写入锁没有被其它线程持有，则立即获取读取锁，否则当前线程阻塞直到获取读取锁；
unlock()：释放读操作锁，同时唤醒其它等待获取该锁的线程；
tryLock()：尝试获取读操作锁，如果写入锁没有被其它线程持有，则立即获取读取锁并返回true值，否则立即返回false值；
tryLock(long timeout, TimeUnit unit)：在指定的等待时间内尝试获取读操作锁，如果写入锁没有被其它线程持有，并且当前线程未被中断，则立即获取读取锁并返回true值，否则当前线程阻塞直到等待时间结束并返回false。
WriteLock类
ReentrantReadWriteLock.WriteLock(ReentrantReadWriteLock lock)：使用ReentrantReadWriteLock创建WriteLock对象；
lock()：当前线程获取写入锁时，如果其它线程既没有持有读取锁也没有持有写入锁，则可以获取写入锁并立即返回，并将写入锁持有计数设置为1；如果当前线程已经持有写入锁，则写入锁计数增加1，该方法立即返回；如果锁被其它线程持有，当前线程阻塞直到获取写入锁；
tryLock()：当前线程获取写入锁时，如果其它线程既没有持有读取锁也没有持有写入锁，则可以获取写入锁并立即返回，并将写入锁持有计数设置为1；如果当前线程已经持有写入锁，则写入锁计数增加1，该方法立即返回；如果锁被其它线程持有，立即返回false值；
tryLock(long timeout, TimeUnit unit)：在指定的等待时间内尝试获取写操作锁，如果其它线程既没有持有读取锁也没有持有写入锁，并且当前线程未被中断，则可以获取写入锁并立即返回，并将写入锁持有计数设置为1；如果当前线程已经持有此锁，则将持有计数加1，该方法将返回true值；否则当前线程阻塞直到等待时间结束并返回false；
unlock()：如果当前线程持有此锁，则将持有计数减1；如果持有计数等于0，则释放该锁，同时唤醒其它等待获取该锁的线程；如果当前线程不是此锁的持有者，则抛出 IllegalMonitorStateException。
源码解析
首先，看一下ReentrantReadWriteLock类有哪些属性和构造函数，具体代码如下。
public class ReentrantReadWriteLock
implements ReadWriteLock, java.io.Serializable { //ReadLock是ReentrantReadWriteLock中的静态内部类，它是读取锁的实现
private final ReentrantReadWriteLock.ReadLock readerLock; //WriteLock是ReentrantReadWriteLock中的静态内部类，它是写入锁的实现
private final ReentrantReadWriteLock.WriteLock writerLock; //Sync是ReentrantReadWriteLock中的静态内部类，它继承了AQS
//它是读写锁实现的重点，后面深入分析
final Sync sync; //默认使用非公平策略创建对象
public ReentrantReadWriteLock() { this(false);
} //根据指定策略参数创建对象
public ReentrantReadWriteLock(boolean fair) { //FairSync和NonfairSync都继承自Sync，它们主要提供了对读写是否需要被阻塞的检查方法
sync = fair ? new FairSync() : new NonfairSync();
readerLock = new ReadLock(this);
writerLock = new WriteLock(this);
}
}上面的代码中介绍到Sync类是重点代码，这里先看它的部分重点源码。
//继承了AQS
abstract static class Sync extends AbstractQueuedSynchronizer { //常量值
static final int SHARED_SHIFT = 16; //左移16位后，二进制值是10000000000000000，十进制值是65536
static final int SHARED_UNIT = (1 << SHARED_SHIFT); //左移16位后再减一，十进制值是65535
//这个常量值用于标识最多支持65535个递归写入锁或65535个读取锁
static final int MAX_COUNT = (1 << SHARED_SHIFT) - 1; //左移16位后再减一，二进制值是1111111111111111
static final int EXCLUSIVE_MASK = (1 << SHARED_SHIFT) - 1; //用于计算持有读取锁的线程数
static int sharedCount(int c) { //无符号右移动16位
//如果c是32位，无符号右移后，得到是高16位的值
return c >>> SHARED_SHIFT;
} //用于计算写入锁的重入次数
static int exclusiveCount(int c) { //如果c是32位，和1111111111111111做&运算，得到的低16位的值
return c & EXCLUSIVE_MASK;
} //用于每个线程持有读取锁的计数
static final class HoldCounter { //每个线程持有读取锁的计数
int count = 0; //当前持有读取锁的线程ID
//这里使用线程ID而没有使用引用，避免垃圾收集器保留，导致无法回收
final long tid = Thread.currentThread().getId();
} //通过ThreadLocal维护每个线程的HoldCounter
static final class ThreadLocalHoldCounter
extends ThreadLocal<HoldCounter> { //这里重写了ThreadLocal的initialValue方法
public HoldCounter initialValue() { return new HoldCounter();
}
} //当前线程持有的可重入读取锁的数量，仅在构造方法和readObject方法中被初始化
//当持有锁的数量为0时，移除此对象
private transient ThreadLocalHoldCounter readHolds; //成功获取读取锁的最近一个线程的计数
private transient HoldCounter cachedHoldCounter; //第一个获得读锁的线程
private transient Thread firstReader = null; //第一个获得读锁的线程持有读取锁的次数
private transient int firstReaderHoldCount;
Sync() { //构建每个线程的HoldCounter
readHolds = new ThreadLocalHoldCounter();
setState(getState()); // ensures visibility of readHolds
}
}Sync继承自AQS，Sync使用AQS中的state属性来代表锁的状态，这个state二进制值被设计成32位，其中高16位用作读取锁，低16位用作写入锁。所以，如果要计算持有读取锁的线程数，只要将state二进制值无符号右移动16位；如果要计算写入锁的重入次数，只要将state二进制值和1111111111111111做&运算。
ReadLock源码解析
下面开始分析读取锁ReadLock的实现原理，先看一下它的构造函数源码。
public static class ReadLock implements Lock, java.io.Serializable { private final Sync sync; //通过ReentrantReadWriteLock对象构建ReadLock
protected ReadLock(ReentrantReadWriteLock lock) { //在ReentrantReadWriteLock构造函数中会根据fair参数值选择FairSync或NonfairSync创建不同的对象
//所以，这里赋值给sync的可能是FairSync类的对象，也可能是NonfairSync类的对象
sync = lock.sync;
}
}FairSync和NonfairSync都继承自Sync，不同点是各自实现了对读写是否需要被阻塞的检查方法，这里不做深入分析。
对于lock()方法，如果写入锁没有被其它线程持有，则立即获取读取锁，否则当前线程阻塞直到获取读取锁，具体代码如下。
public void lock() { //调用AQS的acquireShared方法
sync.acquireShared(1);
} //AQS的acquireShared方法
public final void acquireShared(int arg) { //尝试获取读取锁
//tryAcquireShared方法返回值小于0，则获取失败
if (tryAcquireShared(arg) < 0) //AQS中的方法，用于排队尝试再次获取读取锁
doAcquireShared(arg);
} //ReentrantReadWriteLock类的tryAcquireShared方法
protected final int tryAcquireShared(int unused) { //获取当前线程
Thread current = Thread.currentThread(); //获取当前AQS中state值
int c = getState(); //使用exclusiveCount方法计算写入锁是否被持有
//如果exclusiveCount(c)结果不等于0，即写入锁被持有，并且持有写入锁的线程不是当前线程
if (exclusiveCount(c) != 0 &&
getExclusiveOwnerThread() != current) //返回-1
return -1; //使用sharedCount方法计算读取锁被持有的线程数
int r = sharedCount(c); //如果当先线程不需要被阻塞，并且持有读取锁的线程数没有超过最大值，并且使用CAS更新读取锁线程数量成功。
//readerShouldBlock方法用于判断当先线程是否需要被阻塞，它在FairSync和NonfairSync中的实现各不同，
//FairSync中实现是如果有其它线程在当前线程之前等待获取读取锁，则当前线程应该被阻塞并返回true，否则返回false；
//NonfairSync中实现是如果等待队列的第一个节点的线程等待获取写入锁，则当前线程应该被阻塞并返回true，否则返回false；
if (!readerShouldBlock() &&
r < MAX_COUNT &&
compareAndSetState(c, c + SHARED_UNIT)) { //如果读取锁被持有的线程数等于0
if (r == 0) { //则表示当前线程是第一个获取读取锁的
firstReader = current; //第一个获得读锁的线程持有读取锁的次数赋值为1
firstReaderHoldCount = 1;
} else if (firstReader == current) { //如果读取锁被持有的线程数不等于0，并且当前线程是第一个获取读取锁的
//则将持有读取锁的次数加1
firstReaderHoldCount++;
} else { //将当先线程持有读取锁的次数信息，放入线程本地变量中
HoldCounter rh = cachedHoldCounter; if (rh == null || rh.tid != current.getId())
cachedHoldCounter = rh = readHolds.get(); else if (rh.count == 0)
readHolds.set(rh);
rh.count++;
} //成功获取读取锁，返回1
return 1;
} //如果当前线程需要被阻塞，或持有读取锁的线程数超过最大值，或使用CAS更新读取锁线程数量失败
//通过自旋的方式解决这三种获取锁失败的情况
return fullTryAcquireShared(current);
} //ReentrantReadWriteLock类的fullTryAcquireShared方法
final int fullTryAcquireShared(Thread current) {
HoldCounter rh = null; //自旋
for (;;) { //当前锁状态值
int c = getState(); //如果写入锁被线程持有
if (exclusiveCount(c) != 0) { //并且写入锁的持有者不是当前线程，则返回-1，获取锁失败
if (getExclusiveOwnerThread() != current) return -1; // else we hold the exclusive lock; blocking here
// would cause deadlock.
//如果当前线程应该被阻塞
} else if (readerShouldBlock()) { // Make sure we're not acquiring read lock reentrantly
if (firstReader == current) { // assert firstReaderHoldCount > 0;
} else { //如果最近一次获取读取锁的线程或从上下文中获取的当前线程，持有的锁记录数等于0，
//则返回-1获取读取锁失败
if (rh == null) {
rh = cachedHoldCounter; if (rh == null || rh.tid != current.getId()) {
rh = readHolds.get(); if (rh.count == 0)
readHolds.remove();
}
} if (rh.count == 0) return -1;
}
} //如果持有读取锁的线程数等于最大值
if (sharedCount(c) == MAX_COUNT) throw new Error("Maximum lock count exceeded"); //如果使用CAS更新读取锁线程数量成功
if (compareAndSetState(c, c + SHARED_UNIT)) { //更新线程持有的锁次数
if (sharedCount(c) == 0) {
firstReader = current;
firstReaderHoldCount = 1;
} else if (firstReader == current) {
firstReaderHoldCount++;
} else { if (rh == null)
rh = cachedHoldCounter; if (rh == null || rh.tid != current.getId())
rh = readHolds.get(); else if (rh.count == 0)
readHolds.set(rh);
rh.count++;
cachedHoldCounter = rh; // cache for release
} return 1;
}
}
} //AQS中的doAcquireShared方法
private void doAcquireShared(int arg) { //根据当前线程创建一个共享模式的Node节点
//并把这个节点添加到等待队列的尾部
//AQS等待队列不熟悉的可以查看AQS深入解析的内容
//addWaiter方法的解析在其它篇幅已经分析过，这里不再深入分析
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) { //将前驱节点从等待队列中释放
//同时使用LockSupport.unpark方法唤醒前驱节点的后继节点中的线程
setHeadAndPropagate(node, r);
p.next = null; // help GC
if (interrupted)
selfInterrupt();
failed = false; return;
}
} //当前节点的前驱节点不是头结点，或不可以获取到锁
//shouldParkAfterFailedAcquire方法检查当前节点在获取锁失败后是否要被阻塞
//如果shouldParkAfterFailedAcquire方法执行结果是当前节点线程需要被阻塞，则执行parkAndCheckInterrupt方法阻塞当前线程
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
interrupted = true;
}
} finally { if (failed)
cancelAcquire(node);
}
}对于unlock()方法用于释放当前线程持有的读取锁，具体实现是先修改线程持有的读取锁的计数数量，然后通过自旋方式使用CAS修改锁状态高16位值为当前值减1，最后如果锁状态值等于0，即既没有写入锁，也没有读取锁被线程持有，则唤醒等待队列中等待获取锁的线程。具体源码分析如下。
//ReadLock类中的unlock方法
public void unlock() { //调用AQS中的releaseShared方法
sync.releaseShared(1);
} //AQS中的releaseShared方法
public final boolean releaseShared(int arg) { //这里调用了ReentrantReadWriteLock类的tryReleaseShared方法
//在这个方法中做了如下两件事：
//1.修改当前线程持有的读取锁的计数数量，计数数量减一
//2.修改锁状态值的高16位值(读取锁的值)，值减一，并且返回锁状态值是否等于0
if (tryReleaseShared(arg)) { //如果锁状态值等于0，即既没有写入锁，也没有读取锁被线程持有，执行doReleaseShared
//这里调用的是AQS中的doReleaseShared方法
doReleaseShared(); return true;
} return false;
} //ReentrantReadWriteLock类的tryReleaseShared方法
protected final boolean tryReleaseShared(int unused) {
Thread current = Thread.currentThread(); // 如果当前线程是第一个获取读取锁的线程
if (firstReader == current) { // 第一个获取读取锁的线程持有的读取锁的次数等于1
if (firstReaderHoldCount == 1) //要释放掉，所以置为null
firstReader = null; else
//否则，直接将线程持有的读取锁的次数减一
firstReaderHoldCount--;
} else{ //最近一次成功获取读取锁的线程和计数信息
HoldCounter rh = cachedHoldCounter; //如果取得的rh==null或它里面记录的不是当前线程的信息，
//则直接从上下文中获取当前线程的线程和计数信息
if (rh == null || rh.tid != current.getId())
rh = readHolds.get(); //当前线程持有读取锁的次数
int count = rh.count; //如果次数小于等于1，则从上下文中移除
if (count <= 1) {
readHolds.remove(); //如果次数小于等于0，则抛出异常，因为不符合是否锁的条件
if (count <= 0) throw unmatchedUnlockException();
} //将线程持有读取锁的次数减一
--rh.count;
} for (;;) { //获取锁状态值
int c = getState(); //这里转换成二进制计算，是将高16位减1
int nextc = c - SHARED_UNIT; if (compareAndSetState(c, nextc)) //如果既没有写入锁，也没有读取锁被线程持有，则返回true
return nextc == 0;
}
} //AQS中的doReleaseShared方法
private void doReleaseShared() { for (;;) { //等待队列的头结点
Node h = head; //如果头结点存在，并且有后继节点
if (h != null && h != tail) { int ws = h.waitStatus; //如果head指向的节点状态可以被唤醒
if (ws == Node.SIGNAL) { //使用CAS更新节点的等待状态
//如果不成功则继续循环，通过循环来保证更新成功
if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0)) continue; // loop to recheck cases
//更新成功，则唤醒节点对应的线程
unparkSuccessor(h);
} else if (ws == 0 &&
!compareAndSetWaitStatus(h, 0, Node.PROPAGATE)) continue; // loop on failed CAS
} if (h == head) // loop if head changed
break;
}
}WriteLock源码解析
下面开始分析写入锁WriteLock的实现原理，先看一下它的构造函数源码。
public static class WriteLock implements Lock, java.io.Serializable { private final Sync sync; //通过ReentrantReadWriteLock对象构建
protected WriteLock(ReentrantReadWriteLock lock) { //在ReentrantReadWriteLock构造函数中会根据fair参数值选择FairSync或NonfairSync创建不同的对象
//所以，这里赋值给sync的可能是FairSync类的对象，也可能是NonfairSync类的对象
sync = lock.sync;
}
}FairSync和NonfairSync都继承自Sync，不同点是各自实现了对读写是否需要被阻塞的检查方法，这里不做深入分析。
对于lock()方法，如果其它线程既没有持有读取锁也没有持有写入锁，则可以获取写入锁并立即返回，并将写入锁持有计数设置为1；如果当前线程已经持有写入锁，则写入锁计数增加1，该方法立即返回；如果锁被其它线程持有，当前线程阻塞直到获取写入锁。具体代码如下。
public void lock() { //调用AQS中的acquire方法
sync.acquire(1);
} //AQS中的acquire方法
public final void acquire(int arg) { //调用Sync类中重写的tryAcquire方法尝试获取写入锁；
//如果tryAcquire方法返回true，则获取成功；
//如果tryAcquire方法返回false，则获取失败，继续执行acquireQueued方法；
//addWaiter方法根据当前线程创建一个独占模式的Node节点
//并把这个节点添加到等待队列的尾部
//AQS等待队列不熟悉的可以查看AQS深入解析的内容
//addWaiter方法的解析在其它篇幅已经分析过，这里不再深入分析
//acquireQueued方法通过排队方式再次获取写入锁
//acquireQueued方法中如果可以获取锁，则返回false，否则返回true
//如果tryAcquire方法获取不到锁返回false，且acquireQueued方法也获取不到锁返回true
//则中断当前线程
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg)) //中断当前线程
selfInterrupt();
} //Sync类中的tryAcquire
protected final boolean tryAcquire(int acquires) { //获取当前线程
Thread current = Thread.currentThread(); //获取当前锁的状态值
int c = getState(); //使用exclusiveCount方法计算写入锁的重入次数
int w = exclusiveCount(c); //如果锁状态不等于0，则表示存在写入锁或读取锁
if (c != 0) { // w等于0表示不存在写入锁，那就存在读取锁
// 或者存在写入锁，但是不是当前线程持有的，则返回false获取失败
if (w == 0 || current != getExclusiveOwnerThread()) return false; // 如果当前线程已经持有写入锁，并且即将重入的次数超过最大65535次，则抛出Error
if (w + exclusiveCount(acquires) > MAX_COUNT) throw new Error("Maximum lock count exceeded"); // 如果重入次数不超限，则修改锁状态值
setState(c + acquires); //返回获取成功
return true;
} //如果锁状态值等于0，则表示不存在写入锁，也不存在读取锁；
//通过writerShouldBlock方法判断当前线程是否需要被阻塞；
//writerShouldBlock在FairSync类和NonfairSync类中有不同的实现；
//FairSync中实现是如果有其它线程在当前线程之前等待获取写入锁，则当前线程应该被阻塞并返回true，否则返回false；
//NonfairSync中实现是直接返回false，也就是不排队，竞争获取；
//如果writerShouldBlock方法结果是true，则表示当前线程需要被阻塞，此时直接返回false获取失败；
//如果writerShouldBlock方法结果是false，将使用CAS修改锁状态值，修改失败则返回false获取失败，
//修改成功则设置当前线程持有写入锁。
if (writerShouldBlock() ||
!compareAndSetState(c, c + acquires)) return false; //设置当前线程持有写入锁
setExclusiveOwnerThread(current); return true;
} //addWaiter方法继承自AQS
//将当前线程封装成Node节点，并将这个Node节点插入到同步等待队列的尾部
private Node addWaiter(Node mode) {
Node node = new Node(Thread.currentThread(), mode); // Try the fast path of enq; backup to full enq on failure
Node pred = tail; if (pred != null) {
node.prev = pred; if (compareAndSetTail(pred, node)) {
pred.next = node; return node;
}
}
enq(node); return node;
} //acquireQueued方法继承自AQS
final boolean acquireQueued(final Node node, int arg) { boolean failed = true; try { boolean interrupted = false; for (;;) { //获取当前节点的前驱节点
final Node p = node.predecessor(); //如果当前节点的前驱节点是头结点，并且可以获取到锁，跳出循环并返回false
if (p == head && tryAcquire(arg)) {
setHead(node);
p.next = null; // help GC
failed = false; return interrupted;
} //当前节点的前驱节点不是头结点，或不可以获取到锁
//shouldParkAfterFailedAcquire方法检查当前节点在获取锁失败后是否要被阻塞
//如果shouldParkAfterFailedAcquire方法执行结果是当前节点线程需要被阻塞，
//则执行parkAndCheckInterrupt方法阻塞当前线程
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
interrupted = true;
}
} finally { if (failed)
cancelAcquire(node);
}
} //parkAndCheckInterrupt方法继承自AQS，用于阻塞当前线程
private final boolean parkAndCheckInterrupt() { //阻塞当前线程，当前线程执行到这里即被挂起，等待被唤醒
//当当前节点的线程被唤醒的时候，会继续尝试获取锁
LockSupport.park(this); return Thread.interrupted();
}对于unlock()方法用于释放当前线程持有的写入锁，如果当前线程持有此锁，则将持有计数减1；如果持有计数等于0，则释放该锁，同时唤醒其它等待获取该锁的线程；如果当前线程不是此锁的持有者，则抛出IllegalMonitorStateException。具体源码分析如下。
public void unlock() { //调用AQS中的release方法
sync.release(1);
} //AQS中的release方法释放持有的写入锁
public final boolean release(int arg) { //通过tryRelease方法修改写入锁状态值，
//并返回是否还存在写入锁被持有；
if (tryRelease(arg)) { //如果不存在写入锁被持有，
//则唤醒等待队列等待获取锁的线程
Node h = head; if (h != null && h.waitStatus != 0)
unparkSuccessor(h); return true;
} //如果存在写入锁被持有，则直接返回false
return false;
} //Sync中重写的tryRelease方法
protected final boolean tryRelease(int releases) { //如果当前线程没有持有写入锁，则抛出IllegalMonitorStateException
if (!isHeldExclusively()) throw new IllegalMonitorStateException(); //计算锁释放后的锁状态值
int nextc = getState() - releases; //使用exclusiveCount方法计算写入锁的值
//判断是否还存在写入锁被持有
boolean free = exclusiveCount(nextc) == 0; //如果不存在写入锁被线程持有，则设置写入锁持有线程为null
if (free)
setExclusiveOwnerThread(null); //修改锁状态值
setState(nextc); //返回是否还存在写入锁被持有
return free;
} protected final boolean isHeldExclusively() { //判断当前线程是否持有写入锁
//若持有，则返回true，否则返回false
return getExclusiveOwnerThread() == Thread.currentThread();
}
作者：JavaQ
链接：https://www.jianshu.com/p/bd4a8dc540ca
点击查看更多内容
为 TA 点赞
评论
共同学习，写下你的评论
评论加载中...
作者其他优质文章
正在加载中
相关文章推荐
感谢您的支持，我会继续努力的～
扫码打赏，你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫，即可进行扫码打赏哦