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是隨機喚醒一個線程或者所有線程。

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。