Atomic原子类介绍

转载自https://github.com/Snailclimb/JavaGuide (添加小部分笔记)感谢作者!

文章开头先用例子介绍几种类型的api使用

package com.aqs;

import lombok.*;

import java.util.concurrent.atomic.*;

@Data
@Getter
@Setter
@AllArgsConstructor
@ToString
class User {
    private String name;
    //如果要为atomicReferenceFieldUpdater服务,必须加上volatile修饰
    public volatile Integer age;
}

public class AtomicTest {
    public static void main(String[] args) {
        System.out.println("原子更新数值---------------");
        AtomicInteger atomicInteger = new AtomicInteger();
        int i1 = atomicInteger.incrementAndGet();
        System.out.println("原子增加后为" + i1);
        System.out.println("原子更新数组---------------");

        int[] a = new int[3];
        AtomicIntegerArray atomicIntegerArray = new AtomicIntegerArray(a);

        int i = atomicIntegerArray.addAndGet(1, 3);
        System.out.println("数组元素[" + 1 + "]增加后为" + i);
        System.out.println("数组为" + atomicIntegerArray);
        System.out.println("原子更新对象---------------");
        User user1 = new User("ly1", 10);
        User user2 = new User("ly2", 20);
        User user3 = new User("ly3", 30);
        AtomicReference<User> atomicReference = new AtomicReference<>(user1);
        boolean b = atomicReference.compareAndSet(user2, user3);
        System.out.println("更新" + (b ? "成功" : "失败"));
        System.out.println("引用里值为"+atomicReference.get());
        boolean b1 = atomicReference.compareAndSet(user1, user3);
        System.out.println("更新" + (b1 ? "成功" : "失败"));
        System.out.println("引用里值为"+atomicReference.get());
        System.out.println("原子更新对象属性---------------");
        User user4=new User("ly4",40);
        AtomicReferenceFieldUpdater<User, Integer> atomicReferenceFieldUpdater = AtomicReferenceFieldUpdater.newUpdater(User.class, Integer.class, "age");
        boolean b2 = atomicReferenceFieldUpdater.compareAndSet(user4, 41, 400);
        System.out.println("更新"+(b2?"成功":"失败"));
        System.out.println("引用里user4值为"+atomicReferenceFieldUpdater.get(user4));
        boolean b3 = atomicReferenceFieldUpdater.compareAndSet(user4, 40, 400);
        System.out.println("更新"+(b3?"成功":"失败"));
        System.out.println("引用里user4值为"+atomicReferenceFieldUpdater.get(user4));
        System.out.println("其他使用---------------");
        User user5=new User("ly5",50);
        User user6=new User("ly6",60);
        User user7=new User("ly7",70);
        AtomicMarkableReference<User> userAtomicMarkableReference=new AtomicMarkableReference<>(user5,true);
        boolean b4 = userAtomicMarkableReference.weakCompareAndSet(user6, user7, true, false);
        System.out.println("更新"+(b4?"成功":"失败"));
        System.out.println("引用里值为"+userAtomicMarkableReference.getReference());
        boolean b5 = userAtomicMarkableReference.weakCompareAndSet(user5, user7, false, true);
        System.out.println("更新"+(b5?"成功":"失败"));
        System.out.println("引用里值为"+userAtomicMarkableReference.getReference());
        boolean b6 = userAtomicMarkableReference.weakCompareAndSet(user5, user7, true, false);
        System.out.println("更新"+(b6?"成功":"失败"));
        System.out.println("引用里值为"+userAtomicMarkableReference.getReference());
        System.out.println("AtomicStampedReference使用---------------");
        User user80=new User("ly8",80);
        User user90=new User("ly9",90);
        User user100=new User("ly10",100);
        AtomicStampedReference<User> userAtomicStampedReference=new AtomicStampedReference<>(user80,80);//版本80
        //...每次更改stamp都加1
        //这里假设中途被改成81了
        boolean b7 = userAtomicStampedReference.compareAndSet(user80, user100,81,90);
        System.out.println("更新"+(b7?"成功":"失败"));
        System.out.println("引用里值为"+userAtomicStampedReference.getReference());
        boolean b8 = userAtomicStampedReference.compareAndSet(user80, user100,80,90);
        System.out.println("更新"+(b8?"成功":"失败"));
        System.out.println("引用里值为"+userAtomicStampedReference.getReference());
    }
}
/*
原子更新数值---------------
原子增加后为1
原子更新数组---------------
数组元素[1]增加后为3
数组为[0, 3, 0]
原子更新对象---------------
更新失败
引用里值为User(name=ly1, age=10)
更新成功
引用里值为User(name=ly3, age=30)
原子更新对象属性---------------
更新失败
引用里user4值为40
更新成功
引用里user4值为400
其他使用---------------
更新失败
引用里值为User(name=ly5, age=50)
更新失败
引用里值为User(name=ly5, age=50)
更新成功
引用里值为User(name=ly7, age=70)
AtomicStampedReference使用---------------
更新失败
引用里值为User(name=ly8, age=80)
更新成功
引用里值为User(name=ly10, age=100)

Process finished with exit code 0

*/

原子类介绍 #

  • 在化学上,原子是构成一般物质的最小单位,化学反应中是不可分割的,Atomic指一个操作是不可中断的,即使在多个线程一起执行时,一个操作一旦开始就不会被其他线程干扰
  • 原子类–>具有原子/原子操作特征的类
  • 并发包java.util.concurrent 的原子类都放着java.util.concurrent.atomicly-20241212141944757
  • 根据操作的数据类型,可以将JUC包中的原子类分为4类(基本类型、数组类型、引用类型、对象的属性修改类型)
    • 基本类型 使用原子方式更新基本类型,包括AtomicInteger 整型原子类AtomicLong 长整型原子类,AtomicBoolean 布尔型原子类

    • 数组类型 使用原子方式更新数组里某个元素,包括AtomicIntegerArray 整型数组原子类AtomicLongArray 长整型数组原子类AtomicReferenceArray 引用类型数组原子类

    • 引用类型 AtomicReference 引用类型原子类,AtomicMarkableReference 原子更新带有标记的引用类型,该类将boolean标记与引用关联(不可解决CAS进行原子操作出现的ABA问题),AtomicStampedReference 原子更新带有版本号的引用类型 该类将整数值与引用关联,可用于解决原子更新数据和数据的版本号(解决使用CAS进行原子更新时可能出现的ABA问题)

    • 对象的属性修改类型 AtomicIntegerFieldUpdater 原子更新整型字段的更新器AtomicLongFieldUpdater 原子更新长整型字段的更新器AtomicReferenceFieldUpdater 原子更新引用类型里的字段

    • AtomicMarkableReference 不能解决 ABA 问题

      public class SolveABAByAtomicMarkableReference {
      
          private static AtomicMarkableReference atomicMarkableReference = new AtomicMarkableReference(100, false);
      
          public static void main(String[] args) {
      
              Thread refT1 = new Thread(() -> {
                  try {
                      TimeUnit.SECONDS.sleep(1);
                  } catch (InterruptedException e) {
                      e.printStackTrace();
                  }
                  atomicMarkableReference.compareAndSet(100, 101, atomicMarkableReference.isMarked(), !atomicMarkableReference.isMarked());//根据期望值100和false 修改为101和true
                  atomicMarkableReference.compareAndSet(101, 100, atomicMarkableReference.isMarked(), !atomicMarkableReference.isMarked());//根据期望值101和true 修改为100和false
              });
      
              Thread refT2 = new Thread(() -> {
                  //获取原来的marked标记(false)
                  boolean marked = atomicMarkableReference.isMarked();
                  //2s之后进行替换,不应该替换成功
                  try {
                      TimeUnit.SECONDS.sleep(2);
                  } catch (InterruptedException e) {
                      e.printStackTrace();
                  }
                  boolean c3 = atomicMarkableReference.compareAndSet(100, 101, marked, !marked);
                  System.out.println(c3); // 返回true,实际应该返回false
              }); //导致了ABA问题
      
              refT1.start();
              refT2.start();
          }
      }
      
    • CAS ABA问题

      描述: 第一个线程取到了变量 x 的值 A,然后巴拉巴拉干别的事,总之就是只拿到了变量 x 的值 A。这段时间内第二个线程也取到了变量 x 的值 A,然后把变量 x 的值改为 B,然后巴拉巴拉干别的事,最后又把变量 x 的值变为 A (相当于还原了)。在这之后第一个线程终于进行了变量 x 的操作,但是此时变量 x 的值还是 A,所以 compareAndSet 操作是成功。


      也就是说,线程一无法保证自己操作期间,该值被修改了

    • 例子描述(可能不太合适,但好理解): 年初,现金为零,然后通过正常劳动赚了三百万,之后正常消费了(比如买房子)三百万。年末,虽然现金零收入(可能变成其他形式了),但是赚了钱是事实,还是得交税的!

    • 代码描述

      import java.util.concurrent.atomic.AtomicInteger;
      
      public class AtomicIntegerDefectDemo {
          public static void main(String[] args) {
              defectOfABA();
          }
      
          static void defectOfABA() {
              final AtomicInteger atomicInteger = new AtomicInteger(1);
      
              Thread coreThread = new Thread(
                      () -> {
                          final int currentValue = atomicInteger.get();
                          System.out.println(Thread.currentThread().getName() + " ------ currentValue=" + currentValue);
      
                          // 这段目的:模拟处理其他业务花费的时间
                          //也就是说,在差值300-100=200ms内,值被操作了两次(但又改回去了),然后线程coreThread并没有感知到,当作没有修改过来处理
                          try {
                              Thread.sleep(300);
                          } catch (InterruptedException e) {
                              e.printStackTrace();
                          }
      
                          boolean casResult = atomicInteger.compareAndSet(1, 2);
                          System.out.println(Thread.currentThread().getName()
                                  + " ------ currentValue=" + currentValue
                                  + ", finalValue=" + atomicInteger.get()
                                  + ", compareAndSet Result=" + casResult);
                      }
              );
              coreThread.start();
      
              // 这段目的:为了让 coreThread 线程先跑起来
              try {
                  Thread.sleep(100);
              } catch (InterruptedException e) {
                  e.printStackTrace();
              }
      
              Thread amateurThread = new Thread(
                      () -> {
                          int currentValue = atomicInteger.get();
                          boolean casResult = atomicInteger.compareAndSet(1, 2);
                          System.out.println(Thread.currentThread().getName()
                                  + " ------ currentValue=" + currentValue
                                  + ", finalValue=" + atomicInteger.get()
                                  + ", compareAndSet Result=" + casResult);
      
                          currentValue = atomicInteger.get();
                          casResult = atomicInteger.compareAndSet(2, 1);
                          System.out.println(Thread.currentThread().getName()
                                  + " ------ currentValue=" + currentValue
                                  + ", finalValue=" + atomicInteger.get()
                                  + ", compareAndSet Result=" + casResult);
                      }
              );
              amateurThread.start();
          }
      } 
      /*输出内容
       Thread-0 ------ currentValue=1
      Thread-1 ------ currentValue=1, finalValue=2, compareAndSet Result=true
      Thread-1 ------ currentValue=2, finalValue=1, compareAndSet Result=true
      Thread-0 ------ currentValue=1, finalValue=2, compareAndSet Result=true 
      */
      

基本类型原子类 #

  • 使用原子方式更新基本类型:AtomicInteger 整型原子类,AtomicLong 长整型原子类 ,AtomicBoolean 布尔型原子类,下文以AtomicInteger为例子来介绍 常用方法:

    public final int get() //获取当前的值
    public final int getAndSet(int newValue)//获取当前的值,并设置新的值
    public final int getAndIncrement()//获取当前的值,并自增
    public final int getAndDecrement() //获取当前的值,并自减
    public final int getAndAdd(int delta) //获取当前的值,并加上预期的值
    boolean compareAndSet(int expect, int update) //如果输入的数值等于预期值,则以原子方式将该值设置为输入值(update)
    public final void lazySet(int newValue)//最终设置为newValue,使用 lazySet 设置之后可能导致其他线程在之后的一小段时间内还是可以读到旧的值。
    

    常见方法使用

    import java.util.concurrent.atomic.AtomicInteger;
    
    public class AtomicIntegerTest {
    
    	public static void main(String[] args) {
    		// TODO Auto-generated method stub
    		int temvalue = 0;
    		AtomicInteger i = new AtomicInteger(0);
    		temvalue = i.getAndSet(3);
    		System.out.println("temvalue:" + temvalue + ";  i:" + i);//temvalue:0;  i:3
    		temvalue = i.getAndIncrement();
    		System.out.println("temvalue:" + temvalue + ";  i:" + i);//temvalue:3;  i:4
    		temvalue = i.getAndAdd(5);
    		System.out.println("temvalue:" + temvalue + ";  i:" + i);//temvalue:4;  i:9
    	}
    
    } 
    

基本数据类型原子类的优势 #

  • 多线程环境不使用原子类保证线程安全(基本数据类型)

    class Test {
            private volatile int count = 0;
            //若要线程安全执行执行count++,需要加锁
            public synchronized void increment() {
                      count++;
            }
    
            public int getCount() {
                      return count;
            }
    } 
    
  • 多线程环境使用原子类保证线程安全(基本数据类型)

    class Test2 {
            private AtomicInteger count = new AtomicInteger();
    
    	    //使用AtomicInteger之后,不需要加锁,也可以实现线程安全。
            public void increment() {
                      count.incrementAndGet();
            }    
            public int getCount() {
                    return count.get();
            }
    } 
    

AtomicInteger线程安全原理简单分析 #

部分源码:

/ setup to use Unsafe.compareAndSwapInt for updates更新操作时提供比较并替换的作用
    private static final Unsafe unsafe = Unsafe.getUnsafe();
    private static final long valueOffset;

    static {
        try {
            valueOffset = unsafe.objectFieldOffset
                (AtomicInteger.class.getDeclaredField("value"));
        } catch (Exception ex) { throw new Error(ex); }
    }

    private volatile int value; 
  1. AtomicInteger类主要利用CAS(compare and swap) + volatilenative方法来保证原子操作,从而避免synchronized高开销,提高执行效率
  2. CAS的原理是拿期望的值原本的值做比较,如果相同则更新成新值 UnSafe类的objectFieldOffset()方法是一个本地方法,这个方法用来拿到**“原来的值"的内存地址**
  3. value是一个volatile变量,在内存中可见,因此JVM可以保证任何时刻任何线程总能拿到该变量的最新值

数组类型原子类 #

使用原子的方式更新数组里的某个元素

AtomicIntegerArray 整型数组原子类,AtomicLongArray 长整型数组原子类,AtomicReferenceArray 引用类型数组原子类

常用方法:

public final int get(int i) //获取 index=i 位置元素的值
public final int getAndSet(int i, int newValue)//返回 index=i 位置的当前的值,并将其设置为新值:newValue
public final int getAndIncrement(int i)//获取 index=i 位置元素的值,并让该位置的元素自增
public final int getAndDecrement(int i) //获取 index=i 位置元素的值,并让该位置的元素自减
public final int getAndAdd(int i, int delta) //获取 index=i 位置元素的值,并加上预期的值
boolean compareAndSet(int i, int expect, int update) //如果输入的数值等于预期值,则以原子方式将 index=i 位置的元素值设置为输入值(update)
public final void lazySet(int i, int newValue)//最终 将index=i 位置的元素设置为newValue,使用 lazySet 设置之后可能导致其他线程在之后的一小段时间内还是可以读到旧的值。 

常见方法使用

import java.util.concurrent.atomic.AtomicIntegerArray;

public class AtomicIntegerArrayTest {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		int temvalue = 0;
		int[] nums = { 1, 2, 3, 4, 5, 6 };
		AtomicIntegerArray i = new AtomicIntegerArray(nums);
		for (int j = 0; j < nums.length; j++) {
			System.out.println(i.get(j));
		}
		temvalue = i.getAndSet(0, 2);
		System.out.println("temvalue:" + temvalue + ";  i:" + i);
		temvalue = i.getAndIncrement(0);
		System.out.println("temvalue:" + temvalue + ";  i:" + i);
		temvalue = i.getAndAdd(0, 5);
		System.out.println("temvalue:" + temvalue + ";  i:" + i);
	}

} 

引用类型原子类 #

基本类型原子类只能更新一个变量,如果需要原子更新多个变量,则需要使用引用类型原子类

AtomicReference 引用类型原子类;
AtomicStampedReference 原子更新带有版本号的引用类型,该类将整数值与引用关联起来,可用于解决原子的更新数据和数据的版本号,可以解决使用CAS进行原子更新时可能出现的ABA问题;
AtomicMarkableReference:原子更新带有标记的引用类型。该类将boolean标记与引用关联**(注:无法解决ABA问题)**

下面以AtomicReference为例介绍

import java.util.concurrent.atomic.AtomicReference;

public class AtomicReferenceTest {

	public static void main(String[] args) {
		AtomicReference<Person> ar = new AtomicReference<Person>();
		Person person = new Person("SnailClimb", 22);
		ar.set(person);
		Person updatePerson = new Person("Daisy", 20);
		ar.compareAndSet(person, updatePerson);//如果期望值为person,则替换成updatePerson

		System.out.println(ar.get().getName());
		System.out.println(ar.get().getAge());
	}
}

class Person {
	private String name;
	private int age;

	public Person(String name, int age) {
		super();
		this.name = name;
		this.age = age;
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public int getAge() {
		return age;
	}

	public void setAge(int age) {
		this.age = age;
	}

} 

上述代码首先创建了一个 Person 对象,然后把 Person 对象设置进 AtomicReference 对象中,然后调用 compareAndSet 方法,该方法就是通过 CAS 操作设置 ar。如果 ar 的值为 person 的话,则将其设置为 updatePerson。实现原理与 AtomicInteger 类中的 compareAndSet 方法相同。运行上面的代码后的输出结果如下

Daisy
20

AtomicStampedReference类使用示例

import java.util.concurrent.atomic.AtomicStampedReference;

public class AtomicStampedReferenceDemo {
    public static void main(String[] args) {
        // 实例化、取当前值和 stamp 值
        final Integer initialRef = 0, initialStamp = 0;
        final AtomicStampedReference<Integer> asr = new AtomicStampedReference<>(initialRef, initialStamp);
        System.out.println("currentValue=" + asr.getReference() + ", currentStamp=" + asr.getStamp());

        // compare and set
        final Integer newReference = 666, newStamp = 999;
        final boolean casResult = asr.compareAndSet(initialRef, newReference, initialStamp, newStamp);
        System.out.println("currentValue=" + asr.getReference()
                + ", currentStamp=" + asr.getStamp()
                + ", casResult=" + casResult);

        // 获取当前的值和当前的 stamp 值
        int[] arr = new int[1];
        final Integer currentValue = asr.get(arr);
        final int currentStamp = arr[0];
        System.out.println("currentValue=" + currentValue + ", currentStamp=" + currentStamp);

        // 单独设置 stamp 值
        final boolean attemptStampResult = asr.attemptStamp(newReference, 88);
        System.out.println("currentValue=" + asr.getReference()
                + ", currentStamp=" + asr.getStamp()
                + ", attemptStampResult=" + attemptStampResult);

        // 重新设置当前值和 stamp 值
        asr.set(initialRef, initialStamp);
        System.out.println("currentValue=" + asr.getReference() + ", currentStamp=" + asr.getStamp());

        // [不推荐使用,除非搞清楚注释的意思了] weak compare and set
        // 困惑!weakCompareAndSet 这个方法最终还是调用 compareAndSet 方法。[版本: jdk-8u191]
        // 但是注释上写着 "May fail spuriously and does not provide ordering guarantees,
        // so is only rarely an appropriate alternative to compareAndSet."
        // todo 感觉有可能是 jvm 通过方法名在 native 方法里面做了转发
        final boolean wCasResult = asr.weakCompareAndSet(initialRef, newReference, initialStamp, newStamp);
        System.out.println("currentValue=" + asr.getReference()
                + ", currentStamp=" + asr.getStamp()
                + ", wCasResult=" + wCasResult);
    }
}
/* 结果
 currentValue=0, currentStamp=0
currentValue=666, currentStamp=999, casResult=true
currentValue=666, currentStamp=999
currentValue=666, currentStamp=88, attemptStampResult=true
currentValue=0, currentStamp=0
currentValue=666, currentStamp=999, wCasResult=true 
*/
  • AtomicMarkableReference 类使用示例

    import java.util.concurrent.atomic.AtomicMarkableReference;
    
    public class AtomicMarkableReferenceDemo {
        public static void main(String[] args) {
            // 实例化、取当前值和 mark 值
            final Boolean initialRef = null, initialMark = false;
            final AtomicMarkableReference<Boolean> amr = new AtomicMarkableReference<>(initialRef, initialMark);
            System.out.println("currentValue=" + amr.getReference() + ", currentMark=" + amr.isMarked());
    
            // compare and set
            final Boolean newReference1 = true, newMark1 = true;
            final boolean casResult = amr.compareAndSet(initialRef, newReference1, initialMark, newMark1);
            System.out.println("currentValue=" + amr.getReference()
                    + ", currentMark=" + amr.isMarked()
                    + ", casResult=" + casResult);
    
            // 获取当前的值和当前的 mark 值
            boolean[] arr = new boolean[1];
            final Boolean currentValue = amr.get(arr);
            final boolean currentMark = arr[0];
            System.out.println("currentValue=" + currentValue + ", currentMark=" + currentMark);
    
            // 单独设置 mark 值
            final boolean attemptMarkResult = amr.attemptMark(newReference1, false);
            System.out.println("currentValue=" + amr.getReference()
                    + ", currentMark=" + amr.isMarked()
                    + ", attemptMarkResult=" + attemptMarkResult);
    
            // 重新设置当前值和 mark 值
            amr.set(initialRef, initialMark);
            System.out.println("currentValue=" + amr.getReference() + ", currentMark=" + amr.isMarked());
    
            // [不推荐使用,除非搞清楚注释的意思了] weak compare and set
            // 困惑!weakCompareAndSet 这个方法最终还是调用 compareAndSet 方法。[版本: jdk-8u191]
            // 但是注释上写着 "May fail spuriously and does not provide ordering guarantees,
            // so is only rarely an appropriate alternative to compareAndSet."
            // todo 感觉有可能是 jvm 通过方法名在 native 方法里面做了转发
            final boolean wCasResult = amr.weakCompareAndSet(initialRef, newReference1, initialMark, newMark1);
            System.out.println("currentValue=" + amr.getReference()
                    + ", currentMark=" + amr.isMarked()
                    + ", wCasResult=" + wCasResult);
        }
    }
    /* 结果
     currentValue=null, currentMark=false
    currentValue=true, currentMark=true, casResult=true
    currentValue=true, currentMark=true
    currentValue=true, currentMark=false, attemptMarkResult=true
    currentValue=null, currentMark=false
    currentValue=true, currentMark=true, wCasResult=true
    
    */
    

对象的属性修改类型原子类 #

对象的属性修改类型原子类,用来原子更新某个类里的某个字段

包括: AtomicIntegerFieldUpdater 原子更新整型字段的更新器,AtomicLongFieldUpdater 原子更新长整型字段的更新器,AtomicReferenceFieldUpdater 原子更新引用类型里的字段的更新器

原子地更新对象属性需要两步骤:

  1. 对象的属性修改类型原子类都是抽象类,所以每次使用都必须使用静态方法newUpdater()创建一个更新器,且设置想要更新的属性
  2. 更新的对象属性必须使用public volatile修饰符

下面以AtomicIntegerFieldUpdater为例子来介绍

import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;

public class AtomicIntegerFieldUpdaterTest {
	public static void main(String[] args) {
		AtomicIntegerFieldUpdater<User> a = AtomicIntegerFieldUpdater.newUpdater(User.class, "age");

		User user = new User("Java", 22);
		System.out.println(a.getAndIncrement(user));// 22
		System.out.println(a.get(user));// 23
	}
}

class User {
	private String name;
	public volatile int age;

	public User(String name, int age) {
		super();
		this.name = name;
		this.age = age;
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public int getAge() {
		return age;
	}

	public void setAge(int age) {
		this.age = age;
	}

}
/* 结果
 22 
 33 
*/