乐观锁思想在 JAVA 中的实现—

乐观锁思想在 JAVA 中的实现——CAS

news2024/9/23 7:26:47

前言

生活中我们看待一个事物总有不同的态度，比如半瓶水，悲观的人会觉得只有半瓶水了，而乐观的人则会认为还有半瓶水呢。很多技术思想往往源于生活，因此在多个线程并发访问数据的时候，有了悲观锁和乐观锁。

悲观锁认为这个数据肯定会被其他线程给修改了，那我就给它上锁，只能自己访问，要等我访问完，其他人才能访问，我上锁、解锁都得花费我时间。
乐观锁认为这个数据不会被修改，我就直接访问，当我发现数据真的修改了，那我也“礼貌的”让自己访问失败。

悲观锁和乐观锁其实本质都是一种思想，在 JAVA 中对于悲观锁的实现大家可能都很了解，可以通过synchronized、ReentrantLock加锁实现，本文不展开讲解了。那么乐观锁在 JAVA 中是如何实现的呢？底层的实现机制又是什么呢？

问题引入

我们用一个账户取钱的例子来说明乐观锁和悲观锁的问题。

public class AccountUnsafe {     // 余额     private Integer balance;         public AccountUnsafe(Integer balance) {       this.balance = balance;     }        @Override     public Integer getBalance() {       return balance;     }         @Override     public void withdraw(Integer amount) {       balance -= amount;     }}

复制代码

账户类，withdraw()方法是取钱方法。

public static void main(String[] args) {        // 账户10000元        AccountUnsafe account = new AccountUnsafe(10000);        List<Thread> ts = new ArrayList<>();        long start = System.nanoTime();        // 1000个线程，每次取10元        for (int i = 0; i < 1000; i++) {            ts.add(new Thread(() -> {                account.withdraw(10);            }));        }        ts.forEach(Thread::start);        ts.forEach(t -> {            try {                t.join();            } catch (InterruptedException e) {                e.printStackTrace();            }        });        long end = System.nanoTime();        // 打印账户余额和花费时间        log.info("账户余额：{}, 花费时间: {}", account.getBalance(), (end-start)/1000_000 + " ms");    }

复制代码

账户默认有 10000 元，1000 个线程取钱，每次取 10 元，最后账户应该还有多少钱呢？

运行结果：

运行结果显示余额还有 150 元，显然出现并发问题。

原因分析:

原因也很简单，取钱方法withdraw()的操作balance -= amount;看着就一行代码，实际上会生成多条指令，如下图所示：

多个线程运行的时候会进行线程切换，导致这个操作不是原子性，所以不是线程安全的。

悲观锁解决

最简单的方法，我想大家都能想到吧，给withdraw()方法加锁，保证同一时刻只有一个线程能够执行这个方法，保证了原子性。

通过synchronized关键字加锁。

运行结果：

运行结果正常，但是花费时间稍微多了一点

乐观锁解决

关键来了，如果用乐观锁的思想在 JAVA 中该如何实现呢？

大致思路就是我默认不加任何锁，我先把余额减掉 10 元，最后更新余额的时候，发现余额和我一开始不一样了，我就丢弃当前更新操作，重新读取余额的值，直到更新成功。

找啊找，最终发现 JDK 中的Unsafe方法提供了这样的方法compareAndSwapInt。

先获取老的余额oldBalance，计算出新的余额newBalance
调用 unsafe.compareAndSwapInt()方法，如果内存中余额属性的偏移量BALANCE_OFFSET对应的值等于老的余额，说明的确没有被其他线程访问修改过，我就大胆的更新为newBalance，退出方法
否则的话，我就要进入下一次循环，重新获取余额计算。

那么是如何获取unsafe呢?

静态方法中通过反射的方法获取，因为Unsafe类太底层了，它一般不建议程序员直接使用。

这个 Unsafe 类的名称并不是说线程不安全的意思，只是这个类太底层了，不要乱用，对程序员来说不大安全。

最后别忘了余额balance要加 volatile 修饰。

主要为了保证可见性，让线程能够获取到其他线程修改的结果。

运行结果：

余额也为 0，正常，而且运行速度稍微快了一丢丢

完成代码：

@Slf4j(topic = "a.AccountCAS")public class AccountCAS {    // 余额    private volatile int balance;    // Unsafe对象    static final Unsafe unsafe;    // balance 字段的偏移量    static final long BALANCE_OFFSET;    static {        try {            Field theUnsafe = Unsafe.class.getDeclaredField("theUnsafe");            theUnsafe.setAccessible(true);            unsafe = (Unsafe) theUnsafe.get(null);            // balance 属性在 AccountCAS 对象中的偏移量，用于 Unsafe 直接访问该属性            BALANCE_OFFSET = unsafe.objectFieldOffset(AccountCAS.class.getDeclaredField("balance"));        } catch (NoSuchFieldException | IllegalAccessException e) {            throw new Error(e);        }    }
    public AccountCAS(Integer balance) {        this.balance = balance;    }
    public int getBalance() {        return balance;    }
    public void withdraw(Integer amount) {        // 自旋        while (true) {            // 获取老的余额            int oldBalance = balance;            // 获取新的余额            int newBalance = oldBalance - amount;            // 更新余额，BALANCE_OFFSET表示balance属性的偏移量， 返回true表示更新成功， false更新失败，继续更新            if(unsafe.compareAndSwapInt(this, BALANCE_OFFSET, oldBalance, newBalance)) {                return;            }        }    }
    public static void main(String[] args) {        // 账户10000元        AccountCAS account = new AccountCAS(10000);        List<Thread> ts = new ArrayList<>();        long start = System.nanoTime();        // 1000个线程，每次取10元        for (int i = 0; i < 1000; i++) {            ts.add(new Thread(() -> {                account.withdraw(10);            }));        }        ts.forEach(Thread::start);        ts.forEach(t -> {            try {                t.join();            } catch (InterruptedException e) {                e.printStackTrace();            }        });        long end = System.nanoTime();        // 打印账户余额和花费时间        log.info("账户余额：{}, 花费时间: {}", account.getBalance(), (end-start)/1000_000 + " ms");    }}