手写自旋锁

手写自旋锁

• 需要AtomicReference类的cas操作
• 加锁方法cas占用则循环,没有就拿到
• 解锁直接置为null就行
• 具体代码有注释

package com.xiaojie.haha;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;

/**
 * 实现自旋锁
 */
public class SpinLock {
    AtomicReference<Thread> atomicReference = new AtomicReference<>();//要使用此类的cas

    public void MyLock(){//我的自旋锁
        Thread thread = Thread.currentThread();//拿到当前的线程
        while (!atomicReference.compareAndSet(null,thread)){//如果为null证明没人用,则赋值为thread,有值则循环等待

        }
        System.out.println(Thread.currentThread().getName()+"拿到了MyLock");//拿到了说一声
    }

    public void MyUnLock(){//我的解锁
        Thread thread = Thread.currentThread();//拿到当前的线程
        atomicReference.compareAndSet(thread,null);//改回null
        System.out.println(Thread.currentThread().getName()+"解锁了MyUnLock");//解锁了说一声
    }

    public static void main(String[] args) throws InterruptedException {
        SpinLock spinLock = new SpinLock();
        new Thread(()->{
            spinLock.MyLock();//线程加锁
            try {
                TimeUnit.SECONDS.sleep(5);//休息五秒在解锁
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            spinLock.MyUnLock();//线程解锁
        },"t1").start();

        TimeUnit.SECONDS.sleep(1);//保证上面先执行

        new Thread(()->{
            spinLock.MyLock();//线程加锁
            try {
                TimeUnit.SECONDS.sleep(1);//休息1秒在解锁
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            spinLock.MyUnLock();//线程解锁
        },"t2").start();
    }
}