基于CAS手写一把锁

CAS: Compare and Swap，比较并交换。

每个线程都有自己独立的栈空间，而堆内存是多个线程可以共享的，线程运行时，会先copy一份主内存的变量，存在自己的栈内存中，想修改主内存共享变量的值，会先比较栈内存和主内存的值是否一致，一致则更新主内存中的变量，不一致则重新copy主内存的变量到自己的栈内存中。

CAS概念性的理解，可参考下图案例

CAS底层不是java实现的，是通过Jni技术调用C语言的接口实现的，最底层是CPU硬件指令保证原子性问题。AtomicInteger就是基于CAS实现的，它是线程安全的。

现在我们基于AtomicInteger手写一把锁

import java.util.concurrent.atomic.AtomicInteger;

public class AtomicTryLock {
    private static AtomicInteger atomicInteger = new AtomicInteger(0);

    //获取锁
    public static boolean tryLock(){
        //atomicInteger值修改为1 表示获取到锁
        return atomicInteger.compareAndSet(0, 1);
    }

    //释放锁
    public static boolean unLock(){
        //atomicInteger值修改为0 表示释放锁
        return atomicInteger.compareAndSet(1, 0);
    }

}

实现Runnable接口

public class MyRunnable implements Runnable {
    public int a = 0;

    @Override
    public void run() {
        //同一时刻 只有一个线程 会获取到锁
        try {
            if (AtomicTryLock.tryLock()) {
                System.out.println(Thread.currentThread().getName()+"获取到锁成功");
                //模拟业务逻辑
                Thread.sleep(1000);
            }else{
                System.out.println(Thread.currentThread().getName()+"获取到锁失败");
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            AtomicTryLock.unLock();
        }
    }
}

编写测试类验证

public class Test3 {

    public static void main(String[] args) {

        MyRunnable myRunnable = new MyRunnable();
        IntStream.range(0, 10).forEach(i -> {
            new Thread(myRunnable).start();
        });

    }
}

我们同时起了10个线程，只有一个线程获取了锁，因为同一时刻十个线程执行tryLock方法时，只有一个线程能将atomicInteger的值由0置为1