概念
读写锁在同一时刻允许多个读线程访问,但是在写线程访问时,所有的读线程和其他写线程均被阻塞。读写锁维护着一对锁:一个读锁和一个写锁,通过分离读锁和写锁,提高了并发性能。
实现类
ReentrantReadWriteLock是ReadWriteLock接口的具体实现,其主要有以下特性:
特性 | 说明 |
---|---|
公平性选择 | 支持非公平(默认)和公平的获取方式,吞吐量还是非公平优于公平 |
重进入 | 该锁支持重进入,以读线程为例:读线程火获取了读锁之后,能够再次获取读锁。而写线程在获取了写锁之后能够再次获取写锁,同时也可以获取读锁 |
锁降级 | 遵循获取写锁、获取读锁再释放写锁的次序,写锁能够降级为读锁 |
使用示例
public class Cache {
private static Map<String, Object> map = new HashMap<>();
private static ReentrantReadWriteLock rrwl=new ReentrantReadWriteLock();
private static Lock r = rrwl.readLock();
private static Lock w = rrwl.writeLock();
public static final Object get(String key){
r.lock();
try {
return map.get(key);
} catch (Exception e) {
}finally {
r.unlock();
}
return null;
}
public static final void put(String key,Object value){
w.lock();
try {
map.put(key, value);
} catch (Exception e) {
}finally {
w.unlock();
}
}
public static final void clear(){
w.lock();
try {
map.clear();
} catch (Exception e) {
}finally {
w.unlock();
}
}
}
Cache组合一个非线程安全的HashMap作为缓存的实现,同时使用读写锁的读锁和写锁来保证线程安全。在get方法中需要获取读锁,在put和clear方法中需要获取写锁。
读写锁的实现分析
1.读写状态的设计
读写锁同样依赖自定义同步器来实现同步功能,需要在同步状态(一个整型变量)上维护多个读线程和一个写线程的状态,使得该状态的设计成为读写锁实现的关键。
读写锁将同步状态“安慰切割使用”,高16位表示读,低16位表示写。
读写锁通过位运算迅速确定读和写各自的状态。假设当前状态值为S,那么写状态为S&0x0000FFFF(将高16位全部抹去),读状态等于S>>>16(无符号补0右移16位)。当写状态增加1时,等于S+1,当读状态增加1时,等于S+(1<<16),也就是S+0x00010000。根据状态的划分能得出一个结论:S不等于0时,当写状态等于0时,则读状态(S>>>16)大于0,即读锁已被获取
2.写锁的获取与释放
写锁是一个支持重进入的排他锁。如果当前线程已经获取了写锁,则增加写状态。如果当前线程在获取写锁时,读锁已经被获取(读状态不为0)或者该线程不是已经获取写锁的线程,则当前线程进入等待状态。
protected final boolean tryAcquire(int acquires) {
// 1.如果读取计数非零或写入计数非零,并且所有者是不同的线程,则失败。
// 2.如果计数饱和,失败。 (这只有在count已经非零时才会发生。)
//3.否则,如果此线程是可重入获取或队列策略允许,则此线程适合锁定。 如果是,更新状态并设置所有者
Thread current = Thread.currentThread();//获取当前线程
int c = getState();//当前同步状态
int w = exclusiveCount(c);//获取写锁的重进入计数
if (c != 0) {//如果同步状态不为0
// 存在读锁或者当前获取线程不是已经获取写锁的线程
if (w == 0 || current != getExclusiveOwnerThread())
return false;
if (w + exclusiveCount(acquires) > MAX_COUNT)
throw new Error("Maximum lock count exceeded");
//重进入获取
setState(c + acquires);
return true;
}
if (writerShouldBlock() ||
!compareAndSetState(c, c + acquires))
return false;
setExclusiveOwnerThread(current);
return true;
}
上述代码,除了重入条件之外,还增加了一个读锁是否存在的判断。如果存在读锁,则写锁不能被获取,因为读写锁要确保写锁的操作对读锁可见。如果允许读锁在已被获取的情况下对写锁的获取,那么正在运行的其他读线程就无法感知到当前写线程的操作。因此,只有等待其他读线程都释放了读锁,写锁才能被当前线程获取,而写锁一旦获取,其他读写线程的后续访问都会被阻塞。
写锁的释放与ReentrantLock的释放过程基本相似,每次释放均减少写状态,当写状态等于0 时,表示写锁以及被完全释放,从而等待读写线程能够继续访问读写锁,同时前次写线程的修改为后续读写线程可见。
protected final boolean tryRelease(int releases) {
if (!isHeldExclusively())//检查当前线程是否写锁的所有者
throw new IllegalMonitorStateException();
int nextc = getState() - releases;//减少同步状态
boolean free = exclusiveCount(nextc) == 0;
if (free)//表示写状态等于0,即写锁被完全释放
setExclusiveOwnerThread(null);//设置写锁的所有者为null
setState(nextc);//设置写状态
return free;
}
3. 读锁的获取与释放
读锁是一个支持重进入的共享锁,它能够被多个线程同时获取,在没有其他写线程访问时,读锁总会被成功的获取,而所做的也只是线程安全的增加读状态。如果当前线程已经获取了读锁,则增加读状态。如果当前线程在获取读锁时,写锁已经被获取,那么读线程则进入等待状态。读状态是所有线程获取读锁次数的总和,而每个线程各自获取读锁的次数则只能选择保存在ThreadLocal中。
protected final int tryAcquireShared(int unused) {
Thread current = Thread.currentThread();//获取当前线程
int c = getState();//获取当前同步状态
if (exclusiveCount(c) != 0 &&
getExclusiveOwnerThread() != current)//表示读锁被获取了但是读锁不是被当前线程所持有
return -1;
int r = sharedCount(c);//获取读状态的值
if (!readerShouldBlock() &&
r < MAX_COUNT &&
compareAndSetState(c, c + SHARED_UNIT)) {//读线程没有被阻塞并且读计数小于最大允许值而且通过CAS操作成功设置读状态值
if (r == 0) {
firstReader = current;
firstReaderHoldCount = 1;
} else if (firstReader == current) {
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++;
}
return 1;
}
return fullTryAcquireShared(current);
}
final int fullTryAcquireShared(Thread current) {
HoldCounter rh = null;
for (;;) {
int c = getState();
if (exclusiveCount(c) != 0) {//表示读锁已被获取
if (getExclusiveOwnerThread() != current)//持有读锁的线程不是当前线程
return -1;
} else if (readerShouldBlock()) {//否则我们持有排他锁; 阻塞这里会导致死锁
// 确保我们没有可重入地获取读锁
if (firstReader == current) {
// 断言 firstReaderHoldCount > 0;
} else {
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");
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;
}
}
}
如果其他线程已经获取了写锁,则当前线程获取读锁失败,进入等待状态。如果当前线程获取了写锁或者写锁未被获取,则当前线程通过CAS增加读状态,成功获取读锁。
读锁的每次释放均减少读状态,减少值为(1<<16)
protected final boolean tryReleaseShared(int unused) {
Thread current = Thread.currentThread();
//irstReader是获取读取锁的第一个线程。 firstReaderHoldCount是firstReader的保持计数
if (firstReader == current) {//第一个读线程为当前线程时
// 断言 firstReaderHoldCount > 0;
if (firstReaderHoldCount == 1)//如果当前线程只获得一次读锁,那么释放时,则直接设置firstReader为null
firstReader = null;
else
firstReaderHoldCount--;//如果当前线程多次获取了读锁时,则减少读状态
} else {
//cachedHoldCounter是成功获取readLock的最后一个线程的保持计数。 这会在常见的情况下,保存在ThreadLocal中,并且从ThreadLocal中查找,下一个要释放的线程是最后一个要获取的线程。
HoldCounter rh = cachedHoldCounter;
if (rh == null || rh.tid != current.getId())
rh = readHolds.get();//readHolds当前线程持有的可重入读锁的数量
int count = rh.count;
if (count <= 1) {
readHolds.remove();
if (count <= 0)
throw unmatchedUnlockException();
}
--rh.count;
}
for (;;) {
int c = getState();
int nextc = c - SHARED_UNIT;
if (compareAndSetState(c, nextc))
释放读取锁对读取线程没有影响,但是如果读取和写入锁现在都是空闲的,则它可以允许等待写入线程继续
return nextc == 0;
}
}
4. 锁降级
锁降级指的是写锁降级为读锁。如果当前线程拥有写锁,然后将其释放,最后在获取读锁,这种分段完成的过程不能称之为锁降级。锁降级是指把同时持有写锁在获取到读锁,然后释放写锁的过程。
下面示例,当数据变更后,如果当前线程感知到数据变化,则进行数据的准备工作, 同时其他处理线程被阻塞,直到当前线程完成数据的准备工作。
public void procssData(){
readLock.lock();
if(isUpdate){
//必须先释放读锁
readLock.unlock();
//锁的降级必须先从写锁的获取开始
writeLock.lock();
try {
if(isUpdate){
//准备数据的流程(省略)
isUpdate=false;
}
} finally {
writeLock.unlock();
}
//锁降级完成,写锁降级为读锁
}
try {
//使用数据的流程(省略)
} finally {
readLock.unlock();
}
}
锁降级中读锁的获取是必要的。因为这样可以保证数据的可见性。如果当前线程不获取读锁而是直接释放写锁,假设此刻另一个线程T获取了写锁并修改了数据,那么当前线程无法感知到线程T的数据更新。如果当前线程获取了读锁,即遵循锁降级的步骤,则线程T会被阻塞,直到当前线程使用数据并释放读锁之后,线程T才能获取写锁进行数据更新。