Java显式锁学习总结之四:ReentrantLock源码分析

概述

ReentrantLock,即重入锁,是一个和synchronized关键字等价的,支持线程重入的互斥锁。只是在synchronized已有功能基础上添加了一些扩展功能。

除了支持可中断获取锁、超时获取锁、非阻塞获取锁这些显示锁的常见功能外,ReentrantLock还支持公平锁(synchronized只支持非公平锁)。

下面分析源码时将聚焦重入和公平这两个功能点的实现。

结构总览

重入锁的大体结构如下:

public class ReentrantLock implements Lock, java.io.Serializable {
    abstract static class Sync extends AbstractQueuedSynchronizer {}
    static final class NonfairSync extends Sync {}
    static final class FairSync extends Sync {}
}

Sync类是AQS的子类,而NonfairSync和FairSync是Sync的子类。

公平锁和非公平锁的区别就在于获取锁时候的逻辑略有不同,其他操作都是一样的,因此公用的操作都放在Sync类里,NonfairSync和FairSync里只是实现自己的tryAcquire(int acquires)方法。

AQS里的state在重入锁里代表线程重入的次数,state=1代表重入锁当前已被某个线程独占,这个线程每重入一次,state++。因为state是int型变量,因此重入锁可以重入的最大次数是2^31-1。

重入实现

重入实现其实上边已经提到了,就是利用state状态表示重入次数,我们以非公平锁的代码为例看一下,下面是Sync类里的 nonfairTryAcquire(int acquires)方法 (上面我们说过,Sync类里是存放NonfairSync与FairSync的公用代码,那么这个nonfairTryAcquire方法为什么放到Sync里呢?我们后面会解释,不要急:) 

        final boolean nonfairTryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();
//state为0代表当前锁没有线程持有,则让当前线程持有锁
if (c == 0) { if (compareAndSetState(0, acquires)) { setExclusiveOwnerThread(current); return true; } }
//如果锁被当前线程持有,则把state+acquires(就是1)
else if (current == getExclusiveOwnerThread()) { int nextc = c + acquires; if (nextc < 0) // 如果state<0,说明超过了int最大值,溢出了 throw new Error("Maximum lock count exceeded"); setState(nextc); // 这里不用CAS,因为锁已被当前线程独占 return true; } return false; }

公平实现

公平锁的实现非常简单,其实就是一句代码,我们看一下FairSync类里的 tryAcquire(int acquires) 方法:

        protected final boolean tryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();
            if (c == 0) {
                if (!hasQueuedPredecessors() && 
                    compareAndSetState(0, acquires)) {
                    setExclusiveOwnerThread(current);
                    return true;
                }
            }
            else if (current == getExclusiveOwnerThread()) {
                int nextc = c + acquires;
                if (nextc < 0)
                    throw new Error("Maximum lock count exceeded");
                setState(nextc);
                return true;
            }
            return false;
        }
    public final boolean hasQueuedPredecessors() {
        // The correctness of this depends on head being initialized
        // before tail and on head.next being accurate if the current
        // thread is first in queue.
        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());
    }

 

我们可以发现,这段代码与上面的 nonfairTryAcquire方法就只多了一句代码,而就是这一句代码就实现了公平锁。hasQueuedPredecessors()方法判断同步队列中是否有更早开始等待锁的线程。如果有,则tryAcquire方法直接返回false让当前线程进入同步队列排队。

特殊的tryLock

之前我们在将重入实现的时候说到,Sync类里有个诡异的nonfairTryAcquire方法,听名字是和非公平锁相关的,按道理应该放到NonfairSync类啊。之所以有这么别扭的设计是为了服务tryLock()方法。

看一下ReentrantLock类的tryLock方法的实现:

    public boolean tryLock() {
        return sync.nonfairTryAcquire(1);
    }

可以看出,不管是公平锁还是非公平锁,调用的都是 nonfairTryAcquire 方法。

为什么这么实现呢?我们可以想一下tryLock() 的语义,tryLock() 要实现的效果是尝试获取一次锁,如果获取失败不阻塞而是直接返回false。如果在公平锁模式下严格按照公平锁的定义来实现这个方法,那么当同步队列中有其他线程等待的时候,tryLock()都不可能获取到锁,只能返回false。

而事实上,当我们调用tryLock()的时候,很多时候应该都是希望尽可能的成功的,而此时要不要让tryLock()的线程严格排队,其实不是那么重要,因此公平锁下tryLock()方法在获取锁时使用非公平获取模式,即可以插队。

那么如果我们在公平锁模式下就希望tryLock()方法获取锁严格排队呢?可以用tryLock(0, TimeUnit.SECONDS),这个方法等效于一个严格排队的tryLock()方法,之所以等效,是因为tryLock(long timeout, TimeUnit unit)的实现是区分公平锁和非公平锁的,在公平锁的模式下,获取锁的操作是严格按同步队列排队等待的。

那么如果我们在公平锁模式下希望 tryLock(long timeout, TimeUnit unit) 不严格排队,表现的像一个支持超时的tryLock()呢?也是有办法的:

 if (lock.tryLock() || lock.tryLock(timeout, unit)) {
   ...
 }

可以这样组合一下,左边的tryLock()有机会插队获取一次锁,如果没获取到,在用tryLock(timeout, unit)做一次可超时的同步队列排队。

 

总结

有了前面对AQS的理解基础,现在再来看同步组件的实现,就如果快刀切西瓜!所以说,Doug Lea对AQS的设计真的非常巧妙,ReentrantLock没有用多少代码,就实现了一个加强版的synchronizer。

posted @ 2017-02-28 12:45  叫我家宝  阅读(719)  评论(0编辑  收藏  举报