• ReentrantLock和condition源码浅析(一)


     转载请注明出处。。。。。

    一、介绍

    大家都知道,在java中如果要对一段代码做线程安全操作,都用到了锁,当然锁的实现很多,用的比较多的是sysnchronize和reentrantLock,前者是java里的一个关键字,后者是一个java类。这两者的大致区别,在这里罗列下

    相同点:

           1、都能保证了线程安全性

           2、都支持锁的重入

    不同点:

          1、synchronized适用于不是很激烈的情况,reentranLock适用于比较竞争激烈的情况

          2、Synchronized是jvm层面实现的锁机制,而reentranLock是java代码层面实现的锁机制。

          3、Reentranlock比synchronized多了锁投票,定时锁,中断锁等机制

          4、synchronized是隐式获取锁和释放锁,不需要代码手动获取释放,Reentranlock为显示获取锁和释放锁,必须要手动代码获取释放

    要了解reentranlock,那肯定先得会用它,下面通过一个例子来了解它的加锁和释放锁过程

    二、demo

     1 public class Demo {
     2 
     3     private static int count = 0;
     4 
     5     public static void main(String[] args) throws InterruptedException {
     6         ExecutorService executorService = Executors.newFixedThreadPool(15);
     7         for (int i = 0; i < 500; i++){
     8             executorService.execute(() -> {
     9                 add();
    10             });
    11         }
    12         Thread.sleep(1000);
    13         System.out.println(count);
    14     }
    15 
    16     private static int add(){
    17         return ++count;
    18     }
    19 }

     上述代码,安装预期结果 那肯定是500,但是真的是500吗?结果如下

    结果很显然,它是小于500的,把这段代码用锁保证结果和预期结果一致。代码如下

     1 public class Demo {
     2 
     3     private static int count = 0;
     4 
     5     private static Lock lock = new ReentrantLock();
     6 
     7     public static void main(String[] args) throws InterruptedException {
     8         ExecutorService executorService = Executors.newFixedThreadPool(15);
     9         for (int i = 0; i < 500; i++){
    10             executorService.execute(() -> {
    11                 add();
    12             });
    13         }
    14         Thread.sleep(1000);
    15         System.out.println(count);
    16     }
    17 
    18     private static int add(){
    19         lock.lock();
    20         try {
    21             return ++count;
    22         }finally {
    23             lock.unlock();
    24         }
    25 
    26     }
    27 }

    结果,和预期一致。

    那它是怎么保证线程安全性的呢。往下看

    三、ReentrantLock分析

     先来了解这个类的大致结构

    红框圈中的三个类,其中Sync是一个抽象类,另外两个是它的子类,Sync又继承了AQS类,所以它也有锁的操作可能性。

    FairSync是一个公平锁,NonFairSync是一个非公平锁,它们虽然继承了同一个类,但实现上有所不同,

    1、非公平锁获取锁的过程

    进入lock方法

    而sync 是ReentrantLock的一个字段,它在该类的构造函数中初始化,它有两个构造函数,sync默认为非公平锁实现,

    当sync调用了lock方法,也就是调用NonFairSync类的lock方法,继续看下去,下图为该类的结构

     

    lock大致步骤为,先去试着改变state的值,如果改变成功,则state值就变为1了,返回true,失败返回false,先来解释下compareAndSetState方法的作用

    它有两个参数,第一个是期望值,第二个是要更新的值,如果内存中state值和期望值相等,则将内存值变为更新值,这是交换成功的标志。如果不相等,那肯定是false。这个方法其实就是CAS,同时它也是线程安全的,具体实现,这里不作讨论。

    这里也是获取锁成功的标志,当返回true,则将获取锁的线程置为当前线程,同时state值改变了,如果下一个线程进入,那么该方法肯定是返回false。那么获取锁失败的线程就会进入acquire方法。这个方法其实就是AQS的方法,代码如下,可以看到它又调用了tryAcquire方法,而这个方法的实现就是上一个图的nonFairTryAcquire方法,

     1 final boolean nonfairTryAcquire(int acquires) {
     2             // 获取当前线程
     3             final Thread current = Thread.currentThread();
     4             int c = getState();
     5             // 如果状态值不为0,则进一步去获取锁
     6             if (c == 0) {
     7                 if (compareAndSetState(0, acquires)) {
     8                     // 获取锁成功,将锁置给当前线程
     9                     setExclusiveOwnerThread(current);
    10                     return true;
    11                 }
    12             }// 如果相等,则表明为锁的重入
    13             else if (current == getExclusiveOwnerThread()) {
    14                 int nextc = c + acquires;
    15                 if (nextc < 0) // overflow
    16                     throw new Error("Maximum lock count exceeded");
    17                 setState(nextc);
    18                 return true;
    19             }
    20            //  只有获取锁失败才会返回false
    21             return false;
    22         }

     当上面返回false时,又会相继执行addWaiter和acquireQueued方法,其中addWaiter方法主要是将获取锁失败的线程包装成一个Node节点,插入一个队列中,注意头结点不是该节点,而是new了一个新的node节点,它的状态值为0,然后返回该节点。

    具体代码不做分析,下面看acquireQueued方法

     1  final boolean acquireQueued(final Node node, int arg) {
     2         boolean failed = true;
     3         try {
     4             boolean interrupted = false;
     5             // 类似while(true),作无限循环作用
     6             for (;;) {
     7                 // 获取插入的node节点的前一个节点
     8                 final Node p = node.predecessor();
     9                 // 如果前继节点为head节点并且获取锁成功,则跳出无限循环,执行相应业务代码
    10                 if (p == head && tryAcquire(arg)) {
    11                     setHead(node);// 头结点被改变,改变同时其状态也被改变了,节点线程也为空
    12                     p.next = null; // help GC
    13                     failed = false;
    14                     return interrupted;
    15                 }
    16                 // 前继节点不是头结点或获取锁失败
    17                 if (shouldParkAfterFailedAcquire(p, node) &&
    18                     parkAndCheckInterrupt())
    19                     interrupted = true;
    20             }
    21         } finally {
    22             // 防止代码运行过程中,线程突然被中断,中断则将该节点置为取消状态
    23             if (failed)
    24                 cancelAcquire(node);
    25         }
    26     }

    其中shouldParkAfterFailedAcquire方法做了这两件事,

    1、如果p节点前有状态为cancel的节点,则将这些取消的节点放弃掉,简单来说就是排除取消的节点

    2、将p节点状态置为signal状态。等待下一次进入该方法可能会被挂起

    方法parkAndCheckInterrupt,在shouldParkAfterFailedAcquire返回true的时候,线程会被挂起。、

    以上就是获取锁的过程,步骤如下

    1、获取锁成功,则将改变state值,并将锁的拥有者置为当前线程

    2、获取锁失败,则进入同步队列中,直到获取锁成功或当前线程被外因给中断,获取锁的过程中,有的线程可能会被挂起。

    2、非公平锁释放锁的过程

    为了不显得过于啰嗦,下面只列出核心代码

    上述代码只有获取锁的线程调用了unlock方法,才会去修改state值,当state值为0时其他线程又可以获取锁,看到这,或许有的小伙伴迷糊了,上面不是介绍说在获取锁的过程中,有的线程会被挂起,那如果挂起的线程node节点前继恰好是头结点,那岂不是运行不了?,莫慌,往下看。当state值置为0时,该方法会返回true,之后会执行下面方法。

     重点方法在unparkSuccessor方法上,看if(h != null && h.waitStatus !=0) ,为什么要加这个判断呢,因为如果有多个线程在获取锁,无论是获取失败,还是获取成功head节点的状态值都被改变(setHead()和shouldParkAfterFailedAcquire()方法会去改变head节点状态)。即不为0,如果为0,那么就说明就没有线程被挂起,自然就不用去释放这些线程。加这个判断,为了减少无用操作。重点来了,unparkSuccessor方法,代码如下

    private void unparkSuccessor(Node node) {
            // 将node结点状态置为0
            int ws = node.waitStatus;
            if (ws < 0)
                compareAndSetWaitStatus(node, ws, 0);
    
            /* 
             * 如果node结点下一个节点为null或被取消则进入下面的for循环
             *    下面的for循环从尾节点往前寻找没有取消的节点 ,直至最靠近node节点,即node节点下一个状态小于等于0的节点
             *  在这里node节点就是头结点,
             */
            Node s = node.next;
            if (s == null || s.waitStatus > 0) {
                s = null;
                for (Node t = tail; t != null && t != node; t = t.prev)
                    if (t.waitStatus <= 0)
                        s = t;
            }
           // 找到了该节点,释放该节点的线程
            if (s != null)
                LockSupport.unpark(s.thread);
        }    

    或许看到这更迷糊了,它释放锁怎么能确定释放的就是那个被挂起的线程呢,这个呢,确实确定不了,但是如果释放前继节点为头结点的线程,那么在后续获取锁的过程中,该线程肯定能获取到锁(因为这段代码是前一个线程释放锁的操作代码,所以下一个线程肯定能获取到锁),至此又一轮循环。

    在这里,我对那个为啥从尾节点向前遍历也不清楚,如果有清楚的小伙伴,还请评论下方留言,谢谢!

    以上就是非公平锁的释放操作。

    3、公平锁的获取锁过程

    该种锁和非公平锁的不同之处,就是这种锁一定得按照顺序来获取或,不能前一个线程释放了锁 ,然后谁抢到了就算谁的。

    先来看下这种获取锁的代码

     1 protected final boolean tryAcquire(int acquires) {
     2             final Thread current = Thread.currentThread();
     3             int c = getState();
     4             if (c == 0) {
     5                 if (!hasQueuedPredecessors() &&
     6                     compareAndSetState(0, acquires)) {
     7                     setExclusiveOwnerThread(current);
     8                     return true;
     9                 }
    10             }
    11             else if (current == getExclusiveOwnerThread()) {
    12                 int nextc = c + acquires;
    13                 if (nextc < 0)
    14                     throw new Error("Maximum lock count exceeded");
    15                 setState(nextc);
    16                 return true;
    17             }
    18             return false;
    19         }

    和非公平锁的不同点是在前者线程释放锁后(即state值为0),非公平锁是谁抢到锁,锁就是谁的,但是公平锁不一样,获取锁的线程会先去判断同步队列中有没有其他线程,如果没有,再去试着改变state值,如果改变成功则获取锁成功,它不允许没进入同步队列中的线程(此时同步队列中已有等待的线程,如果没有,那就是直接抢)抢占锁。下面看下hasQueuedPrdecessor(),代码如下

     1 public final boolean hasQueuedPredecessors() {
     2         // The correctness of this depends on head being initialized
     3         // before tail and on head.next being accurate if the current
     4         // thread is first in queue.
     5         Node t = tail; // Read fields in reverse initialization order
     6         Node h = head;
     7         Node s;
     8         return h != t &&
     9             ((s = h.next) == null || s.thread != Thread.currentThread());
    10     }

    代码不复杂,就是判断同步队列中有没有等待的线程,且等待的线程不是当前线程,有则返回true,没有则返回false。

    至于公平锁的释放操作,和非公平锁一致。这里不过多叙述。

    获取公平锁操作

    1、先判断同步队列中有没有等待的线程。

    2、有则放弃锁的争夺,进入同步队列排好队,没有则抢占锁

     ----------------------------------------------------------------------------------------------------华丽的分界线---------------------------------------------------------------------------------------------------------------------------------

     本来想继续写condition,但好像篇幅有点啰嗦,就放在下一篇。

    以上就是我的个人见解,如果不足或错误之处,还请指教,谢谢!

  • 相关阅读:
    org.apache.zookeeper.ClientCnxn
    Oracle数据库 number 长度与 Short Integer Long BigDecimal 对应关系
    Spring中 PROPAGATION_REQUIRED 解释
    java的static final和final的区别
    本地调试有数据,发布到服务器没更新的问题
    java的double类型如何精确到一位小数?
    consul配置参数大全、详解、总结
    深入对比TOML,JSON和YAML
    基于Jenkins,docker实现自动化部署(持续交互)【转】
    让Linux任务在后台可靠运行的几种方法
  • 原文地址:https://www.cnblogs.com/qm-article/p/10011485.html
Copyright © 2020-2023  润新知