ReentrantLock原始碼分析
概述
ReentrantLock是一個可重入的互斥鎖,也被稱為獨佔鎖。它支援公平鎖和非公平鎖兩種模式。
ReentrantLock的使用方法
下面看一個最初級的例子:
public class Test { //預設內部採用非公平實現 ReentrantLock lock=new ReentrantLock(); public void myMethor(){ lock.lock(); //需要加鎖的一些操作 //一定要確保unlock能被執行到,尤其是在存在異常的情況下 lock.unlock(); } }
在進入方法後,在需要加鎖的一些操作執行之前需要呼叫lock方法,在jdk文件中對lock方法詳細解釋如下:
獲得鎖。
如果鎖沒有被另一個執行緒佔用並且立即返回,則將鎖定計數設定為1。 如果當前執行緒已經保持鎖定,則保持計數增加1,該方法立即返回。 如果鎖被另一個執行緒保持,則當前執行緒將被禁用以進行執行緒排程,並且在鎖定已被獲取之前處於休眠狀態,此時鎖定保持計數被設定為1。
這裡也很好的解釋了什麼是可重入鎖,如果一個執行緒已經持有了鎖,它再次請求獲取自己已經拿到的鎖,是能夠獲取成功的,這就是可重入鎖。
在需要加鎖的程式碼執行完畢之後,就會呼叫unlock釋放掉鎖。在jdk文件之中對,unlock的解釋如下:
嘗試釋放此鎖。
如果當前執行緒是該鎖的持有者,則保持計數遞減。 如果保持計數現在為零,則鎖定被釋放。 如果當前執行緒不是該鎖的持有者,則丟擲IllegalMonitorStateException 。
在這裡有一個需要注意的地點,lock和unlock都反覆提到了一個計數,這主要是因為ReentrantLock是可重入的。每次獲取鎖(重入)就將計數器加一,每次釋放的時候的計數器減一,直到計數器為0,就將鎖釋放掉了。
以上就是最基礎,最簡單的使用方法。其餘的一些方法,都是一些拓展的功能,檢視jdk文件即可知道如何使用。
原始碼分析
繼承體系
可以看出ReentrantLock繼承自AQS並實現了Lock介面。它內部有公平鎖和非公平鎖兩種實現,這兩種實現都是繼承自Sync。根據ReentrantLock決定到底採用公平鎖還是非公平鎖實現。
public ReentrantLock(boolean fair) {
sync = fair ? new FairSync() : new NonfairSync();
}
核心方法原始碼分析
Lock方法
- 首先呼叫具體的Lock實現.sync可能是非公平鎖實現也可能是公平鎖實現,這取決於你new物件時的引數。
public void lock() {
sync.lock();
}
我們以非公平鎖實現來看下面的下面的程式碼。
- 非公平鎖的lock方法的具體實現如下
final void lock() {
if (compareAndSetState(0, 1))
setExclusiveOwnerThread(Thread.currentThread());
else
acquire(1);
}
首先進來就是一個判斷,其中判斷的條件就是compareAndSetState(0, 1)
.毫無疑問這是一個CAS。它的意思時如果當前的state的值的為0就將1與其交換(可以理解為將1賦值給0)並返回true。其實在這一步如果state的值修改成功了,那麼鎖就獲取成功了。setExclusiveOwnerThread(Thread.currentThread())
這行程式碼就是將當前執行緒設定為該排他鎖的擁有者。
如果CAS失敗了,那麼就呼叫acquire(1);
- 如果初次獲得鎖失敗就呼叫
qcquire(1)
這個方法的具體實現如下;
public final void acquire(int arg) {
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
selfInterrupt();
}
這個方法進來首先第一步就是呼叫tryAcquire(arg)
.
那麼該方法是幹什麼的呢?
非公平鎖實際是呼叫了這個實現:
protected final boolean tryAcquire(int acquires) {
return nonfairTryAcquire(acquires);
}
它具體的實現是在nonfairTryAcquire(acquires)
中。
final boolean nonfairTryAcquire(int acquires) {
final Thread current = Thread.currentThread();
int c = getState(); //獲取鎖的狀態state,這就是前面我們CAS的操作物件
if (c == 0) {
//c==0說明沒被其它獲取
if (compareAndSetState(0, acquires)) { //CAS修改state
//CAS修改成功,說明獲取鎖成功,將當前執行緒設定為該排他鎖的擁有者
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;
}
//鎖被其它執行緒佔有,就返回false
return false;
}
- 第二次嘗試獲取鎖失敗後,就進行下一步操作
我們再會過頭看void acquire(int arg)
首先嚐試獲取鎖,獲取成功就直接返回了,獲取失敗就會執行acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
進行排隊。
這一行程式碼可以分為兩部分看,一部分是addWaiter(Node.EXCLUSIVE)
一部分是acquireQueued
.我們先看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;
}
初始化對立對入隊的具體實現如下:
private Node enq(final Node node) {
for (;;) {
Node t = tail;
if (t == null) { // Must initialize
//初始化佇列
if (compareAndSetHead(new Node()))
tail = head;
} else {
//佇列初始化成功,進行入隊
node.prev = t;
if (compareAndSetTail(t, node)) {
t.next = node;
return t;
}
}
}
}
這裡稍微補充一下這個AQS中的這個等待佇列。
- 節點也建立了,等待佇列也入了
現在該看boolean acquireQueued(final Node node, int arg)
方法了。
這個方法的具體實現如下:
final boolean acquireQueued(final Node node, int arg) {
boolean failed = true;
try {
boolean interrupted = false;
for (;;) {
final Node p = node.predecessor(); //獲取當前節點的先驅節點
if (p == head && tryAcquire(arg)) {
//如果當前節點的前一個節點是頭節點,就會執行tryAcquire(arg)再次嘗試獲取鎖
setHead(node);
p.next = null; // help GC
failed = false;
return interrupted;
}
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
//根據情況進入park狀態
interrupted = true;
}
} finally {
if (failed)
cancelAcquire(node);
}
}
unlock方法
- 和加鎖類似,呼叫具體的實現
public void unlock() {
sync.release(1);
}
- 具體的release實現
public final boolean release(int arg) {
if (tryRelease(arg)) {
Node h = head;
if (h != null && h.waitStatus != 0)
//喚醒等待的執行緒,可以拿鎖了
unparkSuccessor(h);
return true;
}
return false;
}
該方法首先就呼叫了tryRelease(arg)
方法,這個方法就是實現釋放資源的關鍵。釋放的具體操作,也印證了在jdk文件之中的關於unlock和lock的說明。
protected final boolean tryRelease(int releases) {
int c = getState() - releases; //計算釋放後的state的值
if (Thread.currentThread() != getExclusiveOwnerThread())
//如果當前執行緒沒有持有鎖,就拋異常
throw new IllegalMonitorStateException();
boolean free = false; //標記為釋放失敗
if (c == 0) {
//如果state為0了,說沒沒有執行緒佔有該鎖了
//進行重置所有者
free = true;
setExclusiveOwnerThread(null);
}
//重置state的值
setState(c);
return free;
}
- 如果還有執行緒在等待鎖資源,那麼就可以喚醒它們了
回到boolean release(int arg)
看
if (h != null && h.waitStatus != 0)
//喚醒等待的執行緒,可以拿鎖了
unparkSuccessor(h);
ReentrantLock的高階使用方法
我們使用synchronized的時候,可以通過wait和notify來讓執行緒等待,和喚醒執行緒。在ReentrantLock中,我們也可以使用Condition中的await和signal來使執行緒等待和喚醒。
以下面這段程式碼來解釋:
public class Test {
static ReentrantLock lock=new ReentrantLock();
//獲取到condition
static Condition condition=lock.newCondition();
public static class TaskA implements Runnable{
@Override
public void run() {
lock.lock();
System.out.println(Thread.currentThread().getName() + "開始執行");
try {
System.out.println(Thread.currentThread().getName() + "準備釋放掉鎖並等待");
//在此等待,直到其它執行緒喚醒
condition.await();
System.out.println(Thread.currentThread().getName() + "重新拿到鎖並執行");
} catch (InterruptedException e) {
e.printStackTrace();
}finally {
lock.unlock();
}
}
}
public static class TaskB implements Runnable{
@Override
public void run() {
lock.lock();
System.out.println(Thread.currentThread().getName() + "開始執行");
System.out.println(Thread.currentThread().getName() + "開始喚醒等待的執行緒");
//喚醒等待的執行緒
condition.signal();
try {
Thread.sleep(2000);
System.out.println(Thread.currentThread().getName() + "任務執行完畢");
}catch (InterruptedException e){
e.printStackTrace();
}finally {
lock.unlock();
}
}
}
public static void main(String[] args) {
Thread taskA=new Thread(new TaskA(),"taskA");
Thread taskB=new Thread(new TaskB(),"taskB");
taskA.start();
taskB.start();
}
}
輸出結果:
taskA開始執行
taskA準備釋放掉鎖並等待
taskB開始執行
taskB開始喚醒等待的執行緒
taskB任務執行完畢
taskA重新拿到鎖並執行
現象解釋:
首先taskA拿到鎖,並執行,到condition.await();
釋放鎖,並進入阻塞。taskB因此拿到剛才taskA釋放掉的鎖,taskB開始執行。taskB執行到condition.signal();
喚醒了taskA,taskB繼續執行,taskA因為拿不到鎖,因此雖然已經被喚醒了,但是還是要等到taskB執行完畢,釋放鎖後,才有機會拿到鎖,執行自己的程式碼。
那麼這個過程,原始碼到底是如何實現的呢?
Condition原始碼分析
await()的原始碼分析
具體的實現如下:
public final void await() throws InterruptedException {
if (Thread.interrupted())
throw new InterruptedException();
Node node = addConditionWaiter(); //新增一個條件節點
int savedState = fullyRelease(node); //釋放掉所有的資源
int interruptMode = 0;
while (!isOnSyncQueue(node)) {
//如果當前執行緒不在等待佇列中,park阻塞
LockSupport.park(this);
if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
break; //執行緒被中斷就跳出迴圈
}
if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
interruptMode = REINTERRUPT;
if (node.nextWaiter != null) // clean up if cancelled
//取消條件佇列中已經取消的等待節點的連結
unlinkCancelledWaiters();
if (interruptMode != 0)
//等待結束後處理中斷
reportInterruptAfterWait(interruptMode);
}
基本的步驟如下:
- 首先判斷執行緒是否被中斷,如果中斷則丟擲
InterruptedException()
異常 - 添加當前執行緒到條件佇列中去,然後釋放掉所有的資源
- 如果當前執行緒不在等待佇列中,就直接park阻塞當前執行緒
signal()方法原始碼分析
具體的實現程式碼如下:
public final void signal() {
if (!isHeldExclusively())
throw new IllegalMonitorStateException();
Node first = firstWaiter;
if (first != null)
doSignal(first);
}
這個方法中最重要的也就是doSignal(first)
.
它的實現如下:
private void doSignal(Node first) {
do {
if ( (firstWaiter = first.nextWaiter) == null)
lastWaiter = null;
first.nextWaiter = null; //解除等待佇列中首節點的連結
} while (!transferForSignal(first) && //轉移入等待佇列
(first = firstWaiter) != null);
}
該方法所做的事情就是從等待佇列中移除指定節點,並將其加入等待佇列中去。
轉移節點的方法實現如下:
final boolean transferForSignal(Node node) {
/*
* If cannot change waitStatus, the node has been cancelled.
*/
if (!compareAndSetWaitStatus(node, Node.CONDITION, 0))
//CAS修改狀態失敗,說明節點被取消了,直接返回false
return false;
/*
* Splice onto queue and try to set waitStatus of predecessor to
* indicate that thread is (probably) waiting. If cancelled or
* attempt to set waitStatus fails, wake up to resync (in which
* case the waitStatus can be transiently and harmlessly wrong).
*/
Node p = enq(node); //加入節點到等待佇列
int ws = p.waitStatus;
if (ws > 0 || !compareAndSetWaitStatus(p, ws, Node.SIGNAL))
//如果前節點被取消,說明當前為最後一個等待執行緒,直接unpark喚醒,
LockSupport.unpark(node.thread);
return true;
}
至此ReentrantLock的原始碼分析就結束了