美文网首页
AQS的原理

AQS的原理

作者: baifanger | 来源:发表于2020-11-04 18:37 被阅读0次

        AQS: AbstractQueuedSynchronizer的简称,在jdk中,我们可以发现,这是一个抽象类,需要我们实现tryAcquire和tryRelease方法。是我们可以自己实现锁的一种方式。

    1.使用方法

    • 自定义Sync类,继承自AbstractQueuedSynchronizer,实现其tryAcquire()和tryRelease()方法。
    • 自定义MyLock类,继承自Lock,实现其相应的方法,重要的是lock与unLock方法,在这两个方法中,分别调用Sync.acquire()与Sync.release方法。在AQS类内部,acquire会调用tryAcquire,release会调用tryRelease方法。
    • 在需要锁的地方,使用MyLock.lock及MyLock.unLock来实现锁的添加和消除。而Sync类,则做为MyLock的内部类来调用,不被外界使用

    2.原理
        以下我们会以ReentrantLock为例来讲解,先看AbstractQueueSynchronizer类:

    AbstractQueuedSynchronizer.java
        private transient volatile AbstractQueuedSynchronizer.Node head;
        private transient volatile AbstractQueuedSynchronizer.Node tail;
        private volatile int state;
        protected final int getState() {
            return this.state;
        }
    
        protected final void setState(int var1) {
            this.state = var1;
        }
    
        protected final boolean compareAndSetState(int var1, int var2) {
            return unsafe.compareAndSwapInt(this, stateOffset, var1, var2);
        }
    

        通过上述代码,需要注意关键的点:变量state及方法compareAndSetState,通过这两个定义,我们可以推出,AQS内部锁机制,实际还是通过CAS机制来进行的。

    我们查看ReentrantLock.FairSync的代码

    FairSync extends ReentrantLock.Sync
            protected final boolean tryAcquire(int var1) {
                Thread var2 = Thread.currentThread();
                int var3 = this.getState();
                if (var3 == 0) {
                    if (!this.hasQueuedPredecessors() && this.compareAndSetState(0, var1)) {
                        this.setExclusiveOwnerThread(var2);
                        return true;
                    }
                } else if (var2 == this.getExclusiveOwnerThread()) {
                    int var4 = var3 + var1;
                    if (var4 < 0) {
                        throw new Error("Maximum lock count exceeded");
                    }
    
                    this.setState(var4);
                    return true;
                }
    
                return false;
            }
    

    通过代码可以看出,在ReentrantLock中,其lock的实现,就是用到了AbstractQueuedSynchronizer提供的compareAndSetState,即CAS的原理。所以在理解AQS的时候,一定要知道其内部state的作用。

    3.AQS的队列(CLH队列)
    对于已获得锁的线程可以去执行了,但对于未获得锁的线程如何处理呢,这就要用到AQS中的Q了,从上面的代码中,我们可以看到 ** AbstractQueuedSynchronizer.Node**,这就是AQS中保存的队列,申请所失败的线程,会加到队列当中,并且从代码中可以看出,队列中的结点会保存自己的前驱及后驱节点。

    AbstractQueueSynchronizer.java
        public final void acquire(int var1) {
            if (!this.tryAcquire(var1) && this.acquireQueued(this.addWaiter(AbstractQueuedSynchronizer.Node.EXCLUSIVE), var1)) {
                selfInterrupt();
            }
        }
        final boolean acquireQueued(AbstractQueuedSynchronizer.Node var1, int var2) {
            boolean var3 = true;
    
            try {
                boolean var4 = false;
    
                while(true) {
                    AbstractQueuedSynchronizer.Node var5 = var1.predecessor();
                    if (var5 == this.head && this.tryAcquire(var2)) {
                        this.setHead(var1);
                        var5.next = null;
                        var3 = false;
                        boolean var6 = var4;
                        return var6;
                    }
    
                    if (shouldParkAfterFailedAcquire(var5, var1) && this.parkAndCheckInterrupt()) {
                        var4 = true;
                    }
                }
            } finally {
                if (var3) {
                    this.cancelAcquire(var1);
                }
    
            }
        }
    

    在tryAcquire()拿锁时,如果失败了,会调用acquireQueued方法,将线程添加到队列中。
    (在有的资料中,说Node添加到队列后,会进行自旋,检查自己的上一节点是否已释放锁,但检查几次后,就会停止检查,这个我还有点没明白,后续补充一下)

    4.公平锁与非公平锁
        在使用ReentrantLock时,其构造方法中可以传入一个boolean类型数据,来表明是否为公平锁,那什么是公平锁与非公平锁
    公平锁:多线程按照申请锁的顺序来获取锁,不会出现争抢的情况。
    非公平锁:多线程不会按照顺序获取锁,会先尝试去争抢锁,有可能后申请的线程比先申请的线程先抢到锁,在高并发情况下,可能会造成优先级反转和饥饿的现象。
    ReentrantLock默认为非公平锁,也可指定boolean值,改为公平锁,具体公平与非公平的实现,还要比较其NonfairSync与FairSync的实现

    static final class FairSync extends ReentrantLock.Sync {
            protected final boolean tryAcquire(int var1) {
                Thread var2 = Thread.currentThread();
                int var3 = this.getState();
                if (var3 == 0) {
                    if (!this.hasQueuedPredecessors() && this.compareAndSetState(0, var1)) {
                        this.setExclusiveOwnerThread(var2);
                        return true;
                    }
                } else if (var2 == this.getExclusiveOwnerThread()) {
                    int var4 = var3 + var1;
                    if (var4 < 0) {
                        throw new Error("Maximum lock count exceeded");
                    }
    
                    this.setState(var4);
                    return true;
                }
                return false;
            }
        }
    
    
    static final class NonfairSync extends ReentrantLock.Sync {
            final void lock() {
                if (this.compareAndSetState(0, 1)) {
                    this.setExclusiveOwnerThread(Thread.currentThread());
                } else {
                    this.acquire(1);
                }
            }
    
            protected final boolean tryAcquire(int var1) {
                return this.nonfairTryAcquire(var1);
            }
        }
    

    比较其tryAquire方法后,两者最大的区别在于!this.hasQueuedPredecessors(),在公平锁中,调用了此方法,而在非公平锁中,则没有。在此方法中,会判断线程的前驱结点线程是否在队列前面,如果是,就去尝试获取锁,达到队列前面的线程能优先获取锁,从而实现公平锁。而在非公平锁中,则不会检查此项,每个结点都会尝试拿锁,但最后不一定是最先入队的拿到锁。

    5.可重入锁
        广义上的可重入锁指的是可重复可递归调用的锁,在外层使用锁之后,在内层仍然可以使用,并且不发生死锁(前提得是同一个对象或者class),这样的锁就叫做可重入锁。我们所知的ReentransLock和Synchronized.看源码,了解ReentransLock如何实现可重入的:

    FairSync.java
     protected final boolean tryAcquire(int var1) {
                Thread var2 = Thread.currentThread();
                int var3 = this.getState();
                if (var3 == 0) {
                    if (!this.hasQueuedPredecessors() && this.compareAndSetState(0, var1)) {
                        this.setExclusiveOwnerThread(var2);
                        return true;
                    }
                //可重入机制的关键
                } else if (var2 == this.getExclusiveOwnerThread()) {
                    int var4 = var3 + var1;
                    if (var4 < 0) {
                        throw new Error("Maximum lock count exceeded");
                    }
    
                    this.setState(var4);
                    return true;
                }
    
                return false;
            }
    

    关键代码:var2 == this.getExclusiveOwnerThread(),通过上面,我们知道,在AQS中,通过state字段是否为0来进行CAS操作的,如果当前线程存在递归,第一次获取锁之后,state的值必然会变为了1,但递归仍在此线程中,就会再进行到此处加锁的代码,此时如果再进行CAS操作,就必然是拿锁失败,则会走到上面标粗的关键代码上,此时我们只要判断当前线程是之前拿到锁的线程,则让state++,接着往下处理即可。否则,会出现死锁的现象。

    相关文章

      网友评论

          本文标题:AQS的原理

          本文链接:https://www.haomeiwen.com/subject/vxegvktx.html