有了CopyOnWrite为何又要有ReadWriteLock?

2021-04-27  本文已影响0人  java金融

引言

前文我们有介绍《看了CopyOnWriteArrayList后自己实现了一个CopyOnWriteHashMap》 关于CopyOnWrite容器的,但是它也有一些缺点:

什么是读写锁

根据名字我们也可以猜个大概,就是有两把锁,分别是读锁和写锁。读锁在同一时刻可以允许多个读线程获取,但是在写线程访问的时候,所有的读线程和其他写线程都会被阻塞。写锁同一时刻只能有一个写线程获取成功,其他都会被阻塞。读写锁实际维护了两把锁,一个读锁和一个写锁,通过读锁和写锁进行区分,在读多写少的情况下并发性比独占锁有了很大的提升。在java里面对读写锁的实现就是ReentrantReadWriteLock,它有以下特性:

  • 公平性选择:支持非公平性(默认)和公平的锁获取方式,吞吐量还是非公平优于公平;

  • 重入性:支持重入,读锁获取后能再次获取,写锁获取之后能够再次获取写锁,同时也能够获取读锁;

  • 锁降级:遵循获取写锁,获取读锁再释放写锁的次序,写锁能够降级成为读锁

ReentrantReadWriteLock 的使用

我们先从官网来个事例https://docs.oracle.com/javase/8/docs/api/java/util/concurrent/locks/ReentrantReadWriteLock.html,看看它是如何使用的

 `class RWDictionary {`
 `private final Map<String, Data> m = new TreeMap<String, Data>();`
 `private final ReentrantReadWriteLock rwl = new ReentrantReadWriteLock();`
 `private final Lock r = rwl.readLock();`
 `private final Lock w = rwl.writeLock();`
 `public Data get(String key) {`
 `r.lock();`
 `try { return m.get(key); }`
 `finally { r.unlock(); }`
 `}`
 `public String[] allKeys() {`
 `r.lock();`
 `try { return m.keySet().toArray(); }`
 `finally { r.unlock(); }`
 `}`
 `public Data put(String key, Data value) {`
 `w.lock();`
 `try { return m.put(key, value); }`
 `finally { w.unlock(); }`
 `}`
 `public void clear() {`
 `w.lock();`
 `try { m.clear(); }`
 `finally { w.unlock(); }`
 `}`
 `}`

这个使用起来还是非常简单明了的,跟ReentrantLock的用法基本一致,写的时候获取写锁,写完了释放写锁,读的时候获取读锁,读完了就释放读写。

读写锁的实现分析

我们知道ReentrantLock是通过state来控制锁的状态,以及前面所介绍的《Java高并发编程基础三大利器之Semaphore》《Java高并发编程基础三大利器之CountDownLatch》《Java高并发编程基础三大利器之CyclicBarrier》 都是通过state来进行实现的那ReentrantReadWriteLock毋庸置疑肯定也是通过AQSstate来实现的,不过state是一个int值它是如何来读锁和写锁的。

读写锁状态的实现分析

如果我们有看过线程池的源码,我们知道线程池的状态和线程数是通过一个int类型原子变量(高3位保存运行状态,低29位保存线程数)来控制的。同样的ReentrantReadWriteLock也是通过一个state的高16位和低16位来分别控制读的状态和写状态。

image

下面我们就来看看它是如何通过一个字段来实现读写分离的,

 `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;`
 `/** Returns the number of shared holds represented in count  */`
 `static int sharedCount(int c)    { return c >>> SHARED_SHIFT; }`
 `/** Returns the number of exclusive holds represented in count  */`
 `static int exclusiveCount(int c) { return c & EXCLUSIVE_MASK; }`

源码分析

基于jdk1.8 既然ReentrantReadWriteLock也是基于AQS来实现的,那么它肯定是重写了AQS的获取锁的方法,那我们就直接去ReentrantReadWriteLock这个类里面看看lock的地方我们先看看获取读锁的地方

 `protected final boolean tryAcquire(int acquires) {`
 `/*`
 `* Walkthrough:`
 `* 1. If read count nonzero or write count nonzero`
 `*    and owner is a different thread, fail.`
 `* 2. If count would saturate, fail. (This can only`
 `*    happen if count is already nonzero.)`
 `* 3. Otherwise, this thread is eligible for lock if`
 `*    it is either a reentrant acquire or`
 `*    queue policy allows it. If so, update state`
 `*    and set owner.`
 `*/`
 `Thread current = Thread.currentThread();`
 `// 获取写锁当前的同步状态`
 `int c = getState();`
 `// 写锁次数`
 `int w = exclusiveCount(c);`
 `if (c != 0) {`
 `// (Note: if c != 0 and w == 0 then shared count != 0)`
 `// 当前状态不为0,但是写锁为0 就说明读锁不为0`
 `// 当读锁已被读线程获取或者当前线程不是已经获取写锁的线程的话获取写锁失败`
 `if (w == 0 || current != getExclusiveOwnerThread())`
 `return false;`
 `if (w + exclusiveCount(acquires) > MAX_COUNT)`
 `throw new Error("Maximum lock count exceeded");`
 `// Reentrant acquire 获取到写锁`
 `setState(c + acquires);`
 `return true;`
 `}`
 `//writerShouldBlock 公平锁和非公平锁的判断`
 `if (writerShouldBlock() ||`
 `!compareAndSetState(c, c + acquires))`
 `return false;`
 `setExclusiveOwnerThread(current);`
 `return true;`
 `}`

写锁完了,接下来肯定就是读锁了由于读锁是共享锁,所以也应该重写了tryAcquireShared 这个就不贴代码了,和读锁差不多这个就不做分析了。其实把AQS弄明白了再来看这些基于AQS来实现的玩意还是比较容易的。

读写锁的升级与降级

前面我们有提到读写锁是可以降级的,但是没有说是否可以升级。我们先看看什么是锁降级和锁升级

主要是为了保证数据的可见性,如果当前线程不获取读锁而是直接释放写锁, 假设此刻另一个线程(记作线程T)获取了写锁并修改了数据,那么当前线程无法感知线程T的数据更新。如果当前线程获取读锁,即遵循锁降级的步骤,则线程T将会被阻塞,直到当前线程使用数据并释放读锁之后,线程T才能获取写锁进行数据更新。来源于《Java 并发编程的艺术》

思考题

结束

往期精选

****推荐👍** :Java高并发编程基础三大利器之CyclicBarrier**

推荐👍Java高并发编程基础三大利器之CountDownLatch

推荐👍Java高并发编程基础三大利器之Semaphore

推荐👍Java高并发编程基础之AQS

推荐👍可恶的爬虫直接把生产6台机器爬挂了!

上一篇 下一篇

猜你喜欢

热点阅读