深入理解ReentrantReadWriteLock源码

200 阅读10分钟

之前提到锁(如Mutex和ReentrantLock)基本都是排他锁,这些锁在同一时刻只允许一个线程进行访问,而读写锁在同一时刻可以允许多个读线程访问,但是在写线程访问时,所有的读线程和其他写线程均被阻塞。读写锁维护了一对锁,一个读锁和一个写锁,通过分离读锁和写锁,使得并发性相比一般的排他锁有了很大提升。

在没有读写锁支持的(Java 5之前)时候,如果需要完成上述工作就要使用Java的等待通知机制,就是当写操作开始时,所有晚于写操作的读操作均会进入等待状态,只有写操作完成并进行通知之后,所有等待的读操作才能继续执行(写操作之间依靠synchronized关键进行同步),这样做的目的是使读操作能读取到正确的数据,不会出现脏读。

改用读写锁实现上述功能,只需要在读操作时获取读锁,写操作时获取写锁即可。当写锁被获取到时,后续(非当前写操作线程)的读写操作都会被阻塞,写锁释放之后,所有操作继续执行,编程方式相对于使用等待通知机制的实现方式而言,变得简单明了。

ReentrantReadWriteLock的特性

ReadWriteLock仅定义了获取读锁和写锁的两个方法,即readLock()方法和writeLock()方法,而其实现——ReentrantReadWriteLock,除了接口方法之外,还提供了一些便于外界监控其内部工作状态的方法.

接下来分析ReentrantReadWriteLock的实现,主要包括:1.读写状态的设计、2.写锁的获取与释放、3.读锁的获取与释放以及锁降级:

读写锁同样依赖自定义同步器来实现同步功能,而读写状态就是其同步器的同步状态。回想ReentrantLock(www.jianshu.com/p/5d57573b0…)中自定义同步器的实现,同步状态表示锁被一个线程重复获取的次数,而读写锁的自定义同步器需要在同步状态(一个整型变量)上维护多个读线程和一个写线程的状态,使得该状态的设计成为读写锁实现的关键。

如果在一个整型变量上维护多种状态,就一定需要“按位切割使用”这个变量,读写锁将变量切分成了两个部分,高16位表示读,低16位表示写,划分方式如图5-8所示

当前同步状态表示一个线程已经获取了写锁,且重进入了两次,同时也连续获取了两次读锁。读写锁是如何迅速确定读和写各自的状态呢?答案是通过位运算。假设当前同步状态值为S,写状态等于S&0x0000FFFF(将高16位全部抹去),读状态等于S>>>16(无符号补0右移16位)。当写状态增加1时,等于S+1,当读状态增加1时,等于S+(1<<16),也就是S+0x00010000。

在搞清楚读写锁的获取和释放之前,我们需要对读写锁的整体设计有一个轮廓。

看类源码:

public interface ReadWriteLock {
    Lock readLock();
    Lock writeLock();
}

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

    private final ReentrantReadWriteLock.ReadLock readerLock;
    private final ReentrantReadWriteLock.WriteLock writerLock;
    final Sync sync;
}

实现了ReadWriteLock这个接口,把读锁和写锁的获取抽离出来,还有同样有一个继承了AQS的子类Aync的实现,这跟大多数锁的设计是一样的。 然后接着看下构造方法:

public ReentrantReadWriteLock() {
        //默认还是实现非公平调度策略。
        this(false);
    }

    public ReentrantReadWriteLock(boolean fair) {
        sync = fair ? new FairSync() : new NonfairSync();
        readerLock = new ReadLock(this);
        writerLock = new WriteLock(this);
    }

可以看出,默认还是实现非公平调度策略。初始化会各自创建读锁和写锁。

写锁加锁获取

写锁是一个支持重进入的排它锁。如果当前线程已经获取了写锁,则增加写状态。如果当前线程在获取写锁时,读锁已经被获取(读状态不为0)或者该线程不是已经获取写锁的线程,则当前线程进入等待状态,获取写锁的代码。

//执行的是WriteLock中的lock()方法
public void lock() {
            sync.acquire(1);
        }
//该方法调用的是对应的sync中的父类AQS的方法。而tryAcquire(int arg)则是已经在sync中重写了,如代码二,
//该方法是AQS专门提供给子类重写的。然后先来分析下代码二tryAcquire(int acquires) 
//在AQS代码中
public final void acquire(int arg) {
        if (!tryAcquire(arg) &&
            acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            selfInterrupt();
    }

//代码二,在ReentrantReadWriteLock中
protected final boolean tryAcquire(int acquires) {
          //1.获取当前线程
            Thread current = Thread.currentThread();
            //获取读写锁的同步状态
            int c = getState();
          // 2. 获取写锁获取的次数
          // exclusiveCount() 里面执行的是return c & EXCLUSIVE_MASK;  
        //看下面的标注二
            int w = exclusiveCount(c);   
            //如果读写状态不等于0
            if (c != 0) {
                // (Note: if c != 0 and w == 0 then shared count != 0)
                 // 3.1 当读锁已被读线程获取或者当前线程不是已经获取写锁的线程的话
                // 当前线程获取写锁失败
              //看下面标注一
                if (w == 0 || current != getExclusiveOwnerThread())
                    return false;
              //如果超过获取锁的次数的最大值,则抛异常
                if (w + exclusiveCount(acquires) > MAX_COUNT)
                    //超过最大锁计数
                    throw new Error("Maximum lock count exceeded");
                //3.2 当前线程获取写锁,支持可重复加锁
                setState(c + acquires);
                return true;
            }
            //写锁未被任何线程获取,当前线程可获取写锁
            if (writerShouldBlock() ||
                !compareAndSetState(c, c + acquires))
                return false;
            //写锁第一次获取成功,设置当前线程为写锁的线程,返回获取成功。
            setExclusiveOwnerThread(current);
            return true;
        }

标注一:如果存在读锁,则写锁不能被获取,原因在于:读写锁要确保写锁的操作对读锁可见,如果允许读锁在已被获取的情况下对写锁的获取,那么正在运行的其他读线程就无法感知到当前写线程的操作。因此,只有等待其他读线程都释放了读锁,写锁才能被当前线程获取,而写锁一旦被获取,则其他读写线程的后续访问均被阻塞。

标注二: 其中EXCLUSIVE_MASK为: static final int EXCLUSIVE_MASK = (1 << SHARED_SHIFT) - 1; EXCLUSIVE _MASK为1左移16位然后减1,即为0x0000FFFF。而exclusiveCount方法是将同步状态(state为int类型)与0x0000FFFF相与,即取同步状态的低16位。那么低16位代表什么呢?根据exclusiveCount方法的注释为独占式获取的次数即写锁被获取的次数,现在就可以得出来一个结论同步状态的低16位用来表示写锁的获取次数。所以该方法返回的是写状态数。

然后trylock的加锁实现也是差不多的。

public boolean tryLock( ) {
            return sync.tryWriteLock();
        }

final boolean tryWriteLock() {
            Thread current = Thread.currentThread();
            int c = getState();
            if (c != 0) {
                int w = exclusiveCount(c);
                if (w == 0 || current != getExclusiveOwnerThread())
                    return false;
                if (w == MAX_COUNT)
                    throw new Error("Maximum lock count exceeded");
            }
            if (!compareAndSetState(c, c + 1))
                return false;
            setExclusiveOwnerThread(current);
            return true;
        }

这里不做具体的说明的,不懂得看上面注释。

写锁获取相应中断的锁方法如下(该线程被中断时,会释放锁):

 public void lockInterruptibly() throws InterruptedException {
            sync.acquireInterruptibly(1);
        }

public final void acquireInterruptibly(int arg)
            throws InterruptedException {
        if (Thread.interrupted())
            throw new InterruptedException();
        if (!tryAcquire(arg))
            doAcquireInterruptibly(arg);
    }

这步如果看过AQS独占式获取同步状态的,基本也懂,实现逻辑是一样的。可以看这篇文章。(www.jianshu.com/p/e0066f934…)写锁超时获取响应中断的锁的实现也是一样的。

然后我们来研究下如何释放锁:

//在ReentrantReadWriteLock的内部中的WriteLock的重写了unLock方法,
//调用的是sync类的release,同样是重写了AQS的方法
public void unlock() {
            sync.release(1);
        }

//该方法在AQS类中
public final boolean release(int arg) {
        //尝试释放锁成功,如果当前头节点不等于null,并且等待状态不等于0,则唤醒后继节点unparkSuccessor()
        if (tryRelease(arg)) {
            Node h = head;
            if (h != null && h.waitStatus != 0)
                unparkSuccessor(h);
            return true;
        }
        return false;
    }
//在ReentrantReadWriteLock的内部中的Sync重写了
protected final boolean tryRelease(int releases) {
            //如果当前线程不是持有的线程,抛出异常
            if (!isHeldExclusively())
                throw new IllegalMonitorStateException();
            //判断释放后,写状态的值,
            int nextc = getState() - releases;
            //如果是最后一次释放,则设置当前独占线程为null。
            boolean free = exclusiveCount(nextc) == 0;
            if (free)
                setExclusiveOwnerThread(null);
            setState(nextc);
            return free;
        }

读锁的获取

看完了写锁,现在来看看读锁,读锁不是独占式锁,即同一时刻该锁可以被多个读线程获取也就是一种共享式锁。 如果当前线程在获取读锁时,写锁已被其他线程获取,则进入等待状态。获取读锁的实现从Java 5到Java 6变得复杂许多,主要原因是新增了一些功能,例如getReadHoldCount()方法,作用是返回当前线程获取读锁的次数。读状态是所有线 程获取读锁次数的总和,而每个线程各自获取读锁的次数只能选择保存ThreadLocal中,由线程自身维护,这使获取读锁的实现变得复杂。按照之前对AQS介绍,实现共享式同步组件的同步语义需要通过重写AQS的tryAcquireShared和tryReleaseShared方法。

读锁的获取实现方法为(取tryAcquireShared重写的):

public boolean tryLock() {
            return sync.tryReadLock();
        }

final boolean tryReadLock() {
            Thread current = Thread.currentThread();
            for (;;) {
                int c = getState();
                //1. 如果写锁已经被获取并且获取写锁的线程不是当前线程的话,当前
              // 线程获取读锁失败返回false
                if (exclusiveCount(c) != 0 &&
                    getExclusiveOwnerThread() != current)
                    return false;
                //获取读状态
                int r = sharedCount(c);
                if (r == MAX_COUNT)
                    throw new Error("Maximum lock count exceeded");
                //如果读状态使用CAS操作加一也就是获取读锁成功。
                if (compareAndSetState(c, c + SHARED_UNIT)) {
                    //如果读状态为0,表示第一次读取成功,设置读状态为1
                    if (r == 0) {
                        firstReader = current;
                        firstReaderHoldCount = 1;
                    //如果读线程为当前线程,读状态加一
                    } else if (firstReader == current) {
                        firstReaderHoldCount++;
                    } else {
                    //如果都不满足
                        HoldCounter rh = cachedHoldCounter;
                        if (rh == null || rh.tid != getThreadId(current))
                            cachedHoldCounter = rh = readHolds.get();
                        else if (rh.count == 0)
                            readHolds.set(rh);
                        rh.count++;
                    }
                    return true;
                }
            }
        }

读锁的释放:

读锁释放的实现主要通过方法tryReleaseShared,读锁的每次释放(线程安全的,可能有多个读线程同时释放读锁)均减少读状态,减少的值是(1<<16)源码如下

这种的实现跟写锁是一样的。还有另外一种,在下面
 protected final boolean tryRelease(int releases) {
            if (!isHeldExclusively())
                throw new IllegalMonitorStateException();
            int nextc = getState() - releases;
            boolean free = exclusiveCount(nextc) == 0;
            if (free)
                setExclusiveOwnerThread(null);
            setState(nextc);
            return free;
        }

//第二种:
//在ReadLock类中
public void unlock() {
            sync.releaseShared(1);
        }

//在AQS中
public final boolean releaseShared(int arg) {
        if (tryReleaseShared(arg)) {
            doReleaseShared();
            return true;
        }
        return false;
    }

//主要还是ReentrantReadWriteLock的sync类中重写tryReleaseShared

protected final boolean tryReleaseShared(int unused) {
            Thread current = Thread.currentThread();
          //当前线程为持有读锁的线程 
           if (firstReader == current) {
                // assert firstReaderHoldCount > 0;
                //如果当前读锁的状态值为1,则读锁只有一个。把firstReader = null;
                if (firstReaderHoldCount == 1)
                    firstReader = null;
                else
            //否则firstReaderHoldCount减一
                    firstReaderHoldCount--;
            } else {
            其他情况下。
                HoldCounter rh = cachedHoldCounter;
                if (rh == null || rh.tid != getThreadId(current))
                    rh = readHolds.get();
                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))
                    // 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;
            }
        }

锁降级

锁降级指的是写锁降级成为读锁。如果当前线程拥有写锁,然后将其释放,最后再获取读锁,这种分段完成的过程不能称之为锁降级。锁降级是指把持住(当前拥有的)写锁,再获取到读锁,随后释放(先前拥有的)写锁的过程。

void processCachedData() {
        rwl.readLock().lock();
        if (!cacheValid) {
            // Must release read lock before acquiring write lock
            rwl.readLock().unlock();
            rwl.writeLock().lock();
            try {
                // Recheck state because another thread might have
                // acquired write lock and changed state before we did.
                if (!cacheValid) {
                    data = ...
            cacheValid = true;
          }
          // Downgrade by acquiring read lock before releasing write lock
          rwl.readLock().lock();
        } finally {
          rwl.writeLock().unlock(); // Unlock write, still hold read
        }
      }
 
      try {
        use(data);
      } finally {
        rwl.readLock().unlock();
      }
    }
}