之前的文章中简单的为大家介绍了重入锁JAVA并发之多线程基础(2)。这里面也是简单的为大家介绍了重入锁的几种性质,这里我们就去探索下里面是如何实现的。
我们知道在使用的时候,必须锁先有定义,然后我们再拿着当前的锁进行加锁操作,然后处理业务,最后是释放锁的操作(这里就拿里面非公平锁的实现来讲解)。
字节码操作
public class com.montos.lock.ReentrantLockDemo implements java.lang.Runnable {
public static java.util.concurrent.locks.ReentrantLock lock;
public static int k;
public com.montos.lock.ReentrantLockDemo();
Code:
0: aload_0
1: invokespecial #1 // Method java/lang/Object."":()V
4: return
public void run();
Code:
0: iconst_0
1: istore_1
2: iload_1
3: sipush 1000
6: if_icmpge 29 //int类型的值进行栈顶比较
9: getstatic #2 // Field lock:Ljava/util/concurrent/locks/ReentrantLock;
12: invokevirtual #3 // Method java/util/concurrent/locks/ReentrantLock.lock:()V
15: getstatic #4 // Field k:I
18: iconst_1
19: iadd
20: putstatic #4 // Field k:I
23: iinc 1, 1
26: goto 2
29: iconst_0
30: istore_1
31: iload_1
32: sipush 1000
35: if_icmpge 50
38: getstatic #2 // Field lock:Ljava/util/concurrent/locks/ReentrantLock;
41: invokevirtual #5 // Method java/util/concurrent/locks/ReentrantLock.unlock:()V
44: iinc 1, 1
47: goto 31
50: return
public static void main(java.lang.String[]) throws java.lang.InterruptedException;
Code:
0: new #6 // class com/montos/lock/ReentrantLockDemo
3: dup
4: invokespecial #7 // Method "":()V
7: astore_1
8: new #8 // class java/lang/Thread
11: dup
12: aload_1
13: invokespecial #9 // Method java/lang/Thread."":(Ljava/lang/Runnable;)V
16: astore_2
17: new #8 // class java/lang/Thread
20: dup
21: aload_1
22: invokespecial #9 // Method java/lang/Thread."":(Ljava/lang/Runnable;)V
25: astore_3
26: aload_2
27: invokevirtual #10 // Method java/lang/Thread.start:()V
30: aload_3
31: invokevirtual #10 // Method java/lang/Thread.start:()V
34: aload_2
35: invokevirtual #11 // Method java/lang/Thread.join:()V
38: aload_3
39: invokevirtual #11 // Method java/lang/Thread.join:()V
42: getstatic #12 // Field java/lang/System.out:Ljava/io/PrintStream;
45: getstatic #4 // Field k:I
48: invokevirtual #13 // Method java/io/PrintStream.println:(I)V
51: return
static {};
Code:
0: new #14 // class java/util/concurrent/locks/ReentrantLock
3: dup
4: invokespecial #15 // Method java/util/concurrent/locks/ReentrantLock."":()V
7: putstatic #2 // Field lock:Ljava/util/concurrent/locks/ReentrantLock;
10: iconst_0
11: putstatic #4 // Field k:I
14: return
}
复制代码
这里面无非就是入栈,栈元素比较,出栈放入变量中这些操作,没有之前的synchronized里面的监视器相关指令限制,只是简单的一些栈操作。
加锁操作
final void lock(){
if (compareAndSetState(0, 1)) //将同步状态从0变成1 采用cas进行更新
setExclusiveOwnerThread(Thread.currentThread());//设置当前拥有独占访问权的线程。
else
acquire(1);//没有获取到锁,则进行尝试操作
}
复制代码
往下面的选择走:
public final void acquire(int arg){
//先进行再次尝试获取锁的操作,如果获取失败则将当前加入队列中,并设置中断标志。
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
selfInterrupt();
}
复制代码
首先走尝试获取锁的操作(这里还是走非公平锁的):
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;
}
复制代码
接着往下走:
private Node addWaiter(Node mode){
//独占模式进行封装当前线程
Node node = new Node(Thread.currentThread(), mode);
Node pred = tail;
if (pred != null) {如果尾节点不为null,将当前的节点接入并返回
node.prev = pred;
if (compareAndSetTail(pred, node)) {
pred.next = node;
return node;
}
}
enq(node);
return node;
}
复制代码
继续往下走:
private Node enq(final Node node){
for (;;) {//
Node t = tail;
if (t == null) { // 初始化尾节点
if (compareAndSetHead(new Node()))
tail = head;
} else {
node.prev = t;
if (compareAndSetTail(t, node)) {//尾节点与当前的节点互换
t.next = node;
return t;//返回当前节点
}
}
}
}
复制代码
接着回去往下走:
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; // 手动清除引用 帮助GC
failed = false;
return interrupted;
}
//检测获取锁失败的节点状态 以及暂时挂起并返回当前的中断标志
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
interrupted = true;
}
} finally {
if (failed)
cancelAcquire(node);//取消正在进行的获取尝试。
}
}
复制代码
说真的,咱们直接看失败的情况,我们接着往下走:
private static boolean shouldParkAfterFailedAcquire(Node pred, Node node){
//检查和更新无法获取的节点的状态。
int ws = pred.waitStatus;
if (ws == Node.SIGNAL)
//该节点已经设置了请求释放信号状态,所以可以进行安全挂起
return true;
if (ws > 0) {
do {//清除不需要执行的节点
node.prev = pred = pred.prev;
} while (pred.waitStatus > 0);
pred.next = node;
} else {
//waitstatus必须为0或传播。表明我们需要信号,但不要挂起。调用者重试以确保在挂起前无法获取。
compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
}
return false;
}
复制代码
然后看向下一个方法:
private final boolean parkAndCheckInterrupt(){
LockSupport.park(this);//挂起当前线程
return Thread.interrupted();//返回中断标识
}
复制代码
上面的取消获取队列里面的节点就不看了..cancelAcquire(node),里面就是取消正在进行的获取尝试。同时将无需的节点移除。当上面的操作走完之后就设置当前线程中断标识。这里面主要流程是说如果加锁不成功之后,对于当前线程是怎么执行操作的,我们可以看到,里面的方法中大部分在获取不到锁之后,下一步操作中会再次尝试获取下,如果获取不到才会继续执行,获取到了我们就可以直接使用,这里也是多线程操作里面的魅力,每一个空隙中就可能会让当前线程进行获得锁的操作。
释放锁操作
释放锁的步骤就简单许多了:
public final boolean release(int arg){
if (tryRelease(arg)) {//尝试释放锁
Node h = head;
if (h != null && h.waitStatus != 0)
unparkSuccessor(h);//唤醒节点的后续节点
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;//只有当前重入次数为0,才能返回true
setExclusiveOwnerThread(null);//当前独占线程设为NULL
}
setState(c);//重新设置同步状态
return free;
}
复制代码
然后往下走:
private void unparkSuccessor(Node node){
//当前状态为负数,则尝试清除当前的线程状态
int ws = node.waitStatus;
if (ws < 0)
compareAndSetWaitStatus(node, ws, 0);
//清除取消或无效的节点,从尾部向后移动以找到实际节点
Node s = node.next;
if (s == null || s.waitStatus > 0) {
s = null;
for (Node t = tail; t != null && t != node; t = t.prev)
if (t.waitStatus <= 0)
s = t;
}
if (s != null)
LockSupport.unpark(s.thread);//释放当前线程
}
复制代码从上面的顺序往下面来看,我们主要发现线程在拿锁阶段是有许多的操作的,要根据线程的状态再将线程从等待队列中移除。释放的时候就显得简洁了许多,我们只需要看到当前线程的状态-1,然后看看是否是重入的。
我们通过一个简单的重入锁代码可以看到,作者在用无锁的操作去获得锁,这个整体的步骤里面考虑的东西很多,每一个时刻,线程都有可能千变万化,我们需要了解的是我们每一个步骤都需要可能发生的情况。如果能够考虑到发生的情况,那么有些步骤就可以直接跳过,我们直接就可以获得最后的结果(这块在线程尝试获锁的阶段可以体现)。有小伙伴对于重入锁还有什么看法的可以在下面进行留言,我们可以相互学习,共同进步~