写点什么

Java Core 「12」ReentrantLock 再探析

作者:Samson
  • 2022 年 6 月 19 日
  • 本文字数:3388 字

    阅读完需:约 11 分钟

在《Java Core 「11」AQS-AbstractQueuedSynchronizer》中介绍过 AQS 之后,再回过头来看一下 ReentrantLock 中的实现细节。

ReentrantLock 中有三个内部类,Sync 继承自 AQS,FairSync 实现公平锁机制,NonfairSync 实现非公平锁机制。

01-公平锁 & 非公平锁中的 tryAcquire

非公平锁中获取锁的过程:

final Thread current = Thread.currentThread();int c = getState();if (c == 0) {    /** 锁空闲,则当前线程获取成功 */    if (compareAndSetState(0, acquires)) {        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;j
复制代码

公平锁中获取锁的过程:

final Thread current = Thread.currentThread();int c = getState();if (c == 0) {    /** 锁空闲时 */    if (!hasQueuedPredecessors() &&  /** 检查 sync 队列总是否有其他线程在等待 */        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;
复制代码

通过上面的代码可以发现,在锁空闲时,当前线程能否获得锁的条件是不一样的:

  • 公平锁,若 sync 队列中存在其他线程在等待,则当前线程不能获得锁。公平的含义指线程按照“先来先得“的原则获得锁;

  • 非公平锁,不考虑其他线程,若锁空闲,则当前线程立即获得锁。不公平的含义是不遵循”先来先得“原则,当前线程立即抢占锁。

02-Sync 中的 tryRelease

公平锁 & 非公平锁释放锁时使用的 tryRelease 都是同一份,即 Sync 中的 tryRelease 方法:

int c = getState() - releases;if (Thread.currentThread() != getExclusiveOwnerThread())    throw new IllegalMonitorStateException();boolean free = false;/** 先判断释放后,锁状态是否归零,若归零说明锁空闲,允许其他线程获取锁 */if (c == 0) {    free = true;    setExclusiveOwnerThread(null);}setState(c);return free;
复制代码

03-公平锁示例分析

我们结合一个实例分析一下多个线程获得公平锁的执行情况,我们创建三个线程 t1/t2/t3,执行同一份代码runnable,使用同一公平锁lock

ReentrantLock lock = new ReentrantLock(true);Runnable runnable = () -> {    lock.lock();    try {        System.out.println(Thread.currentThread() + " running");        try {            TimeUnit.MILLISECONDS.sleep(500);        } catch (InterruptedException ignored) {}    } finally {        lock.unlock();    }};
new Thread(runnable, "t1").start();new Thread(runnable, "t2").start();new Thread(runnable, "t3").start();
复制代码

不妨假设线程被调度的顺序是 t1 → t2 → t3,所以存在如下一种时序:t1 执行lock.lock();获得锁,开始执行并 sleep 500ms,此时 t2 执行lock.lock();被阻塞,t3 执行lock.lock();同样被阻塞。

注:以下的阅读顺序,可以按照先 Lock 流程,再 Unlock 流程。

03.1-t1 的执行流程

Lock 流程

ReentrantLock#lock → AbstractQueuedSynchronizer#acquire → ReentrantLock.FairSync#tryAcquire (true, state 是 0,sync 队列为空)

Unlock 流程

ReentrantLock#unlock → AbstractQueuedSynchronizer#release

public final boolean release(int arg) {    /** state 由1变0,返回结果 true */    if (tryRelease(arg)) {        Node h = head;        /** h 不为 null,而且 waitStatus 为 SIGNAL(-1),由 t2 自旋时 park 之前设置,具体看下节 03.2 */        if (h != null && h.waitStatus != 0)            /** 会将 h (head) 的 waitStatus 设置为 0,             * h.next != null,会 LockSupport.unpark(h.next.thread)             * t2 会在 LockSupport.park 处被唤醒             */            unparkSuccessor(h);        return true;    }    return false;}
复制代码

03.2-t2 的执行流程

Lock 流程

ReentrantLock#lock → AbstractQueuedSynchronizer#acquire → ReentrantLock.FairSync#tryAcquire (false,state 是 1,为 t1 tryAcquire 时设置) → AbstractQueuedSynchronizer#addWaiter (会初始化 sync 队列,并将 t2 包装到 Node 中加入队列尾部)→ AbstractQueuedSynchronizer#acquireQueued

final boolean acquireQueued(final Node node, int arg) {    boolean interrupted = false;    try {        for (;;) {	          /** t2 执行时,p 为 head */            final Node p = node.predecessor();            { /** ReentrantLock.FairSync#tryAcquire 仍会失败*/            if (p == head && tryAcquire(arg))                setHead(node);                p.next = null; // help GC                return interrupted;            }	          /**              * 这段代码在无限循环中(线程 t2 在自旋)             * 第1次,p (head) 的 waitStatus 为0,然后被设置为 SIGNAL(-1)	           * 第2次,线程 t2 会在 parkAndCheckInterrupt 中 LockSupport.park(this);	           * 当 t1 在 unlock 中唤醒 t2 后,             * 根据 t2.isInterrupted(true) 判断是否被中断过更新 interrupted 的值             * 无限循环继续,再次执行到 ttryAcquire 时会获得锁             * 此时,setHead(node) 会把 head 指向 t2 所在节点             */            if (shouldParkAfterFailedAcquire(p, node))                interrupted |= parkAndCheckInterrupt();        }    } catch (Throwable t) {        cancelAcquire(node);        if (interrupted)            selfInterrupt();        throw t;    }}
复制代码

Unlock 流程

与 03.1 中 unlock 流程一致

03.3-t3 的执行流程

Lock 流程

ReentrantLock#lock → AbstractQueuedSynchronizer#acquire → ReentrantLock.FairSync#tryAcquire (false,state 是 1,为 t1 tryAcquire 时设置) → AbstractQueuedSynchronizer#addWaiter (将 t3 包装到 Node 中加入 sync 队列尾部) → AbstractQueuedSynchronizer#acquireQueued

final boolean acquireQueued(final Node node, int arg) {    boolean interrupted = false;    try {        for (;;) {	          /** t3 执行时,p 为 t2 所在的节点  */            final Node p = node.predecessor();            { /** 这里不会执行 */            if (p == head && tryAcquire(arg))                setHead(node);                p.next = null; // help GC                return interrupted;            }	          /**              * 这段代码在无限循环中(线程 t3 在自旋)             * 第1次,p (t2 节点) 的 waitStatus 为0,然后被设置为 SIGNAL(-1)	           * 第2次,线程 t3 会在 parkAndCheckInterrupt 中 LockSupport.park(this);             */            if (shouldParkAfterFailedAcquire(p, node))                interrupted |= parkAndCheckInterrupt();        }    } catch (Throwable t) {        cancelAcquire(node);        if (interrupted)            selfInterrupt();        throw t;    }}
复制代码

Unlock 流程

与 03.1 中 unlock 流程一致


历史文章推荐

Java Core 「11」AQS-AbstractQueuedSynchronizer

Java Core 「10」J.U.C 同步工具类 -2

Java Core 「9」J.U.C 同步工具类 -1

Java Core 「8」字节码增强技术

Java Core 「7」各种不同类型的锁

Java Core「6」反射与 SPI 机制

Java Core「5」自定义注解编程

Java Core「4」java.util.concurrent 包简介

Java Core「3」volatile 关键字

Java Core「2」synchronized 关键字

Java Core「1」JUC- 线程基础

发布于: 刚刚阅读数: 5
用户头像

Samson

关注

还未添加个人签名 2019.07.22 加入

还未添加个人简介

评论

发布
暂无评论
Java Core 「12」ReentrantLock 再探析_学习笔记_Samson_InfoQ写作社区