用AtomicStampedReference解决ABA问题

在运用CAS做Lock-Free操作中有一个经典的ABA问题:

线程1准备用CAS将变量的值由A替换为B,在此之前,线程2将变量的值由A替换为C,又由C替换为A,然后线程1执行CAS时发现变量的值仍然为A,所以CAS成功。但实际上这时的现场已经和最初不同了,尽管CAS成功,但可能存在潜藏的问题,例如下面的例子:

现有一个用单向链表实现的堆栈,栈顶为A,这时线程T1已经知道A.next为B,然后希望用CAS将栈顶替换为B:

head.compareAndSet(A,B);

在T1执行上面这条指令之前,线程T2介入,将A、B出栈,再pushD、C、A,此时堆栈结构如下图,而对象B此时处于游离状态:

此时轮到线程T1执行CAS操作,检测发现栈顶仍为A,所以CAS成功,栈顶变为B,但实际上B.next为null,所以此时的情况变为:

其中堆栈中只有B一个元素,C和D组成的链表不再存在于堆栈中,平白无故就把C、D丢掉了。

以上就是由于ABA问题带来的隐患,各种乐观锁的实现中通常都会用版本戳version来对记录或对象标记,避免并发操作带来的问题,在Java中,AtomicStampedReference<E>也实现了这个作用,它通过包装[E,Integer]的元组来对对象标记版本戳stamp,从而避免ABA问题,例如下面的代码分别用AtomicInteger和AtomicStampedReference来对初始值为100的原子整型变量进行更新,AtomicInteger会成功执行CAS操作,而加上版本戳的AtomicStampedReference对于ABA问题会执行CAS失败:

public class Test {
    private static AtomicInteger atomicInt = new AtomicInteger(100);
    private static AtomicStampedReference atomicStampedRef = new AtomicStampedReference(100, 0); //核心

    public static void main(String[] args) throws InterruptedException {
       Thread intT1 = new Thread(new Runnable() {
           @Override
           public void run() {
              atomicInt.compareAndSet(100, 101);
              atomicInt.compareAndSet(101, 100);
           }
       });

       Thread intT2 = new Thread(new Runnable() {
           @Override
           public void run() {
              try {
                  TimeUnit.SECONDS.sleep(1);
              } catch (InterruptedException e) {
              }
              boolean c3 = atomicInt.compareAndSet(100, 101);
              System.out.println(c3); // true
           }
       });

       intT1.start();
       intT2.start();
       intT1.join();
       intT2.join();

       Thread refT1 = new Thread(new Runnable() {
           @Override
           public void run()
              try {
                  TimeUnit.SECONDS.sleep(1);
              } catch (InterruptedException e) {
              }
              atomicStampedRef.compareAndSet(100, 101, atomicStampedRef.getStamp(), atomicStampedRef.getStamp() + 1); //核心用法
              atomicStampedRef.compareAndSet(101, 100, atomicStampedRef.getStamp(), atomicStampedRef.getStamp() + 1);
           }
       });

       Thread refT2 = new Thread(new Runnable() {
           @Override
           public void run() {
              int stamp = atomicStampedRef.getStamp();
              try {
                  TimeUnit.SECONDS.sleep(2);
              } catch (InterruptedException e) {
              }
              boolean c3 = atomicStampedRef.compareAndSet(100, 101, stamp, stamp + 1);
              System.out.println(c3); // false
           }
       });

       refT1.start();
       refT2.start();
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在多线程编程中,CAS(Compare and Swap)机制被广泛使用。它可以实现无锁并发,提高程序的性能。但是,CAS 机制存在 ABA 问题,即当一个值从 A 变为 B,再从 B 变回 A,这时另一个线程也会执行相同的操作,而我们无法区分这两次操作是否真正修改了值。为了解决这个问题,Java 提供了一个原子类 AtomicStampedReferenceAtomicStampedReference 可以保证在进行 CAS 操作时,不仅比较对象值是否相等,还会比较对象的时间戳是否相等。时间戳是一个整数值,每次对象值的改变都会导致时间戳的变化。因此,即使对象值从 A 变为 B,再从 B 变回 A,时间戳也会发生变化,从而避免了 ABA 问题的出现。 下面是一个使用 AtomicStampedReference 解决 ABA 问题的示例代码: ```java import java.util.concurrent.atomic.AtomicStampedReference; public class AtomicStampedReferenceDemo { static AtomicStampedReference<Integer> reference = new AtomicStampedReference<>(1, 0); public static void main(String[] args) { new Thread(() -> { int stamp = reference.getStamp(); System.out.println(Thread.currentThread().getName() + " 第 1 次版本号:" + stamp); reference.compareAndSet(1, 2, stamp, stamp + 1); System.out.println(Thread.currentThread().getName() + " 第 2 次版本号:" + reference.getStamp()); reference.compareAndSet(2, 1, reference.getStamp(), reference.getStamp() + 1); System.out.println(Thread.currentThread().getName() + " 第 3 次版本号:" + reference.getStamp()); }, "线程 1").start(); new Thread(() -> { int stamp = reference.getStamp(); System.out.println(Thread.currentThread().getName() + " 第 1 次版本号:" + stamp); // 等待线程 1 完成 CAS 操作 try { Thread.sleep(1000); } catch (InterruptedException e) { e.printStackTrace(); } boolean isSuccess = reference.compareAndSet(1, 3, stamp, stamp + 1); System.out.println(Thread.currentThread().getName() + " 是否修改成功:" + isSuccess); System.out.println(Thread.currentThread().getName() + " 当前版本号:" + reference.getStamp()); System.out.println(Thread.currentThread().getName() + " 当前值:" + reference.getReference()); }, "线程 2").start(); } } ``` 输出结果: ``` 线程 1 第 1 次版本号:0 线程 1 第 2 次版本号:1 线程 1 第 3 次版本号:2 线程 2 第 1 次版本号:0 线程 2 是否修改成功:false 线程 2 当前版本号:2 线程 2 当前值:1 ``` 通过输出结果可以看出,线程 2 尝试将值从 1 改为 3,但是由于版本号已经被线程 1 修改过了,因此 CAS 操作失败,避免了 ABA 问题的出现。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值