ReentrantLock

本文详细剖析了ReentrantLock的工作原理,包括其内部实现机制、公平性和非公平性锁的区别,以及通过AQS(AbstractQueuedSynchronizer)实现线程间的同步与等待。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

title:ReentrantLock

date:2017年11月12日12:40:32


​ 在生产者消费者问题中,我们提到了利用锁和Condition条件来解决问题。

今天我们就来看下锁是怎么实现的。

private final Lock lock=new ReentrantLock();
lock.lock();

ReentrantLock锁在同一个时间点只能被一个线程锁持有;而可重入的意思是,ReentrantLock锁,可以被单个线程多次获取。
ReentrantLock分为“公平锁”和“非公平锁”。它们的区别体现在获取锁的机制上是否公平。“锁”是为了保护竞争资源,防止多个线程同时操作线程而出错,ReentrantLock在同一个时间点只能被一个线程获取(当某线程获取到“锁”时,其它线程就必须等待);ReentraantLock是通过一个FIFO的等待队列来管理获取该锁所有线程的。在“公平锁”的机制下,线程依次排队获取锁;而“非公平锁”在锁是可获取状态时,不管自己是不是在队列的开头都会获取锁。

当new RReentrantLock();的时候,默认使用非公平锁。

  private final Sync sync;

  public ReentrantLock() {
        sync = new NonfairSync();
    }

当创建一个实例的时候,在构造方法里里面创建了一个NonfairSync对象。

NonfairSync是ReentrantLock中的一个静态内部类,他继承了Sync类。而Sync是ReentrantLock中的一个静态类,而且是abstract修饰的抽象类,他继承了AbstractQueuedSynchronizer,这个类就是AQS,

它是一个非常有用的超类,可用来定义锁以及依赖于排队阻塞线程的其他同步器;ReentrantLock,ReentrantReadWriteLock,CountDownLatch,CyclicBarrier和Semaphore等这些类都是基于AQS类实现的。AbstractQueuedLongSynchronizer 类提供相同的功能但扩展了对同步状态的 64 位的支持。两者都扩展了类 AbstractOwnableSynchronizer(一个帮助记录当前保持独占同步的线程的简单类)。

    abstract static class Sync extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = -5179523762034025860L;

        /**
         * Performs {@link Lock#lock}. The main reason for subclassing
         * is to allow fast path for nonfair version.
         */
        abstract void lock();

        /**
         * Performs non-fair tryLock.  tryAcquire is implemented in
         * subclasses, but both need nonfair try for trylock method.
         */
        final boolean nonfairTryAcquire(int acquires) {
            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;
        }

        protected final boolean tryRelease(int releases) {
            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;
        }

        protected final boolean isHeldExclusively() {
            // While we must in general read state before owner,
            // we don't need to do so to check if current thread is owner
            return getExclusiveOwnerThread() == Thread.currentThread();
        }

        final ConditionObject newCondition() {
            return new ConditionObject();
        }

        // Methods relayed from outer class

        final Thread getOwner() {
            return getState() == 0 ? null : getExclusiveOwnerThread();
        }

        final int getHoldCount() {
            return isHeldExclusively() ? getState() : 0;
        }

        final boolean isLocked() {
            return getState() != 0;
        }

        /**
         * Reconstitutes the instance from a stream (that is, deserializes it).
         */
        private void readObject(java.io.ObjectInputStream s)
            throws java.io.IOException, ClassNotFoundException {
            s.defaultReadObject();
            setState(0); // reset to unlocked state
        }
    }
  • 接着我们来看看lock.lock()到底的具体实现是什么:
  public void lock() {
        sync.lock();
    }

默认是使用的非公平锁,则sync.lock()的具体实现为:

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

我们分析下这个方法

lock()会先通过compareAndSet(0, 1)来判断“锁”是不是空闲状态。是的话,“当前线程”直接获取“锁”;否则的话,调用acquire(1)获取锁。

(01) compareAndSetState()是CAS函数,它的作用是比较并设置当前锁的状态。若锁的状态值为0,则设置锁的状态值为1。

compareAndSetState()在AQS中实现。它的源码如下:

        // See below for intrinsics setup to support this
        return unsafe.compareAndSwapInt(this, stateOffset, expect, update);
    }

compareAndSwapInt() 是sun.misc.Unsafe类中的一个本地方法。对此,我们需要了解的是 compareAndSetState(expect, update) 是以原子的方式操作当前线程;若当前线程的状态为expect,则设置它的状态为update。我们要明白AQS在实现锁和阻塞队列时时非常有用的类。

(02) setExclusiveOwnerThread(Thread.currentThread())的作用是,设置“当前线程”为“锁”的持有者。

  • acquire()方法

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

    acquire()方法也是在AQS中实现的,不得不说,在任何框架,甚至JDK的设计中,都大量的运用到了模板方法设计模式。

    • 当前前程首先通过tryAcquire()尝试获取锁,获取成功的话,直接返回。获取失败的话,

    进入到等待队列依次排序,然后获取锁。

    这里再说一下这句代码的设计:

     if (!tryAcquire(arg) &&
                acquireQueued(addWaiter(Node.EXCLUSIVE), arg))

    这里if里面的是与逻辑,我们在学习&&逻辑符号的时候就说过,&&具有短路功能,也就是说只有第一个分支是true才会执行第二个,如果第一个分支(这里是!tryAcquire(arg))是false的话,就会直接跳出if了,而在这里就是直接返回了。这里就有点类似于if return 提前返回。

    if(flag){
      return;
    }
    ...
    ....

    这里JDK这样的设计也是蛮有灵性的。

  • tryAcquire()方法

       protected final boolean tryAcquire(int acquires) {
              return nonfairTryAcquire(acquires);
          }

    tryAcquire()方法在AOS中是一个抽象方法,在非公平锁中,具体实现在NonfairSync这个静态内部类中。

    protected final boolean tryAcquire(int acquires) {
              return nonfairTryAcquire(acquires);
          }
          final boolean nonfairTryAcquire(int acquires) {
            //获取当前线程
              final Thread current = Thread.currentThread();
            //获取锁的状态  
            int c = getState();
               // c=0意味着锁没有被任何线程锁拥有
    
              if (c == 0) {
                // 若“锁没有被任何线程锁拥有”,则通过CAS函数设置锁的状态为acquires。
          // 同时,设置当前线程为锁的持有者。       
                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;
          } 
    

    • getState(), setState()

    getState()和setState()都在AQS中实现,源码如下:

    // 锁的状态
    private volatile int state;
    // 设置锁的状态
    protected final void setState(int newState) {
        state = newState;
    }
    // 获取锁的状态
    protected final int getState() {
        return state;
    }

    state表示锁的状态,对于“独占锁”而已,state=0表示锁是可获取状态(即,锁没有被任何线程锁持有)。由于java中的独占锁是可重入的,state的值可以>1。

    tryAcquire()的作用就是让当前线程尝试获取锁。获取成功返回true,失败则返回false。返回false之后,if的第一分支就是true,此时会执行第二分支 acquireQueued(addWaiter(Node.EXCLUSIVE), arg)

  • addWaite()方法

    在说acquireQueued()方法之前需要说明下他的第一个参数的方法addWaiter()

    addWaiter()方法的实现是在AQS中。

      private Node addWaiter(Node mode) {
        //新建一个Node节点,node节点对应的是当前线程,当前线程锁的模型是Node.EXCLUSIVE
          Node node = new Node(Thread.currentThread(), mode);
          // Try the fast path of enq; backup to full enq on failure
    //取CLH队列的末尾元素      
          Node pred = tail;
        //如果末尾元素不为null,即CLH队列不为空,则将node节点插入在CLH阻塞队列末尾
          if (pred != null) {
              node.prev = pred;
              if (compareAndSetTail(pred, node)) {
                  pred.next = node;
                  return node;
              }
          }
            // 若CLH队列为空,则调用enq()新建CLH队列,然后再将当前线程添加到CLH队列中。
    
          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;
                  }
              }
          }
      }

​ 1.compareAndSetTail()

 private final boolean compareAndSetTail(Node expect, Node update) {
        return unsafe.compareAndSwapObject(this, tailOffset, expect, update);
    }

compareAndSetTail()也是在AQS中实现的,也属于CAS函数,也是通过本地native方法实现的。compareAndSetTail(expect, update)会以原子的方式进行操作,它的作用是判断CLH队列的队尾是不是为expect,是的话,就将队尾设为update。

​ 2.enq()

enq()的作用很简单。如果CLH队列为空,则新建一个CLH表头;然后将node添加到CLH末尾。否则,直接将node添加到CLH末尾

​ 3.Node类

Node就是CLH队列的节点。Node在AQS中实现,是AQS的静态内部类

注: CLH队列是AQS中“等待锁”的线程队列。在多线程中,为了保护竞争资源不被多个线程同时操作而起来错误,我们常常需要通过锁来保护这些资源。在独占锁中,竞争资源在一个时间点只能被一个线程锁访问;而其它线程则需要等待。CLH就是管理这些“等待锁”的线程的队列。
    CLH是一个非阻塞的 FIFO 队列。也就是说往里面插入或移除一个节点的时候,在并发条件下不会阻塞,而是通过自旋锁和 CAS 保证节点插入和移除的原子性。

CAS函数:

CAS函数,是比较并交换函数,它是原子操作函数;即,通过CAS操作的数据都是以原子方式进行的。例如,compareAndSetHead(), compareAndSetTail(), compareAndSetNext()等函数。它们共同的特点是,这些函数所执行的动作是以原子的方式进行的。

private transient volatile Node head;    // CLH队列的队首
private transient volatile Node tail;    // CLH队列的队尾

// CLH队列的节点
static final class Node {
    static final Node SHARED = new Node();
    static final Node EXCLUSIVE = null;

    // 线程已被取消,对应的waitStatus的值
    static final int CANCELLED =  1;
    // “当前线程的后继线程需要被unpark(唤醒)”,对应的waitStatus的值。
    // 一般发生情况是:当前线程的后继线程处于阻塞状态,而当前线程被release或cancel掉,因此需要唤醒当前线程的后继线程。
    static final int SIGNAL    = -1;
    // 线程(处在Condition休眠状态)在等待Condition唤醒,对应的waitStatus的值
    static final int CONDITION = -2;
    // (共享锁)其它线程获取到“共享锁”,对应的waitStatus的值
    static final int PROPAGATE = -3;

    // waitStatus为“CANCELLED, SIGNAL, CONDITION, PROPAGATE”时分别表示不同状态,
    // 若waitStatus=0,则意味着当前线程不属于上面的任何一种状态。
    volatile int waitStatus;

    // 前一节点
    volatile Node prev;

    // 后一节点
    volatile Node next;

    // 节点所对应的线程
    volatile Thread thread;

    // nextWaiter是“区别当前CLH队列是 ‘独占锁’队列 还是 ‘共享锁’队列 的标记”
    // 若nextWaiter=SHARED,则CLH队列是“独占锁”队列;
    // 若nextWaiter=EXCLUSIVE,(即nextWaiter=null),则CLH队列是“共享锁”队列。
    Node nextWaiter;

    // “共享锁”则返回true,“独占锁”则返回false。
    final boolean isShared() {
        return nextWaiter == SHARED;
    }

    // 返回前一节点
    final Node predecessor() throws NullPointerException {
        Node p = prev;
        if (p == null)
            throw new NullPointerException();
        else
            return p;
    }

    Node() {    // Used to establish initial head or SHARED marker
    }

    // 构造函数。thread是节点所对应的线程,mode是用来表示thread的锁是“独占锁”还是“共享锁”。
    Node(Thread thread, Node mode) {     // Used by addWaiter
        this.nextWaiter = mode;
        this.thread = thread;
    }

    // 构造函数。thread是节点所对应的线程,waitStatus是线程的等待状态。
    Node(Thread thread, int waitStatus) { // Used by Condition
        this.waitStatus = waitStatus;
        this.thread = thread;
    }
}

Node是CLH队列的节点,代表“等待锁的线程队列”。
(01) 每个Node都会一个线程对应。
(02) 每个Node会通过prev和next分别指向上一个节点和下一个节点,这分别代表上一个等待线程和下一个等待线程。
(03) Node通过waitStatus保存线程的等待状态。
(04) Node通过nextWaiter来区分线程是“独占锁”线程还是“共享锁”线程。如果是“独占锁”线程,则nextWaiter的值为EXCLUSIVE;如果是“共享锁”线程,则nextWaiter的值是SHARED。

  • acquireQueued()方法

    该方法依旧是在AQS中实现的:

    final boolean acquireQueued(final Node node, int arg) {
          boolean failed = true;
          try {
              boolean interrupted = false;
            //使用无限循坏,直到return返回
              for (;;) {
                //定位到当前线程对应节点在CLH的前继节点,
                  final Node p = node.predecessor();
                //如果前继节点是头结点,且当前锁尝试获取锁成功,才会return,才会退出这个方法
               //将当前节点设置为头节点,返回false,即第二个逻辑判断分支也为false,     //lock.lock()得以返回,因为在tryAcquire(arg)中以及成功获取锁了
                  if (p == head && tryAcquire(arg)) {
                        //设置为头结点
                    setHead(node);
                      p.next = null; // help GC
                      failed = false;
                      return interrupted;
                  }
                //否则,执行shouldParkAfterFailedAcquire,这个方法需要具体分析
                  if (shouldParkAfterFailedAcquire(p, node) &&
                      parkAndCheckInterrupt())
                      interrupted = true;
              }
          } finally {
              if (failed)
                  cancelAcquire(node);
          }
      }

    由上面代码中的分析可见,acquireQueued()的目的是从CLH队列中获取锁。

  • shouldParkAfterFailedAcquire()

shouldParkAfterFailedAcquire


private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
        //取出当前线程前继节点表示的状态
        int ws = pred.waitStatus;
  //如果前继节点是SIGNAL状态,则意味这当前线程需要被unpark唤醒。此时,返回true。
        if (ws == Node.SIGNAL)
            /*
             * This node has already set status asking a release
             * to signal it, so it can safely park.
             */
            return true;
        if (ws > 0) {
            /*
             * Predecessor was cancelled. Skip over predecessors and
             * indicate retry.
             */
             // 如果前继节点是“取消”状态,则设置 “当前节点”的 “当前前继节点”  为  “‘原前继节点’的前继节点”。就是删除了前继节点

            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.
             */
        // 如果前继节点为“0”或者“共享锁”状态,则设置前继节点为SIGNAL状态。显然这也是一个CAS函数

            compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
        }
        return false;
    }

shouldParkAfterFailedAcquire()方法是通过如下流程判断当前线程是否需要被阻塞的。

1、如果前继节点状态为SIGNAL,表明当前节点需要被unpark(唤醒),此时则返回true2、如果前继节点状态为CANCELLED(ws>0),说明前继节点已经被取消,则通过先前回溯找到一个有效(非CANCELLED状态)的节点,并返回false3、如果前继节点状态为非SIGNAL、非CANCELLED,则设置前继的状态为SIGNAL,并返回false

如果1发生了返回true,则会执行parkAndCheckInterrupt()来阻塞当前线程。

  • parkAndCheckInterrupt()

    该方法依然是在AQS中实现的。

    private final boolean parkAndCheckInterrupt() {
          LockSupport.park(this);
          return Thread.interrupted();
      }

    parkAndCheckInterrupt()的作用是阻塞当前线程,并且返回线程被唤醒之后的中断状态。
    它会先通过LockSupport.park()阻塞当前线程,然后通过Thread.interrupted()返回线程的中断状态。注意:Thread.interrupted()方法会清空线程中断状态。

    这里介绍一下线程被阻塞之后如何唤醒。一般有2种情况:
    第1种情况:unpark()唤醒。“前继节点对应的线程”使用完锁之后,通过unpark()方式唤醒当前线程。
    第2种情况:中断唤醒。其它线程通过interrupt()中断当前线程。

    补充:LockSupport()中的park(),unpark()的作用 和 Object中的wait(),notify()作用类似,是阻塞/唤醒。
    它们的用法不同,park(),unpark()是轻量级的,而wait(),notify()是必须先通过Synchronized获取同步锁。

  • selfInterrupt()

    selfInterrupt() 依然是在AQS中实现的,这里是让当前线程自行中断,当调用parkAndCheckInterrupt()返回true后,会执行这个方法,这个方法就是让线程再中断一次,之所以需要重新中断一次,是因为parkAndCheckInterrupt()中调用Thread.interrupted会清空中断标志,即在该线程“成功获取锁并真正执行起来”之前,它的中断会被忽略并且中断标记会被清除!

      static void selfInterrupt() {
          Thread.currentThread().interrupt();
      }
  • 总结

    再回过头看看acquire()函数,它最终的目的是获取锁!

(01) 先是通过tryAcquire()尝试获取锁。获取成功的话,直接返回;尝试失败的话,再通过acquireQueued()获取锁。
(02) 尝试失败的情况下,会先通过addWaiter()来将“当前线程”加入到”CLH队列”末尾;然后调用acquireQueued(),在CLH队列中排序等待获取锁,在此过程中,线程处于休眠状态。直到获取锁了才返回。 如果在休眠等待过程中被中断过,则调用selfInterrupt()来自己产生一个中断。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值