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