自旋锁—简单JAVA示例

1、自旋锁定义： 实际就是while/do...while+CAS

• 尝试获取锁的线程不会立即阻塞，而是采用循环的方法尝试获取锁
• 这样的好处是减少线程上下文切换的消耗，缺点是循环会消耗CPU

2、示例代码

SpinLock.java

package com.mzj.javabase.thread;

import java.util.concurrent.atomic.AtomicReference;

public class SpinLock {

    AtomicReference<Thread> atomicReference = new AtomicReference<>();

    public void myLock() {
        //获取当前线程对象
        Thread thread = Thread.currentThread();
        System.out.println(Thread.currentThread().getName() + "\t come in ####");
        //当owner持有线程不为空时，循环等待
        while (!atomicReference.compareAndSet(null, thread)) {
            //当owner持有线程为空时，将owner持有线程设为当前线程，退出循环
        }
    }

    public void myUnLock() {
        Thread thread = Thread.currentThread();
        atomicReference.compareAndSet(thread, null);
        //执行完成后，将owner持有线程重新置为空，相当于释放锁
        System.out.println(Thread.currentThread().getName() + "\t invoke myUnLock()");
    }
}

AtomicReference类提供了对引用对象的原子操作。AtomicReference类持有一个对象，核心是CAS(Compare and Swap, 比较并交换）算法。对应于上面代码的compareAndSet方法。

CAS算法：CAS算法有3个操作数，当前值value，预期值expect，修改的新值update。当value与预期值expect相同时，将更新当前值value为新值update。

SpinLockDemo.java

package com.mzj.javabase.thread;

import java.util.concurrent.TimeUnit;

/**
 * 描述：
 * <p>
 * 线程AA首先进入MyLock()方法，输出 AA     come in ####
 * compareAndSet比较内存值是否为空，true，将内存值改为AA线程，A睡眠5秒线程
 * 线程BB进入MyLock()方法，输出 BB     come in ####
 * 此时compareAndSet，内存值为AA线程，while语句返回false，BB线程一直在此处判断，直到AA线程进入myUnLock()方法，将内存值重新改为null，输出 AA     invoke myUnLock()，AA线程运行结束
 * 线程BB进入myUnLock()方法，输出 BB     invoke myUnLock()，BB线程运行结束
 */
public class SpinLockDemo {

    public static void main(String[] args) {

        SpinLock spinLock = new SpinLock();
        //AA线程
        new Thread(() -> {
            spinLock.myLock();
            try {
                TimeUnit.SECONDS.sleep(5);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            spinLock.myUnLock();
        }, "AA").start();

        //BB线程 自旋等待AA使用完成
        try {
            TimeUnit.SECONDS.sleep(1);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        new Thread(() -> {
            spinLock.myLock();
            spinLock.myUnLock();
        }, "BB").start();
    }


}

3、总结

可以看到CAS操作巧妙的实现了自旋锁。

4、完整工程代码

https://github.com/mazhongjia/javabase/tree/master/src/main/java/com/mzj/javabase/thread/spinlock