简单看看ReentrantLock

  前面我们分析了AQS的基本原理,然后也试着基于AQS实现了一个可重入的锁了,现在我们再来看看官方的ReentrantLock锁,这个锁是可重入的独占锁,也就是说同时只有一个线程可以获取该锁,而且这个线程还能继续尝试获取锁;

 

一.简单的使用

  我们先根据ReentrantLock来简单实现一个线程安全的List,然后再分析常用的方法;

package com.example.demo.study;

import java.util.ArrayList;
import java.util.concurrent.locks.ReentrantLock;

public class Study0204 {
    //线程不安全的List
    private ArrayList<String> list = new ArrayList<String>();
    //独占锁,默认是非公平锁,传入true可以是公平锁
    private volatile ReentrantLock lock = new ReentrantLock();
    
    //往集合中添加元素
    public void add(String str) {
        lock.lock();
        try {
            list.add(str);
        } finally {
            lock.unlock();
        }
    }
    //删除集合中的元素
    public void remove(String str) {
        lock.lock();
        try {
            list.remove(str);
        } finally {
            lock.unlock();
        }
    }
    //根据索引获取集合中某个元素
    public String get(int index) {
        lock.lock();
        try {
            return list.get(index);
        } finally {
            lock.unlock();
        }
    }
}

  其实就是在每一步都会进行一个加锁的过程,用法和synchronized关键字一样,但是要注意一定要在finally中释放锁,而且不知道大家有没有发现一个问题,在get方法中,需要用锁吗?为什么多线程去读数据也要加锁呢,又没有改变集合中的数据,这也是ReentrantLock这个锁的一个缺陷,使用于写多读少的情况;

  所以后面我们会说一下ReentrantReadWriteLock这个锁,这个锁应用于读多写少的情景,可以把读锁和写锁分开使用,如果是读数据可以多个线程都可以获取读锁,后面会说到的

 

二.看看ReentrantLock锁结构

  这个锁其实也很简单,下图所示:实现Lock接口,还有一个内部工具类Sync继承自AQS,然后还有两个类NonfairSync和FairSync继承Sync实现自己的方法,看名字就知道这两个类其实就是非公平锁和公平锁的实现策略;

 

  首先我们看看ReentrantLock的lock方法,发现就是调用sync的lock方法,所以下一步我们看看这个sync对象是怎么构建出来的;

  

  看看构造器,可以知道根据传入的参数是true和false构建sync对象是公平策略还是非公平策略;顺便一说,在这里AQS中的state表示的是锁的可重入次数,默认情况state为0;

  当第一次一个线程CAS成功获取了该锁,那么state就加一,当这个已经获取该锁的线程继续获取锁,state继续加一,释放锁state就减一,直到变为0,那么说明该锁就没有线程占有了,此时AQS中阻塞队列中的某个线程就可以获取锁了;

 

  我们可以看看这里的Sync实现了一些什么方法,下图所示,只有lock()方法没有实现,留给NonfairSync和FairSync各自根据自己的场景去实现;

 

 

  公平锁策略实现如下,实现了lock方法还有tryAcquire()方法,其中这个tryAcquire方法以前说活是AQS中特意留给子类根据实际场景去实现的

 

  非公平策略实现如下,和公平策略实现的方法一样;

 

三.非公平策略获取锁

  前面说了ReentrantLock的基本结构,然后我们分析看看是怎么获取锁的,从NonfairSync中的lcok方法开始,我们梳理一下:首先当前线程A使用CAS尝试将AQS中的state从0设置为1,如果成功,那就直接将当前线程设置为锁的占有者;失败的话,那就去获取state的值,如果为0,还是用CAS设置为1,将当前线程设置为锁的占有者,不为0,那就看看占有锁的是不是当前线程;如果是当前线程,就将state加一;不是当前线程,那就是其他线程占用该锁,这才真正说明当前线程获取锁失败,我们才将当前线程封装成一个Node.EXCLUSIVE类型的节点,丢到阻塞队列中去;

  这里我们需要想一下为什么说这里是非公平的?假如在下面nonfairTryAcquire方法中,线程A获取到的c的值为1,占有锁的不是线程A,那么线程就被封装成节点丢到阻塞队列中去了,这个时候线程B来获取c的值,刚好是0(因为这个时候可能那个占有了该锁的线程释放了锁),于是线程B就能成功的将state从0变为1,能获得该锁;

  虽然线程A先尝试获取该锁,线程B后去获取,然而线程B居然可以先成功获取到锁,就好像你排队吃饭,一个后来的人居然先打到饭去吃了,真的是日了狗了!

final void lock() {
    //尝试通过CAS将state从0改为1,如果成功那就将当前线程占用该锁
    if (compareAndSetState(0, 1))
        setExclusiveOwnerThread(Thread.currentThread());
    //CAS失败,说明有其他线程已经获取锁了,于是就将当前线程丢到阻塞队列中去,重点看看acquire方法
    else
        acquire(1);
}

//主要看tryAcquire方法,在NonfairSync中实现
//当前方法中,if判断中第二个条件已经看过了,主要是将当前线程封装成一个Node.EXCLUSIVE类型的节点
//然后丢到阻塞队列的最后面去
public final void acquire(int arg) {
    if (!tryAcquire(arg) && acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
        selfInterrupt();
}

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

final boolean nonfairTryAcquire(int acquires) {
    final Thread current = Thread.currentThread();
    //获取state的值
    int c = getState();
    //state的值位0的话,那就CAS设置为1,并且设置当前线程占用该锁
    if (c == 0) {
        if (compareAndSetState(0, acquires)) {
            setExclusiveOwnerThread(current);
            return true;
        }
    }
    //CAS设置失败,肯定有其他线程占用该锁,由于是可重入锁,所以这里先判断占有该锁的是不是当前线程
    else if (current == getExclusiveOwnerThread()) {
        //是,那就将state加一
        int nextc = c + acquires;
        if (nextc < 0) // overflow
            throw new Error("Maximum lock count exceeded");
        setState(nextc);
        return true;
    }
    //能到这里,说明state的值不是0,占有锁的也不是当前线程,肯定是其他线程,那就返回false
    return false;
}

 

四.公平策略获取锁

  其实公平策略和非公平策略的实现基本一样,我们只看看tryAcquire方法的实现:

protected final boolean tryAcquire(int acquires) {
    final Thread current = Thread.currentThread();
    int c = getState();
    if (c == 0) {
        //重点是这里不同,不会直接去通过CAS获取锁,而是调用hasQueuedPredecessors方法查看阻塞队列中有没有前驱节点,
        //这个方法是核心,如果阻塞队列中当前线程节点的前面有节点,那么这里不会进去,直接会走下面的else if
        //如果前面没有节点,那么说明当前阻塞队列为null或者只有当前线程这一个节点,那就可以获取CAS修改state,获取锁成功
        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;
}

//判断阻塞队列中当前线程节点前面有没有节点?
public final boolean hasQueuedPredecessors() {

    Node t = tail; // Read fields in reverse initialization order
    Node h = head;
    Node s;
    //注意,头节点head指向的是哨兵节点
    //如果头节点和尾节点一样即h==t,前面博客画图都说了都指向哨兵节点,此时阻塞队列为空,没有前驱节点,就返回false
    //如果h != t同时(s = h.next) == null,说明阻塞队列中哨兵节点后面正在插入一个节点,此时表示有前驱节点,返回true
    //如果h != t同时(s = h.next) != null,而且s.thread != Thread.currentThread(),说明哨兵节点后面有一个节点,而且这个节点还不是当前线程节点
    //也就是说有前驱节点,返回true
    return h != t && ((s = h.next) == null || s.thread != Thread.currentThread());
}

 

 

五.其他一些方法

  上面我们看了公平策略和非公平策略的实现方式,其实没什么,比较容易,继续看看一些其他的方法,这些方法是通用的;

  1.lockInterruptibly()方法

//前面说过加了Interruptibly表示如果当前线程在调用该方法时,其他线程调用了当前线程的interrupt()方法的时候,那么
//当前线程就会抛出InterruptedException异常,我们可以看看内在的机制是怎么实现的
public void lockInterruptibly() throws InterruptedException {
    sync.acquireInterruptibly(1);
}

public final void acquireInterruptibly(int arg) throws InterruptedException {
    //如果当前线程被中断,就抛出异常
    if (Thread.interrupted())
        throw new InterruptedException();
    //尝试获取资源,这里分为公平策略和非公平策略,前面已经说过了;
    //获取资源失败的话,就调用AQS可被中断的方法
    if (!tryAcquire(arg))
        doAcquireInterruptibly(arg);
}

 

 

  2.tyLock()方法

//很明显这个tryLock方法是非公平策略
public boolean tryLock() {
    return sync.nonfairTryAcquire(1);
}
//这个方法在前面说非公平策略的时候tryAcquire方法调用的也是这个方法
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;
}

  

  3.tryLock(long timeout, TimeUnit unit)方法

  这个方法和上面的方法不一样的就是可以设置超时时间,其实和前面说的也差不多,就是多了一个时间的判断,这个也是会对线程中断有响应的;

public boolean tryLock(long timeout, TimeUnit unit)
        throws InterruptedException {
    return sync.tryAcquireNanos(1, unit.toNanos(timeout));
}

public final boolean tryAcquireNanos(int arg, long nanosTimeout) throws InterruptedException {
    if (Thread.interrupted())
        throw new InterruptedException();
    return tryAcquire(arg) ||
        doAcquireNanos(arg, nanosTimeout);
}

 

 

  4.unlock()释放锁

public void unlock() {
    sync.release(1);
}

public final boolean release(int arg) {
    //从下面的方法我们可以知道tryRelease只有当state为0的时候才会返回true,也就是当前锁没有被线程持有
    //此时,如果头节点不为null,而且头节点的waitStatus不为初始状态,就唤醒头节点
    if (tryRelease(arg)) {
        Node h = head;
        if (h != null && h.waitStatus != 0)
            unparkSuccessor(h);
        return true;
    }
    return false;
}

protected final boolean tryRelease(int releases) {
    //将AQS中的state减一
    int c = getState() - releases;
    //当前线程如果不是锁的拥有者,却调用了unlock方法,那么就会抛出IllegalMonitorStateException异常
    //在这里就能判断了如果原先的state为0,那么上面的c应该就是负一,就会走到这里来抛错,如果state为1,就会到下面的if(c==0)里面
    //如果state大于1,那么只会到最下面的setState(c)将减一之后的state更新到AQS中
    if (Thread.currentThread() != getExclusiveOwnerThread())
        throw new IllegalMonitorStateException();
    boolean free = false;
    //可重入数字为0,就将当前持有锁的线程设置为null
    if (c == 0) {
        free = true;
        setExclusiveOwnerThread(null);
    }
    //这里不管重入数字是不是0,我们只会将最新的state更新获取
    setState(c);
    return free;
}

 

 

六.总结

  我们已经看到了ReentrantLock这个锁的基本功能,其实就是一个独占锁,而且是可重入的,这里的重入指的是一个已经占有了该锁的线程,还可以继续获取该锁!

  下图所示,同时有三个线程去争夺ReentrantLock锁,此时,只有Thread1成功占有锁了,那么其他的两个线程就被丢到AQS阻塞队列中去了(这里是有顺序的,先是Thread2,然后是Thread3);

 

 

  如果这个时候Thread1调用了条件变量1的await方法,那么Thread1就被丢到条件队列1中,并且释放锁,这个时候阻塞队列中就会有一个线程可以获取锁,如果是公平锁,那么就是阻塞队列中最前面的那一个获取锁,此时阻塞队列中只有Thread3了;(如果是非公平锁,那么就是看运气,Thread2和Thread那个先去尝试获取锁那么就获得锁)

posted @ 2020-02-05 14:39  java小新人  阅读(366)  评论(0编辑  收藏  举报