读写锁ReentrantReadWriteLock的原理

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

ReentrantReadWriteLock采用读写分离的策略，允许多个线程可以同时获取读锁。

类图结构

为了了解ReentrantReadWriteLock的内部构造，我们先看下它的类图结构。

读写锁的内部维护了一个ReadLock和一个WriteLock,它们依赖Sync实现具体功能。

而Sync继承自AQS,并且也提供了公平和非公平的实现。

下面只介绍非公平的读写锁实现。我们知道AQS中只维护了一个state状态，而ReentrantReadWriteLock则需要维护读状态和写状态，一个state怎么表示写和读两种状态呢?ReentrantReadWriteLock巧妙地使用state的高16位表示读状态，也就是获取到读锁的次数；使用低16位表示获取到写锁的线程的可重入次数。

其中firstReader用来记录第一个获取到读锁的线程，firstReaderHoldCount则记录第一个获取到读锁的线程获取读锁的可重入次数。

cachedHoldCounter用来记录最后一个获取读锁的线程获取读锁的可重入次数。

readHolds是ThreadLocal变量，用来存放除去第一个获取读锁线程外的其他线程获取读锁的可重入次数。ThreadLocalHoldCounter继承了ThreadLocal,因而initialValue方法返回一个HoldCounter对象。

写锁的获取与释放

在ReentrantReadWriteLock中写锁使用WriteLock来实现。

void lock()

写锁是个独占锁，某时只有一个线程可以获取该锁。 如果当前没有线程获取到读锁和写锁，则当前线程可以获取到写锁然后返回。如果当前已经有线程获取到读锁和写锁，则当前请求写锁的线程会被阻塞挂起。

另外，写锁是可重入锁，如果当前线程已经获取了该锁，再次获取只是简单地把可重入次数加1后直接返回。

如以上代码所示，在lock()内部调用了AQS的acquire方法，其中tryAcquire是ReentrantReadWriteLock内部的sync类重写的。

在代码(1)中，如果当前AQS状态值不为0则说明当前已经有线程获取到了读锁或者写锁。

在代码(2)中，如果w==0说明状态值的低16位为0,而AQS状态值不为0,则说明高16位不为0,这暗示已经有线程获取了读锁，所以直接返回false。

而如果w!=0则说明当前已经有线程获取了该写锁，再看当前线程是不是该锁的持有者，如果不是则返回false。

执行到代码(3)说明当前线程之前已经获取到了该锁，所以判断该线程的可重入次数是不是超过了最大值，是则抛出异常，否则执行代码(4)增加当前线程的可重入次数，然后返回true.如果AQS的状态值等于0则说明目前没有线程获取到读锁和写锁，所以执行代码(5)。

其中，对于writerShouldBlock方法，非公平锁的实现为。

如果代码对于非公平锁来说总是返回false,则说明代码(5)抢占式执行CAS尝试获取写锁，获取成功则设置当前锁的持有者为当前线程并返回true,否则返回false。

公平锁的实现为

这里还是使用hasQueuedPredecessors来判断当前线程节点是否有前驱节点，如果有则当前线程放弃获取写锁的权限，直接返回false。

void locklnterruptibly()

类似于lock()方法，它的不同之处在于，它会对中断进行响应，也就是当其他线程调用了该线程的interrupt()方法中断了当前线程时，当前线程会抛出异常InterruptedException异常。

boolean tryLock()

尝试获取写锁，如果当前没有其他线程持有写锁或者读锁，则当前线程获取写锁会成功，然后返回true。

如果当前已经有其他线程持有写锁或者读锁则该方法直接返回false,且当前线程并不会被阻塞。如果当前线程已经持有了该写锁则简单增加AQS的状态值后直接返回true。

boolean tryLock(long timeout,TimeUnit unit)

与tryAcquire()的不同之处在于，多了超时时间参数，如果尝试获取写锁失败则会把当前线程挂起指定时间，待超时时间到后当前线程被激活，如果还是没有获取到写锁则返回false。

另外，该方法会对中断进行响应，也就是当其他线程调用了该线程的interrupt)方法中断了当前线程时，当前线程会抛出InterruptedException异常。

void unlock()

尝试释放锁，如果当前线程持有该锁，调用该方法会让该线程对该线程持有的AQS状态值减1,如果减去1后当前状态值为0则当前线程会释放该锁，否则仅仅减1而已。

如果当前线程没有持有该锁而调用了该方法则会抛出IlegalMonitorStateException异常，代码如下。

读锁的获取与释放

ReentrantReadWriteLock中的读锁是使用ReadLock来实现的。

void lock()

获取读锁，如果当前没有其他线程持有写锁，则当前线程可以获取读锁，AQS的状态值state的高16位的值会增加1,然后方法返回。否则如果其他一个线程持有写锁，则当前线程会被阻塞。

读锁的lock方法调用了AQS的acquireShared方法，在其内部调用了ReentrantReadWriteLock中的sync重写的tryAcquireShared方法。

如上代码首先获取了当前AQS的状态值，然后代码(2)查看是否有其他线程获取到了写锁，如果是则直接返回-1,而后调用AQS的doAcquireShared方法把当前线程放入AQS阻塞队列。

如果当前要获取读锁的线程已经持有了写锁，则也可以获取读锁。

但是需要注意，当一个线程先获取了写锁，然后获取了读锁处理事情完毕后，要记得把读锁和写锁都释放掉，不能只释放写锁。

否则执行代码(3),得到获取到的读锁的个数，到这里说明目前没有线程获取到写锁，但是可能有线程持有读锁，然后执行代码(4)。

其中非公平锁的readerShouldBlock实现代码如下。

如上代码的作用是，如果队列里面存在一个元素，则判断第一个元素是不是正在尝试获取写锁，如果不是，则当前线程判断当前获取读锁的线程是否达到了最大值。

最后执行CAS操作将AQS状态值的高16位值增加1。

代码(5)(6)记录第一个获取读锁的线程并统计该线程获取读锁的可重入数。

代码(7)使用cachedHoldCounter记录最后一个获取到读锁的线程和该线程获取读锁的可重入数，readHolds记录了当前线程获取读锁的可重入数。

如果readerShouldBlock返回true则说明有线程正在获取写锁，所以执行代码(8)。

fullTryAcquireShared的代码与tryAcquireShared类似，它们的不同之处在于，前者通过循环自旋获取。

void locklnterruptibly()

类似于lock()方法，不同之处在于，该方法会对中断进行响应，也就是当其他线程调用了该线程的interrupt()方法中断了当前线程时，当前线程会抛出InterruptedException异常。

boolean tryLock()

尝试获取读锁，如果当前没有其他线程持有写锁，则当前线程获取读锁会成功，然后返回true。

如果当前已经有其他线程持有写锁则该方法直接返回false,但当前线程并不会被阻塞。

如果当前线程已经持有了该读锁则简单增加AQS的状态值高16位后直接返回true。

boolean tryLock(long timeout,TimeUnit unit)

与tryLock()的不同之处在于，多了超时时间参数，如果尝试获取读锁失败则会把当前线程挂起指定时间，待超时时间到后当前线程被激活，如果此时还没有获取到读锁则返回false。

另外，该方法对中断响应，也就是当其他线程调用了该线程的interrupt()方法中断了当前线程时，当前线程会抛出InterruptedException异常。

void unlock()

如上代码具体释放锁的操作是委托给Sync类来做的，sync.releaseShared方法的代码如下：


如以上代码所示，在无限循环里面，首先获取当前AQS状态值并将其保存到变量c,然后变量c被减去一个读计数单位后使用CAS操作更新AQS状态值，如果更新成功则查看当前AQS状态值是否为0,为0则说明当前已经没有读线程占用读锁，则tryReleaseShared返回true。

然后会调用doReleaseShared方法释放一个由于获取写锁而被阻塞的线程，如果当前AQS状态值不为0,则说明当前还有其他线程持有了读锁，所以tryReleaseShared返回false。

如果tryReleaseShared中的CAS更新AQS状态值失败，则自旋重试直到成功。