读写锁——ReentrantReadWriteLock原理详解

1.读写锁ReentrantReadWriteLock的原理

解决线程安全问题使用ReentrantLock就可以了，但是ReentrantLock是独占锁，某一时刻只有一个线程可以获取该锁，而实际中会有写少读多的场景，显然ReentrantLock满足不了这个需求，所以ReentrantReadWriteLock应运而生。ReentrantReadWriteLock采用读写分离的策略，允许多个线程可以同时获取读锁。

1.1 ReentrantReadWriteLock类图结构

由类图可知，读写锁内部维护了一个ReadLock和一个WriteLock,他们依赖Sync实现具体功能，而Sync继承自AQS，并且提供了公平和非公平的实现。

1.2 源码解读

我们先看下ReentrantReadWriteLock类的整体结构

public class ReentrantReadWriteLock implements ReadWriteLock, java.io.Serializable {
   
    private final ReentrantReadWriteLock.ReadLock readerLock;//读锁?
 
    private final ReentrantReadWriteLock.WriteLock writerLock;//写锁?
    
    final Sync sync;

    public ReentrantReadWriteLock() {//使用默认（非公平）的排序属性创建一个新的ReentrantReadWriteLock
        this(false);
    }
    public ReentrantReadWriteLock(boolean fair) {//使用给定的公平策略创建一个新的ReentrantReadWriteLock
        sync = fair ? new FairSync() : new NonfairSync();
        readerLock = new ReadLock(this);
        writerLock = new WriteLock(this);
    }

    public ReentrantReadWriteLock.WriteLock writeLock() { return writerLock; }//返回用于写入操作的锁?

    public ReentrantReadWriteLock.ReadLock  readLock()  { return readerLock; }//返回用于读取操作的锁?

    abstract static class Sync extends AbstractQueuedSynchronizer { ....

    static final class NonfairSync extends Sync {....}//非公平策略

    static final class FairSync extends Sync {.... }//公平策略

    public static class ReadLock implements Lock, java.io.Serializable {}//读锁?
       ....
    public static class WriteLock implements Lock, java.io.Serializable {}//写锁?
       ....

}

1.2.1 ReentrantReadWriteLock类的继承关系

public class ReentrantReadWriteLock implements ReadWriteLock, java.io.Serializable {}

public interface ReadWriteLock {
  
    Lock readLock();

    Lock writeLock();
}

说明：可以看到ReentrantReadWriteLock实现了ReadWriteLock接口，ReadWriteLock接口规范了读写锁方法，具体操作由子类去实现，同时还实现了Serializable接口，表示可以进行序列化操作。

1.2.2 Sync类的继承关系

abstract static class Sync extends AbstractQueuedSynchronizer {}

说明：Sync抽象类继承自AQS抽象类，Sync类提供了对ReentrantReadWriteLock的支持。

1.2.2.1 Sync类的内部类

Sync类的内部类存在两个，分别为HoldCounter和ThreadLocalHoldCounter，其中HoldCounter主要与读锁配套使用，HoldCounter源码如下

static final class HoldCounter {//计数器
    int count = 0;//计数
    // Use id, not reference, to avoid garbage retention
    final long tid = getThreadId(Thread.currentThread());//线程id
}

说明：HoldCounter有两个属性，count和tid，其中count表示某个读线程重入次数，tid表示该线程的tid字段的值，该字段可以用来唯一标识一个线程。

ThreadLocalHoldCounter源码如下：

static final class ThreadLocalHoldCounter
    extends ThreadLocal<HoldCounter> {//本地线程计数器
    public HoldCounter initialValue() {//重写初始化方法，在没有进行set的情况下，获取的都是该HoldCounter的值
        return new HoldCounter();
    }
}

说明：ThreadLocalHoldCounter重写了ThreadLocal的initialValue方法，ThreadLocal类可以将线程与对象相关联。在没用进行set的情况下，get到的均是initialValue方法里面生成的那个HoldCounter对象。

1.2.2.2 Sync类的属性

abstract static class Sync extends AbstractQueuedSynchronizer {
    //版本序号
    private static final long serialVersionUID = 6317671515068378041L;
    //高16位为读锁?，低16位为写锁?
    static final int SHARED_SHIFT   = 16;
    //读锁?单位
    static final int SHARED_UNIT    = (1 << SHARED_SHIFT);
    //读锁?的最大数量
    static final int MAX_COUNT      = (1 << SHARED_SHIFT) - 1;
    //写锁?的最大数量
    static final int EXCLUSIVE_MASK = (1 << SHARED_SHIFT) - 1;
    //本地线程计数器
    private transient ThreadLocalHoldCounter readHolds;
    //缓存计数器
    private transient HoldCounter cachedHoldCounter;
    //第一个读线程
    private transient Thread firstReader = null;
    //第一个读线程的计数
    private transient int firstReaderHoldCount;

说明：该属性中包括了读锁，写锁线程的最大量。本地线程计数器等。

1.2.2.3 Sync类的构造函数

//构造函数
Sync() {
    //本地线程计数器
    readHolds = new ThreadLocalHoldCounter();
    //设置AQS的状态
    setState(getState()); // ensures visibility of readHolds
}

说明：在Sync的构造函数中设置了本地线程计数器和AQS的状态state。

1.3 读写锁的状态设计

读写锁需要在同步状态（一个整形变量）上维护多个读线程和一个写线程的状态。

读写锁对于同步状态的实现是在一个整形变量上通过“按位切割使用”：将变量切割成两部分，高16位表示读状态，也就是获取到读锁的次数，低16位表示获取到写线程的可重入次数。

http://static.open-open.com/lib/uploadImg/20151031/20151031223319_397.png

假设当前同步状态值为S，get和set的操作如下： （1）获取写状态： S&0x0000FFFF:将高16位全部抹去 （2）获取读状态： S>>>16:无符号补0，右移16位 （3）写状态加1： S+1 （4）读状态加1： 　　S+（1<<16）即S + 0x00010000 在代码层的判断中，如果S不等于0，当写状态（S&0x0000FFFF），而读状态（S>>>16）大于0，则表示该读写锁的读锁已被获取。

1.4 写锁的获取与释放

看下WriteLock类中的lock和unlock方法：

public void lock() { sync.acquire(1); } public void unlock() { sync.release(1); }

说明：可以看到就是调用独占式同步状态的获取与释放，因此真实的实现就是Sync的tryAcquire和tryRelease。

1.4.1 写锁的获取，看下tryAcquire:

protected final boolean tryAcquire(int acquires) {
    //当前线程
    Thread current = Thread.currentThread();
    //获取状态
    int c = getState();
    //写线程数量（即获取独占锁的重入数）
    int w = exclusiveCount(c);
    //c!=0说明读锁或者写锁已经被某线程获取
    if (c != 0) {
        //w=0说明已经有线程获取了读锁返回false，w!=0并且当前线程不是写锁的拥有者，则返回false
        if (w == 0 || current != getExclusiveOwnerThread())
            return false;
        //说明当前线程获取了写锁，判断可重入次数（最大次数65535）
        if (w + exclusiveCount(acquires) > MAX_COUNT)
            throw new Error("Maximum lock count exceeded");
        //次数当前线程已经持有写锁，设置可重入次数
        setState(c + acquires);
        return true;
    }
    //到这里说明此时c=0，读锁和写锁都没有被获取，writerShouldBlock表示是否阻塞
    if (writerShouldBlock() ||
        !compareAndSetState(c, c + acquires))
        return false;
    //设置锁位当前线程所持有
    setExclusiveOwnerThread(current);
    return true;
}

其中exclusiveCount方法表示占有写锁的线程数量，源码如下：

static int exclusiveCount(int c) { return c & EXCLUSIVE_MASK; }

static final int EXCLUSIVE_MASK = (1 << SHARED_SHIFT) - 1;

static final int SHARED_SHIFT   = 16;

说明：直接将状态state和（2^16 - 1）做与运算，其等效于将state模上2^16。写锁数量由state的低十六位表示。

writerShouldBlock（）判断是否需要阻塞，公平和非公平方式实现不同。

非公平实现方式：

在非公平策略方式下总是不会被阻塞

final boolean writerShouldBlock() {
    return false; // writers can always barge
}

公平实现方式，在公平策略下会进行判断，判断同步队列中是否有等待时间更长的线程，若存在，则需要被阻塞，否则无需阻塞。

final boolean writerShouldBlock() {
    return hasQueuedPredecessors();
}

public final boolean hasQueuedPredecessors() {
    Node t = tail; // Read fields in reverse initialization order
    Node h = head;
    Node s;
    return h != t &&
        ((s = h.next) == null || s.thread != Thread.currentThread());
}

说明：如果不需要阻塞，则CAS更新同步状态，若CAS成功则返回true，失败则说明当前锁?被别的线程抢去了，返回false，如果需要阻塞也返回false。

成功获取写锁?后，将当前线程设置位占有写锁的线程，返回true。

方法流程图如下：

1.4.2 写锁的释放，tryRelease方法：

protected final boolean tryRelease(int releases) {
    //若锁的持有则不是当前线程，抛出异常
    if (!isHeldExclusively())
        throw new IllegalMonitorStateException();
     //写锁的新线程数
    int nextc = getState() - releases;
    //如果独占模式重入数为0了，说明独占模式被释放
    boolean free = exclusiveCount(nextc) == 0;
    if (free)
        //若写锁的新线程数为0，则将锁的持有则设置为null
        setExclusiveOwnerThread(null);
     //设置写锁的新线程数，不管独占模式是否被释放，更新独占重入数
    setState(nextc);
    return free;
}

说明： 写锁的释放过程还是相对而言比较简单的：首先查看当前线程是否为写锁的持有者，如果不是抛出异常。然后检查释放后写锁的线程数是否为0，如果为0则表示写锁空闲了，

释放锁资源将锁的持有线程设置为null，否则释放仅仅只是一次重入锁而已，并不能将写锁的线程清空。 此方法用于释放写锁资源，首先会判断该线程是否为独占线程，若不为独占线程，

则抛出异常，否则，计算释放资源后的写锁的数量，若为0，表示成功释放，资源不将被占用，否则，表示资源还被占用。其方法流程图如下。

1.5 读锁的获取与释放

类似于写锁，读锁的lock和unlock的实际实现对应Sync的 tryAcquireShared 和 tryReleaseShared方法。

1.5.1读锁的获取，看下tryAcquireShared方法

protected final int tryAcquireShared(int unused) {
   //获取当前线程
    Thread current = Thread.currentThread();
    //获取状态
    int c = getState();
    //如果写锁线程数！=0，且独占锁不是当前线程则返回失败，因为存在锁?降级
    if (exclusiveCount(c) != 0 &&
        getExclusiveOwnerThread() != current)
        return -1;
    //读锁数量
    int r = sharedCount(c);
    if (!readerShouldBlock() &&//读锁是否需要等待（公平锁原则）
        r < MAX_COUNT &&//持有线程小于最大数（65535）
        compareAndSetState(c, c + SHARED_UNIT)) {//设置读取锁状态
        //r=0，表示第一个读锁线程，第一个读锁firstReader是不会加入到readHolds中
        if (r == 0) {
            firstReader = current;//设置第一个读线程
            firstReaderHoldCount = 1;//读线程占用的资源数为1
        } else if (firstReader == current) {//当前线程为第一个读线程，表示第一个读锁线程重入
            firstReaderHoldCount++;//占用资源数+1
        } else {//读锁数量不为0，并且不为当前线程
            HoldCounter rh = cachedHoldCounter;//获取计数器
            if (rh == null || rh.tid != getThreadId(current))//计数器为空，或者计数器的tid不为当前正在运行的线程的tid
                cachedHoldCounter = rh = readHolds.get();//获取当前线程的计数器
            else if (rh.count == 0)//计数器为0
                readHolds.set(rh);//加入到readHolds中
            rh.count++;//计数器+1
        }
        return 1;
    }
    return fullTryAcquireShared(current);
}

其中sharedCount方法表示占有读锁的线程数量，源码如下：

static int sharedCount(int c)    { return c >>> SHARED_SHIFT; }

说明：直接将state右移16位，就可以得到读锁的线程数量，因为state的高16位表示读锁，对应的第十六位表示写锁数量。

读锁获取锁的过程比写锁稍微复杂些，首先判断写锁是否为0并且当前线程不占有独占锁，直接返回；否则，判断读线程是否需要被阻塞并且读锁数量是否小于最大值并且比较设置状态成功，

若当前没有读锁，则设置第一个读线程firstReader和firstReaderHoldCount；若当前线程线程为第一个读线程，则增加firstReaderHoldCount；否则，将设置当前线程对应的HoldCounter对象的值。流程图如下。

注意：更新成功后会在firstReaderHoldCount中或readHolds（ThreadLocal类型的）本地线程副本中记录当前线程的重入数，如果当前只有一个线程的话，还不需要动用ThreadLocal，直接往firstReaderHoldCount这个成员变量里存重入数， 当第二个线程来的时候，就要动用ThreadLocal变量的readHolds了，每个线程拥有自己的副本，用来保存自己的重入数。

1.5.1.1 fullTryAcquireShared方法：

final int fullTryAcquireShared(Thread current) { HoldCounter rh = null; for (;;) { // 无限循环 // 获取状态 int c = getState(); if (exclusiveCount(c) != 0) { // 写线程数量不为0 if (getExclusiveOwnerThread() != current) // 不为当前线程 return -1; } else if (readerShouldBlock()) { // 写线程数量为0并且读线程被阻塞 // Make sure we're not acquiring read lock reentrantly if (firstReader == current) { // 当前线程为第一个读线程 // assert firstReaderHoldCount > 0; } else { // 当前线程不为第一个读线程 if (rh == null) { // 计数器不为空 // rh = cachedHoldCounter; if (rh == null || rh.tid != getThreadId(current)) { // 计数器为空或者计数器的tid不为当前正在运行的线程的tid 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"); if (compareAndSetState(c, c + SHARED_UNIT)) { // 比较并且设置成功 if (sharedCount(c) == 0) { // 读线程数量为0 // 设置第一个读线程 firstReader = current; // firstReaderHoldCount = 1; } else if (firstReader == current) { firstReaderHoldCount++; } else { if (rh == null) rh = cachedHoldCounter; if (rh == null || rh.tid != getThreadId(current)) rh = readHolds.get(); else if (rh.count == 0) readHolds.set(rh); rh.count++; cachedHoldCounter = rh; // cache for release } return 1; } } }

说明：在tryAcquireShared函数中，如果下列三个条件不满足（读线程是否应该被阻塞、小于最大值、比较设置成功）则会进行fullTryAcquireShared函数中，它用来保证相关操作可以成功。其逻辑与tryAcquireShared逻辑类似，不再累赘。

1.5.2 读锁的释放，tryReleaseShared方法

protected final boolean tryReleaseShared(int unused) { // 获取当前线程 Thread current = Thread.currentThread(); if (firstReader == current) { // 当前线程为第一个读线程 // assert firstReaderHoldCount > 0; if (firstReaderHoldCount == 1) // 读线程占用的资源数为1 firstReader = null; else // 减少占用的资源 firstReaderHoldCount--; } else { // 当前线程不为第一个读线程 // 获取缓存的计数器 HoldCounter rh = cachedHoldCounter; if (rh == null || rh.tid != getThreadId(current)) // 计数器为空或者计数器的tid不为当前正在运行的线程的tid // 获取当前线程对应的计数器 rh = readHolds.get(); // 获取计数 int count = rh.count; if (count <= 1) { // 计数小于等于1 // 移除 readHolds.remove(); if (count <= 0) // 计数小于等于0，抛出异常 throw unmatchedUnlockException(); } // 减少计数 --rh.count; } for (;;) { // 无限循环 // 获取状态 int c = getState(); // 获取状态 int nextc = c - SHARED_UNIT; if (compareAndSetState(c, nextc)) // 比较并进行设置 // Releasing the read lock has no effect on readers, // but it may allow waiting writers to proceed if // both read and write locks are now free. return nextc == 0; } }

说明：此方法表示读锁线程释放锁。首先判断当前线程是否为第一个读线程firstReader，若是，则判断第一个读线程占有的资源数firstReaderHoldCount是否为1，若是，则设置第一个读线程firstReader为空，否则，将第一个读线程占有的资源数firstReaderHoldCount减1；若当前线程不是第一个读线程，那么首先会获取缓存计数器（上一个读锁线程对应的计数器 ），若计数器为空或者tid不等于当前线程的tid值，则获取当前线程的计数器，如果计数器的计数count小于等于1，则移除当前线程对应的计数器，如果计数器的计数count小于等于0，则抛出异常，之后再减少计数即可。无论何种情况，都会进入无限循环，该循环可以确保成功设置状态state。其流程图如下。

在读锁的获取、释放过程中，总是会有一个对象存在着，同时该对象在获取线程获取读锁是+1，释放读锁时-1，该对象就是HoldCounter。

要明白HoldCounter就要先明白读锁。前面提过读锁的内在实现机制就是共享锁，对于共享锁其实我们可以稍微的认为它不是一个锁的概念，它更加像一个计数器的概念。

一次共享锁操作就相当于一次计数器的操作，获取共享锁计数器+1，释放共享锁计数器-1。只有当线程获取共享锁后才能对共享锁进行释放、重入操作。所以HoldCounter的作用就是当前线程持有共享锁的数量，这个数量必须要与线程绑定在一起，否则操作其他线程锁就会抛出异常。

先看读锁获取锁的部分：

if (r == 0) {//r == 0，表示第一个读锁线程，第一个读锁firstRead是不会加入到readHolds中 firstReader = current; firstReaderHoldCount = 1; } else if (firstReader == current) {//第一个读锁线程重入 firstReaderHoldCount++; } else { //非firstReader计数 HoldCounter rh = cachedHoldCounter;//readHoldCounter缓存 //rh == null 或者 rh.tid != current.getId()，需要获取rh if (rh == null || rh.tid != current.getId()) cachedHoldCounter = rh = readHolds.get(); else if (rh.count == 0) readHolds.set(rh); //加入到readHolds中 rh.count++; //计数+1 }

这里为什么要搞一个firstRead、firstReaderHoldCount呢？而不是直接使用else那段代码？这是为了一个效率问题，firstReader是不会放入到readHolds中的，

如果读锁仅有一个的情况下就会避免查找readHolds。可能就看这个代码还不是很理解HoldCounter。我们先看firstReader、firstReaderHoldCount的定义：

private transient Thread firstReader = null;
private transient int firstReaderHoldCount;

这两个变量比较简单，一个表示线程，当然该线程是一个特殊的线程，一个是firstReader的重入计数。

HoldCounter的定义：

static final class HoldCounter {
    int count = 0;
    final long tid = Thread.currentThread().getId();
}

在HoldCounter中仅有count和tid两个变量，其中count代表着计数器，tid是线程的id。但是如果要将一个对象和线程绑定起来仅记录tid肯定不够的，而且HoldCounter根本不能起到绑定对象的作用，只是记录线程tid而已。

诚然，在java中，我们知道如果要将一个线程和对象绑定在一起只有ThreadLocal才能实现。所以如下：

static final class ThreadLocalHoldCounter
    extends ThreadLocal<HoldCounter> {
    public HoldCounter initialValue() {
        return new HoldCounter();
    }
}

ThreadLocalHoldCounter继承ThreadLocal，并且重写了initialValue方法。 故而，HoldCounter应该就是绑定线程上的一个计数器，而ThradLocalHoldCounter则是线程绑定的ThreadLocal。从上面我们可以看到ThreadLocal将HoldCounter绑定到当前线程上，同时HoldCounter也持有线程Id，这样在释放锁的时候才能知道ReadWriteLock里面缓存的上一个读取线程（cachedHoldCounter）是否是当前线程。这样做的好处是可以减少ThreadLocal.get()的次数，因为这也是一个耗时操作。需要说明的是这样HoldCounter绑定线程id而不绑定线程对象的原因是避免HoldCounter和ThreadLocal互相绑定而GC难以释放它们（尽管GC能够智能的发现这种引用而回收它们，但是这需要一定的代价），所以其实这样做只是为了帮助GC快速回收对象而已。

三、总结

通过上面的源码分析，我们可以发现一个现象：

在线程持有读锁的情况下，该线程不能取得写锁(因为获取写锁的时候，如果发现当前的读锁被占用，就马上获取失败，不管读锁是不是被当前线程持有)。 在线程持有写锁的情况下，该线程可以继续获取读锁（获取读锁时如果发现写锁被占用，只有写锁没有被当前线程占用的情况才会获取失败）。 仔细想想，这个设计是合理的：因为当线程获取读锁的时候，可能有其他线程同时也在持有读锁，因此不能把获取读锁的线程“升级”为写锁；而对于获得写锁的线程，它一定独占了读写锁，因此可以继续让它获取读锁，当它同时获取了写锁和读锁后，还可以先释放写锁继续持有读锁，这样一个写锁就“降级”为了读锁.

综上：

一个线程要想同时持有写锁和读锁，必须先获取写锁再获取读锁；写锁可以“降级”为读锁；读锁不能“升级”为写锁。

参考书籍

Java并发编程之美