实现TOLock过程中的一处多线程bug
背景
最近在啃《多处理器编程的艺术》，书中的7.6节介绍了时限锁——实现了tryLock方法的队列锁。
书中重点讲解了tryLock的实现，也就是如何实现在等待超时后退出队列，放弃锁请求，并且能让后继线程感知到。
在实现的过程中，我为TOLock补充了lock方法的实现。代码如下所示：
public class TOLock implements Lock {
private static final QNode AVAILABLE = new QNode();
private AtomicReference<QNode> tail;
private ThreadLocal<QNode> myNode;
public TOLock() {
this.tail = new AtomicReference<>(null);
this.myNode = new ThreadLocal<>();
}
@Override
public void lock() {
QNode qNode = new QNode();
qNode.pred = null;
myNode.set(qNode);
QNode myPred = tail.getAndSet(qNode);
if (myPred == null) {
return;
}
while (myPred.pred != AVAILABLE) {
if (myPred.pred != null) {
myPred = myPred.pred;
}
}
}
@Override
public void unlock() {
QNode qNode = myNode.get();
if (!tail.compareAndSet(qNode, null)) {
qNode.pred = AVAILABLE;
}
}
@Override
public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
long startTime = System.currentTimeMillis();
long patience = TimeUnit.MILLISECONDS.convert(time, unit);
QNode qNode = new QNode();
myNode.set(qNode);
qNode.pred = null;
QNode myPred = tail.getAndSet(qNode);
if (myPred == null || myPred.pred == AVAILABLE) {
return true;
}
while (System.currentTimeMillis() - startTime < patience) {
QNode predPred = myPred.pred;
if (predPred == AVAILABLE) {
return true;
} else if (predPred != null) {
myPred = predPred;
}
}
if (!tail.compareAndSet(qNode, myPred)) {
qNode.pred = myPred;
}
return false;
}
private static class QNode {
volatile QNode pred;
}
}
问题
在编写lock方法的单元测试的时候发现，TOLock偶现卡死，当加大线程池中线程数量，几乎是稳定复现卡死。遂debug，发现lock方法卡死是因为
while (myPred.pred != AVAILABLE) {
if (myPred.pred != null) {
myPred = myPred.pred;
}
}
这段代码中的myPred.pred为null，所以就一直走不出去。myPred.pred为什么会是null，再定睛一看，myPred居然就是AVAILABLE。这怎么会呢？
myPred的取值路径只有两种，一种是从tail通过GAS操作拿到tail之前的值，另一种就是走循环拿到myPred.pred。
原子引用tail一开始为null，每次都是吸一个new出来的QNode进队列，绝对不可能会有AVAILABLE的情况。
排除了所有不可能的情况，剩下的即使在不可能，也都是真相了。
事实上就是如此,myPred变为AVAILABLE是通过循环的if分支拿到的。
if (myPred.pred != null) {
myPred = myPred.pred;
}
这段while循环在多线程的情况下是有bug的，在并发环境下，在后继争用线程读取while条件的时候，有可能前驱线程还没有释放锁，所以此时的myPred.pred应该为null，接下去前驱线程释放了锁，此时myPred.pred为AVAILABLE，这时当前线程理应具备进入临界区的条件，但是因为内存循环的判断导致myPred被赋值为AVAILABLE，此后myPred.pred永远为null，线程无限时原地旋转，后继线程也无法进入临界区。
我随后将此处代码改为如下：
QNode predPred;
while ((predPred = myPred.pred) != AVAILABLE) {
if (predPred != null) {
myPred = predPred;
}
}
其实就和tryLock中的写法是类似的，将myPred.pred赋值到局部变量，封闭在栈上即可。
后记
后来仔细想了想，其实类似的写法在我以前读源码的时候我有注意到过，例如BufferedInputStream里面也有很多类似的将变量封闭在栈上保证线程安全的做法。可以参考这篇帖子。
在FilterInputStream里面定义的in是
protected volatile InputStream in可以被子类访问到，或者BufferedInputStream本身的byte[] buf也是protected volatile的并且还被volatile修饰，本身就是为多线程环境设计的，所以我们可以看到
private InputStream getInIfOpen() throws IOException {
InputStream input = in;
if (input == null)
throw new IOException("Stream closed");
return input;
}
JDK中getInIfOpen是这样写的，虽然看上去挺丑的，像是冗余了一个input变量，但还真就是改不得。
否则就会出现在读取input的时候还不是null，return出去就已经是null的情况了。
看过类似代码，积累过知识，自己真正操刀练习的时候还是犯错，只说明了一句话：纸上得来终觉浅，绝知此事要躬行。
完整源码实现
关于TOLock的完整代码实现，可以参考我的github上的实现。
