ReentantLock核心代码

Lock接口

Reentrant实现了Lock接口,Lock接口主要定义了如下几个方法:

//加锁
void lock();
//获取锁
boolearn tryLock();
//获取锁(设置超时等待时间)
boolearn tryLock(long time, TimeUnit unit) throw InterruptedException;
//释放锁
void unlock();

ReentrantLock结构

ReentrantLock结构

默认构造创建的是一个非公平锁,可以通过参数控制创建的是公平的还是非公平的;

    //默认的构造为非公平重入锁
    public ReentrantLock() {
        sync = new NonfairSync();
    }
    
    //根据fair参数判断创建的是公平锁还是非公平
    public ReentrantLock(boolean fair) {
        sync = fair ? new FairSync() : new NonfairSync();
    }

公平锁:按照线程先来后到顺序获取锁,后到的线程只能等前面的线程都获取锁完毕才执行获取锁的操作,执行有序
非公平锁:不按照线程先来后到的时间顺序进行竞争锁,后到的线程也能够获取到锁

ReentrantLock内部类:NonfairSync,FairSync

Reentrant都是通过NonfairSync和FairSync来实现锁的功能,它俩的父类Sync则继承了AbstractQueuedSynchronizer,AQS是JUC框架核心。

加锁lock()

//公平锁
//static final class FairSync extends Sync
final void lock() {
  acquire(1);
}

//非公平锁
// static final class NonfairSync extends Sync {
final void lock() {
  //CAS更新同步状态值state,锁占有线程设置为当前线程
  if (compareAndSetState(0, 1))
    setExclusiveOwnerThread(Thread.currentThread());
  else
    acquire(1);
}

先看一下acquire(1)方法,此方法定义在Sync的父类AbstractQueuedSynchronizer中

//AbstractQueuedSynchronizer.java
//获取锁,此部分逻辑定义在父类中,NonfairSync和FairSync通用,只是tryAcquire(arg) 有不同的实现
public final void acquire(int arg) {
        //获取锁失败  &&  acquireQueued方法会将线程排队到队列并且返回线程是否需要中断
        if (!tryAcquire(arg) && acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            selfInterrupt();
}

/**
* 中断当前线程
**/
static void selfInterrupt() {
        Thread.currentThread().interrupt();
}

公平锁的tryAcquire实现

//static final class FairSync extends Sync
//尝试获取锁的逻辑(公平锁),获取成功返回true,失败返回false
protected final boolean tryAcquire(int acquires) {
    //获取当前线程
    final Thread current = Thread.currentThread();
    //获取同步状态的当前值
    int c = getState();
    if (c == 0) {
        //判断前面是否有排队的线程,没有的话CAS设置线程的state值为1
        //与非公平锁的获取锁实现唯一区别的地方
        //非公平锁不需要hasQueuedPredecessors() 判断前面是否有排队的线程
        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;
}

非公平锁的tryAcquire实现

//Sync object for non-fair locks
 protected final boolean tryAcquire(int acquires) {
    return nonfairTryAcquire(acquires);
 }

//nonfairTryAcquire实现在父类Sync中
final boolean nonfairTryAcquire(int acquires) {
    //获取当前线程
    final Thread current = Thread.currentThread();
    //获取同步状态值
    int c = getState();
    if (c == 0) {
        //与公平锁的获取锁实现唯一区别的地方
        //公平锁要hasQueuedPredecessors() 判断前面是否有排队的线程
        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;
}

tryAcquire(arg)返回false的话,执行acquireQueued进行添加到队列,此方法定义在AbstractQueuedSynchronizer中,参数需要addWaiter(Node.EXCLUSIVE)返回的Node。

//AbstractQueuedSynchronizer.java
/**
* 为当前线程和给定模式创建和排队节点
**/
private Node addWaiter(Node mode) {
    //创建当前线程节点
    Node node = new Node(Thread.currentThread(), mode);
    Node pred = tail;
    if (pred != null) {
        node.prev = pred;
        if (compareAndSetTail(pred, node)) {
            pred.next = node;
            return node;
        }
    }
    enq(node);
    return node;
}

/**
* 尝试排队
**/
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)) {
                setHead(node);
                p.next = null; // help GC
                failed = false;
                return interrupted;
            }
            //获取锁失败后暂停线程
            if (shouldParkAfterFailedAcquire(p, node) && parkAndCheckInterrupt()) 
              interrupted = true;
        }
    } finally {
        if (failed) 
          cancelAcquire(node);
    }
}

解锁lock()

解锁方法定义在AbstractQueuedSynchronizer中,实现在ReentrantLock的内部类Sync中

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

//AbstractQueuedSynchronizer.java
public final boolean release(int arg) {
    //尝试获取锁
    if (tryRelease(arg)) {
        Node h = head;
        if (h != null && h.waitStatus != 0) 
            //唤醒等待的线程获取锁
            unparkSuccessor(h);
        return true;
    }
    return false;
}

//ReentrantLock.Sync
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;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,245评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,749评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,960评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,575评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,668评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,670评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,664评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,422评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,864评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,178评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,340评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,015评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,646评论 3 323
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,265评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,494评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,261评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,206评论 2 352

推荐阅读更多精彩内容