【Java】CLH 自旋锁

看java重入锁的代码,遇到了CLH队列锁,发现实现很巧妙,学习一下。

什么是自旋锁?说的是锁等待的实现方式,可以改变线程的状态,让其进入等待或者睡眠(具体状态还没研究),这就是使用线程最原生的方式实现。也可以让线程进入一个while循环,这是一种轻量级实现,没有设计线程状态的转换,节省了转换的开销,但是cpu开销可能很大,进入循环的方式就是自旋锁。

下面是一段很常见的理解CLH的代码:

class ClhSpinLock {
    private final ThreadLocal<Node> prev;
    private final ThreadLocal<Node> node;
    private final AtomicReference<Node> tail = new AtomicReference<Node>(new Node());

    public ClhSpinLock() {
        this.node = new ThreadLocal<Node>() {
            protected Node initialValue() {
                return new Node();
            }
        };

        this.prev = new ThreadLocal<Node>() {
            protected Node initialValue() {
                return null;
            }
        };
    }

    public void lock() {
        final Node node = this.node.get();
        node.locked = true;
        // 一个CAS操作即可将当前线程对应的节点加入到队列中,
        // 并且同时获得了前继节点的引用,然后就是等待前继释放锁
        Node pred = this.tail.getAndSet(node);
        this.prev.set(pred);
        while (pred.locked) {// 进入自旋
        }
    }

    public void unlock() {
        final Node node = this.node.get();
        node.locked = false;
        this.node.set(this.prev.get());
    }

    private static class Node {
        private volatile boolean locked;
    }
}

摘自:
http://www.cnblogs.com/zhanjindong/p/java-concurrent-package-aqs-clh-and-spin-lock.html

一个锁类有两种共享变量,第一种是线程私有的,即ThreadLocal类型,线程之间不会影响。另一种就是共享的,线程之间可见。

共享的是一个tail指针,指向整个等待队列的最后一个元素。这个也必须是共享的,否则各个线程就没有关系了。通过这个共享变量,把整个队列联系起来。另外两个变量是线程私有的,每一个线程在内部都有副本。myNode指的是当前线程对应的node,pred指的是当前线程的前驱节点。

在lock方法内,首先得到当前线程的myNode节点,这个是通过ThreadLocal的intialize方法实现的,默认返回一个新的节点代表当前线程。设置node的属性locked为true表明在使用锁,其后继节点就会在这个变量上循环。这时当前线程还没有和整个队列建立关系,所以接下来要通过全局的tail变量把自己接入队列。调用tail的getAndSet,得到当前的tail命名为pred,然后把自己设置为tail,那么pred就是当前节点的前驱节点。这个过程是通过CAS控制并发的。这时当前线程已经得到了前驱,而且把自己设置为tail,那么就先用自己的prev变量缓存前驱,在前驱上while循环。只有前驱把自己的locked设置为false,才会停止while。

这样每一个线程都通过这样的方法,用tail形成一个队列,然后在前驱自旋。整个构成是通过ThreadLocal来创建新的节点。

    原文作者:java锁
    原文地址: https://blog.csdn.net/u010900754/article/details/77476936
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞