ReentrantLock原理详解

@[TOC]

前言

了解ReentrantLock的原理和底层实现

一、ReentrantLock是如何实现的?

ReentrantLock 主要利用CAS 和 AQS队列实现,并且支持公平锁和非公平锁。

CAS:Compare and Swap,比较并交换。在java中,CAS主要由Unsafe类通过JIN调用CPU底层代码实现。

CAS有三个参数:内存值V、预期值A、新值B。当且仅当,预期值A和内存值V相同时,将内存值修改为B,否则不做修改。

关于AQS,参见文章AQS队列

二、lock 和 unlock的过程分析

2.1 ReentrantLock的两个构造器

public ReentrantLock() {
        sync = new NonfairSync();
    }
public ReentrantLock(boolean fair) {
        sync = fair ? new FairSync() : new NonfairSync();
    }

默认构造器会将其实例化为NonfairSync对象,即非公平锁,而带参数的构造器可以指定使用公平锁还是非公平锁。

2.2 lock过程

NonfairSync.lock

final void lock() {
            if (compareAndSetState(0, 1))
                setExclusiveOwnerThread(Thread.currentThread());
            else
                acquire(1);
        }

非公平锁的lock方法中,首先是一个CAS操作,判断state是否是0,即当前锁是否被其他线程占用了;如果是0,则设置为1,且设置当前线程为独占线程。这里CAS操作避免了并发操作的问题。

这里也是第一处体现非公平锁:如果此处占用锁的线程刚刚好释放锁,则当前线程可以直接获取锁,而无需排队,相当于直接“插队”。

若此处没有设置成功,即没有获取锁,则会执行acquire()方法。
acquire()

public final void acquire(int arg) {
        if (!tryAcquire(arg) &&
            acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            selfInterrupt();
    }

第一步是调用tryAcquire()方法,尝试获取锁。

final boolean nonfairTryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();
            if (c == 0) { //状态为0,没有线程占用锁
                if (compareAndSetState(0, acquires)) {
                    //与lock方法里一样,直接获取锁
                    setExclusiveOwnerThread(current);
                    return true;
                }
            }
            else if (current == getExclusiveOwnerThread()) { // 当前线程已经占用该锁
                int nextc = c + acquires; // 锁重入次数
                if (nextc < 0) // overflow
                    throw new Error("Maximum lock count exceeded");
                setState(nextc);
                return true;
            }
            return false; // 获取锁失败,返回false
        }

这里第二次体现了不公平锁。

这里还检验了是不是当前线程已经获取锁,而又重新获取锁。这里体现了重入锁。会更新状态值为重入的次数。

第二步addWaiter(Node.EXCLUSIVE)。

private Node addWaiter(Node mode) {
        Node node = new Node(Thread.currentThread(), mode);
        // Try the fast path of enq; backup to full enq on failure
        Node pred = tail;
        if (pred != null) {
            node.prev = pred;
            if (compareAndSetTail(pred, node)) {
                pred.next = node;
                return node;
            }
        }
        enq(node);
        return node;
    }

这个添加到队列的方法在上篇文章AQS已经分析过了,就不详细分析了。

第三步acquireQueued(),此方法是让队列中的线程尝试获取锁,如果失败则挂起线程。

final boolean acquireQueued(final Node node, int arg) {
        boolean failed = true; // 标记是否获取锁成功
        try {
            boolean interrupted = false; // 线程中断标志
            for (;;) {  // 自旋
                final Node p = node.predecessor();  // 获得前驱节点
                // 如果该节点的前驱节点是head,则该节点是队列中的第一个线程,可以尝试互殴去锁
                if (p == head && tryAcquire(arg)) { 
                    setHead(node); // 获取锁成功
                    p.next = null; // help GC,不然GC无法回收
                    failed = false; // 获取锁成功
                    return interrupted; //返回中断标志,没有被中断
                }
                // 如果获取失败,是否可以挂起
                if (shouldParkAfterFailedAcquire(p, node) &&
                    parkAndCheckInterrupt())
                    interrupted = true;
            }
        } finally {
            if (failed)
                cancelAcquire(node); // 最后取消尝试获取锁
        }
    }

当上述方法获取锁失败之后,就会判断线程是否需要挂起。下面说明下shouldParkAfterFailedAcquire()和parkAndCheckInterrupt()方法。

private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
        // 前驱节点的状态
        int ws = pred.waitStatus;
        if (ws == Node.SIGNAL)
            /*
             * This node has already set status asking a release
             * to signal it, so it can safely park.
             */
            return true; // 如果前驱节点状态是SIGNAL,则返回true,可以挂起
        // 前驱节点是CANCELLED
        if (ws > 0) {
            /*
             * Predecessor was cancelled. Skip over predecessors and
             * indicate retry.
             */
            // 从队尾向前循环查找,直到找到第一个不为CANCELLED的节点
            do {
                node.prev = pred = pred.prev;
            } while (pred.waitStatus > 0);
            pred.next = node;
        } else {
            /*
             * waitStatus must be 0 or PROPAGATE.  Indicate that we
             * need a signal, but don't park yet.  Caller will need to
             * retry to make sure it cannot acquire before parking.
             */
            // 设置前驱节点状态为SIGNAL
            compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
        }
        return false;
    }
 // 挂起当前线程,并返回线程中断状态
 private final boolean parkAndCheckInterrupt() {
        LockSupport.park(this);
        return Thread.interrupted();
    }

这里解释下,线程挂起的前提是,前驱节点是SIGNAL状态。那SIGNAL状态的含义是什么?SIGNAL表示其前驱节点线程如果获取锁,退出队列,则将其唤醒。

2.3 unlock过程

unlock

public void unlock() {
        sync.release(1);
    }
public final boolean release(int arg) {
        if (tryRelease(arg)) { // 尝试释放锁
            Node h = head; // 获取头节点
            if (h != null && h.waitStatus != 0)
                unparkSuccessor(h); // 如果头节点不为空且不为状态不为0,则取消挂起
            return true; // 释放成功
        }
        return false; // 释放失败
    }

这里先尝试释放锁,如果成功,就将头节点取消挂起。

tryRealase

protected final boolean tryRelease(int releases) {
            // 当前锁状态值减去释放值,注意考虑重入锁,不是直接置为0
            int c = getState() - releases;
            // 如果不是当前线程占用锁,则抛出异常
            if (Thread.currentThread() != getExclusiveOwnerThread())
                throw new IllegalMonitorStateException();
            boolean free = false;
            if (c == 0) { // 如果状态值等于0,则释放锁成功
                free = true;
                // 清空独占线程
                setExclusiveOwnerThread(null);
            }
            // 更新state 值
            setState(c);
            return free;
        }

三、公平锁和非公平锁的区别

下面介绍非公平锁和公平锁的区别,公平锁在获取锁之前, 不会直接CAS去获取锁,而是先判断当前锁状态是不是等于0且当前线程是不是队列中第一个线程,state=0才去尝试获得锁。

protected final boolean tryAcquire(int acquires) {
            // 获取当前线程
            final Thread current = Thread.currentThread();
            // 获取锁状态
            int c = getState();
            if (c == 0) { // 如果锁没有被任何线程拥有
                // 当前线程是CLH队列的的第一个线程,则取获取锁
                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;
        }
    }

四、Synchronized和ReentrantLock的区别

4.1 功能区别

Synchronized是Java语言的关键字,是通过JVM实现的。

ReentrantLock是API层面的一种互斥锁,而且它的锁粒度和灵活性都优于Synchronized。

Synchronized使用较为简单方便,由编译器去保证锁的加锁和释放。而ReentrantLock需要手动加锁和释放锁,如果忘记手工释放锁,会造成死锁。

4.2 性能区别

在Synchronized优化之前,性能是比较差的,因为都是重量级锁。但是Synchronized引入偏向锁,轻量级锁之后,两者的性能相差不大。当不需要使用ReentrantLock的特有性质时,官方建议使用Synchronized。

4.3 ReentrantLock的特有性质

等待可中断:持有锁的线程长期不释放的时候,正在等待的线程可以选择放弃等待,这相当于Synchronized来说可以避免出现死锁的情况。通过lock.lockInterruptibly()来实现这个机制。

公平锁:多个线程等待同一个锁时,必须按照申请锁的时间顺序获得锁,Synchronized锁是非公平锁,ReentrantLock可以实现公平锁。

选择性通知(锁绑定多个条件):一个ReentrantLock可以同时绑定多个对象,然后通过condition(条件)类,来分组唤醒线程,而Synchronized是随机唤醒一个线程或者所有线程。

你可能感兴趣的:(ReentrantLock原理详解)