Java并发5:ReentrantReadWriteLock原理

1、图解流程

读写锁用的是同一个 Sycn 同步器,因此等待队列、state 等也是同一个
t1 w.lock,t2 r.lock
1) t1 成功上锁,流程与 ReentrantLock 加锁相比没有特殊之处,不同是写锁状态占了 state 的低 16 位,而读锁
使用的是 state 的高 16 位
在这里插入图片描述
2)t2 执行 r.lock,这时进入读锁的 sync.acquireShared(1) 流程,首先会进入 tryAcquireShared 流程。如果有写锁占据,那么 tryAcquireShared 返回 -1 表示失败。

tryAcquireShared 返回值表示

  • -1 表示失败
  • 0 表示成功,但后继节点不会继续唤醒
  • 正数表示成功,而且数值是还有几个后继节点需要唤醒,读写锁返回 1

在这里插入图片描述
3)这时会进入 sync.doAcquireShared(1) 流程,首先也是调用 addWaiter 添加节点,不同之处在于节点被设置为Node.SHARED 模式而非 Node.EXCLUSIVE 模式,注意此时 t2 仍处于活跃状态
在这里插入图片描述
4)t2 会看看自己的节点是不是老二,如果是,还会再次调用 tryAcquireShared(1) 来尝试获取锁
5)如果没有成功,在 doAcquireShared 内 for 循环一次,把前驱节点的 waitStatus 改为 -1,再 for 循环一次尝试 tryAcquireShared(1) 如果还不成功,那么在 parkAndCheckInterrupt() 处 park
在这里插入图片描述
t3 r.lock,t4 w.lock
这种状态下,假设又有 t3 加读锁和 t4 加写锁,这期间 t1 仍然持有锁,就变成了下面的样子

在这里插入图片描述
接下来执行唤醒流程 sync.unparkSuccessor,即让老二恢复运行,这时 t2 在 doAcquireShared 内
parkAndCheckInterrupt() 处恢复运行
这回再来一次 for 执行 tryAcquireShared 成功则让读锁计数加一
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
下一个节点不是 shared 了,因此不会继续唤醒 t4 所在节点
t2 r.unlock,t3 r.unlock
t2 进入 sync.releaseShared(1) 中,调用 tryReleaseShared(1) 让计数减一,但由于计数还不为零
在这里插入图片描述
在这里插入图片描述

1、写锁上锁流程

// WriteLock
static final class NonfairSync extends Sync {
	// ... 省略无关代码
	// 外部类 WriteLock 方法, 方便阅读, 放在此处
	public void lock() {
		sync.acquire(1);
	}
	// AQS 继承过来的方法, 方便阅读, 放在此处
	public final void acquire(int arg) {
		if (
		// 尝试获得写锁失败
		!tryAcquire(arg) &&
		// 将当前线程关联到一个 Node 对象上, 模式为独占模式
		// 进入 AQS 队列阻塞
		acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
		) {
			selfInterrupt();
		}
	}

// Sync 继承过来的方法, 方便阅读, 放在此处
protected final boolean tryAcquire(int acquires) {
	// 获得低 16 位, 代表写锁的 state 计数
	Thread current = Thread.currentThread();
	int c = getState();
	
	int w = exclusiveCount(c);
	if (c != 0) {
		if (
			// c != 0 and w == 0 表示有读锁, 或者
			w == 0 ||
			// 如果 exclusiveOwnerThread 不是自己
			current != getExclusiveOwnerThread()
		) {
		// 获得锁失败
		return false;
		}
		// 写锁计数超过低 16 位, 报异常
		if (w + exclusiveCount(acquires) > MAX_COUNT)
			throw new Error("Maximum lock count exceeded");
		// 写锁重入, 获得锁成功
		setState(c + acquires);
		return true;
	}
	if (
		// 判断写锁是否该阻塞, 或者
		writerShouldBlock() ||
		// 尝试更改计数失败
		!compareAndSetState(c, c + acquires)
		) {
		// 获得锁失败
		return false;
	}
		// 获得锁成功
		setExclusiveOwnerThread(current);
		return true;
	}
	// 非公平锁 writerShouldBlock 总是返回 false, 无需阻塞
	final boolean writerShouldBlock() {
		return false;
	}
}

2、写锁释放流程

static final class NonfairSync extends Sync {
	// ... 省略无关代码
	// WriteLock 方法, 方便阅读, 放在此处
	public void unlock() {
		sync.release(1);
	}
	// AQS 继承过来的方法, 方便阅读, 放在此处
	public final boolean release(int arg) {
		// 尝试释放写锁成功
		if (tryRelease(arg)) {
			// unpark AQS 中等待的线程
			Node h = head;
			if (h != null && h.waitStatus != 0)
				unparkSuccessor(h);
			return true;
		}
		return false;
	}
		// Sync 继承过来的方法, 方便阅读, 放在此处
	protected final boolean tryRelease(int releases) {
		if (!isHeldExclusively())
			throw new IllegalMonitorStateException();
		int nextc = getState() - releases;
		// 因为可重入的原因, 写锁计数为 0, 才算释放成功
		boolean free = exclusiveCount(nextc) == 0;
		if (free) {
			setExclusiveOwnerThread(null);
		}
		setState(nextc);
		return free;
	}
}

3、读锁上锁流程

static final class NonfairSync extends Sync {
	// ReadLock 方法, 方便阅读, 放在此处
	public void lock() {
		sync.acquireShared(1);
	}
	// AQS 继承过来的方法, 方便阅读, 放在此处
	public final void acquireShared(int arg) {
		// tryAcquireShared 返回负数, 表示获取读锁失败
		if (tryAcquireShared(arg) < 0) {
			doAcquireShared(arg);
		}
	}
	// Sync 继承过来的方法, 方便阅读, 放在此处
	protected final int tryAcquireShared(int unused) {
		Thread current = Thread.currentThread();
		int c = getState();
		// 如果是其它线程持有写锁, 获取读锁失败
		if (
			exclusiveCount(c) != 0 &&
			getExclusiveOwnerThread() != current
		) {
		return -1;
		}
		int r = sharedCount(c);
		if (
			// 读锁不该阻塞(如果老二是写锁,读锁该阻塞), 并且
			!readerShouldBlock() &&
			// 小于读锁计数, 并且
			r < MAX_COUNT &&
			// 尝试增加计数成功
			compareAndSetState(c, c + SHARED_UNIT)
		) {
		// ... 省略不重要的代码
			return 1;
		}
		return fullTryAcquireShared(current);
	}
	// 非公平锁 readerShouldBlock 看 AQS 队列中第一个节点是否是写锁
	// true 则该阻塞, false 则不阻塞
	final boolean readerShouldBlock() {
		return apparentlyFirstQueuedIsExclusive();
	}
	// AQS 继承过来的方法, 方便阅读, 放在此处
	// 与 tryAcquireShared 功能类似, 但会不断尝试 for (;;) 获取读锁, 执行过程中无阻塞
	final int fullTryAcquireShared(Thread current) {
		HoldCounter rh = null;
		for (;;) {
			int c = getState();
			if (exclusiveCount(c) != 0) {
				if (getExclusiveOwnerThread() != current)
					return -1;
			} else if (readerShouldBlock()) {
				// ... 省略不重要的代码
			}
			if (sharedCount(c) == MAX_COUNT)
				throw new Error("Maximum lock count exceeded");
			if (compareAndSetState(c, c + SHARED_UNIT)) {
				// ... 省略不重要的代码
				return 1;
			}
		}
	}
	// AQS 继承过来的方法, 方便阅读, 放在此处
	private void doAcquireShared(int arg) {
		// 将当前线程关联到一个 Node 对象上, 模式为共享模式
		final Node node = addWaiter(Node.SHARED);
		boolean failed = true;
		try {
			boolean interrupted = false;
			for (;;) {
				final Node p = node.predecessor();
				if (p == head) {
					// 再一次尝试获取读锁
					int r = tryAcquireShared(arg);
					// 成功
					if (r >= 0) {
						// ㈠
						// r 表示可用资源数, 在这里总是 1 允许传播
						//(唤醒 AQS 中下一个 Share 节点)
						setHeadAndPropagate(node, r);
						p.next = null; // help GC
						if (interrupted)
							selfInterrupt();
						failed = false;
						return;
					}
				}
				if (
					// 是否在获取读锁失败时阻塞(前一个阶段 waitStatus == Node.SIGNAL)
					shouldParkAfterFailedAcquire(p, node) &&
					// park 当前线程
					parkAndCheckInterrupt()
				) {
					interrupted = true;
				}
			}
		} finally {
			if (failed)
				cancelAcquire(node);
		}
}
	// ㈠ AQS 继承过来的方法, 方便阅读, 放在此处
	private void setHeadAndPropagate(Node node, int propagate) {
		Node h = head; // Record old head for check below
		// 设置自己为 head
		setHead(node);
		// propagate 表示有共享资源(例如共享读锁或信号量)
		// 原 head waitStatus == Node.SIGNAL 或 Node.PROPAGATE
		// 现在 head waitStatus == Node.SIGNAL 或 Node.PROPAGATE
		if (propagate > 0 || h == null || h.waitStatus < 0 ||
			(h = head) == null || h.waitStatus < 0) {
			Node s = node.next;
			// 如果是最后一个节点或者是等待共享读锁的节点
			if (s == null || s.isShared()) {
				// 进入 ㈡
				doReleaseShared();
			}
		}
	}
	// ㈡ AQS 继承过来的方法, 方便阅读, 放在此处
	private void doReleaseShared() {
		// 如果 head.waitStatus == Node.SIGNAL ==> 0 成功, 下一个节点 unpark
		
		// 如果 head.waitStatus == 0 ==> Node.PROPAGATE, 为了解决 bug, 见后面分析
		for (;;) {
			Node h = head;
			// 队列还有节点
			if (h != null && h != tail) {
				int ws = h.waitStatus;
				if (ws == Node.SIGNAL) {
					if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))
						continue; // loop to recheck cases
					// 下一个节点 unpark 如果成功获取读锁
					// 并且下下个节点还是 shared, 继续 doReleaseShared
					unparkSuccessor(h);
				}
				else if (ws == 0 &&
					!compareAndSetWaitStatus(h, 0, Node.PROPAGATE))
					continue; // loop on failed CAS
			}
			if (h == head) // loop if head changed
				break;
			}
		}
	}

4、读锁释放流程

static final class NonfairSync extends Sync {
	// ReadLock 方法, 方便阅读, 放在此处
	public void unlock() {
		sync.releaseShared(1);
	}
	// AQS 继承过来的方法, 方便阅读, 放在此处
	public final boolean releaseShared(int arg) {
		if (tryReleaseShared(arg)) {
			doReleaseShared();
			return true;
		}
		return false;
	}
	// Sync 继承过来的方法, 方便阅读, 放在此处
	protected final boolean tryReleaseShared(int unused) {
		// ... 省略不重要的代码
		for (;;) {
			int c = getState();
			int nextc = c - SHARED_UNIT;
			if (compareAndSetState(c, nextc)) {
			// 读锁的计数不会影响其它获取读锁线程, 但会影响其它获取写锁线程
			// 计数为 0 才是真正释放
				return nextc == 0;
			}
		}
	}
	// AQS 继承过来的方法, 方便阅读, 放在此处
	private void doReleaseShared() {
		// 如果 head.waitStatus == Node.SIGNAL ==> 0 成功, 下一个节点 unpark
		// 如果 head.waitStatus == 0 ==> Node.PROPAGATE
		for (;;) {
			Node h = head;
			if (h != null && h != tail) {
				int ws = h.waitStatus;
				// 如果有其它线程也在释放读锁,那么需要将 waitStatus 先改为 0
				// 防止 unparkSuccessor 被多次执行
				if (ws == Node.SIGNAL) {
					if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))
						continue; // loop to recheck cases
					unparkSuccessor(h);
				}
				// 如果已经是 0 了,改为 -3,用来解决传播性,见后文信号量 bug 分析
				else if (ws == 0 &&
					!compareAndSetWaitStatus(h, 0, Node.PROPAGATE))
					continue; // loop on failed CAS
			}
			if (h == head) // loop if head changed
				break;
			}
		}
	}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值