Java 并发编程学习笔记 理解CLH队列锁算法

CLH算法实现

CLH队列中的结点QNode中含有一个locked字段，该字段若为true表示该线程需要获取锁，且不释放锁，为false表示线程释放了锁。结点之间是通过隐形的链表相连，之所以叫隐形的链表是因为这些结点之间没有明显的next指针，而是通过myPred所指向的结点的变化情况来影响myNode的行为。CLHLock上还有一个尾指针，始终指向队列的最后一个结点。CLHLock的类图如下所示：

 

 

 

当一个线程需要获取锁时，会创建一个新的QNode，将其中的locked设置为true表示需要获取锁，然后线程对tail域调用getAndSet方法，使自己成为队列的尾部，同时获取一个指向其前趋的引用myPred,然后该线程就在前趋结点的locked字段上旋转，直到前趋结点释放锁。当一个线程需要释放锁时，将当前结点的locked域设置为false，同时回收前趋结点。如下图所示，线程A需要获取锁，其myNode域为true，些时tail指向线程A的结点，然后线程B也加入到线程A后面，tail指向线程B的结点。然后线程A和B都在它的myPred域上旋转，一量它的myPred结点的locked字段变为false，它就可以获取锁扫行。明显线程A的myPred locked域为false，此时线程A获取到了锁。

 

 

 

 

 

整个CLH的代码如下，其中用到了ThreadLocal类，将QNode绑定到每一个线程上，同时用到了AtomicReference,对尾指针的修改正是调用它的getAndSet()操作来实现的，它能够保证以原子方式更新对象引用。

 

 public class CLHLock {  

     AtomicReference<QNode> tail = new AtomicReference<QNode>(new QNode());
     ThreadLocal<QNode> myPred;
     ThreadLocal<QNode> myNode;  

     public static class QNode {
         //注意这个地方 如果不加volatile则会导致线程永远死循环
         //关于volatile的用法在我的另外一篇文章 http://www.cnblogs.com/daxin/p/3364014.html
         public volatile boolean locked = false;
     }

     public CLHLock() {
         myNode = new ThreadLocal<QNode>() {
             protected QNode initialValue() {
                 return new QNode();
             }
         };
         myPred = new ThreadLocal<QNode>() {
             protected QNode initialValue() {
                 return null;
             }
         };
     }  

     public void lock() {
         QNode qnode = myNode.get();
         qnode.locked = true;
         QNode pred = tail.getAndSet(qnode);
         myPred.set(pred);
         while (pred.locked) {
             //非阻塞算法
         }
     }  

     public void unlock() {
         QNode qnode = myNode.get();
         qnode.locked = false;
         myNode.set(myPred.get());
     }
 } 

 

从代码中可以看出lock方法中有一个while循环，这 是在等待前趋结点的locked域变为false，这是一个自旋等待的过程。unlock方法很简单，只需要将自己的locked域设置为false即可。

CLH优缺点

CLH队列锁的优点是空间复杂度低（如果有n个线程，L个锁，每个线程每次只获取一个锁，那么需要的存储空间是O（L+n），n个线程有n个myNode，L个锁有L个tail），CLH的一种变体被应用在了JAVA并发框架中。唯一的缺点是在NUMA系统结构下性能很差，在这种系统结构下，每个线程有自己的内存，如果前趋结点的内存位置比较远，自旋判断前趋结点的locked域，性能将大打折扣，但是在SMP系统结构下该法还是非常有效的。一种解决NUMA系统结构的思路是MCS队列锁。