AQS的原理

2020-11-04  本文已影响0人  baifanger

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

1.使用方法

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++,接着往下处理即可。否则,会出现死锁的现象。

上一篇下一篇

猜你喜欢

热点阅读