Java并发之原子类

作者: 第四单元 | 来源:发表于2018-12-15 11:37 被阅读1次

一.从i++说起

i++这个看是简单的操作其实不是一个原子操作,它是由三步组成的。取出i的值,进行加一操作,写会计算后的值。在多线程竞争环境下,如果多个线程同时调用包含i++的一段代码,就可能出现错误(一个线程将另一个线程的修改覆盖的问题)。

当然可以使用synchronized重量级锁来进行代码块互斥访问。但就i++这一句代码来说还有更轻量级的方式,那就是原子类AtomicInteger的getAndIncrement()方法。

二.AtomicInteger

AtomicInteger位于java.util.concurrent.atomic包下。他提供了一种用法简单、性能高效、线程安全地更新一个Integer变量的方式。

2.1 AtomicInteger的用法
AtomicInteger a = new AtomicInteger(0);

a.getAndIncrement(); 
a.getAndDecrement();
a.compareAndSet(int expect,int update); //返回值true or false
a.getAndAdd(int delta);
2.2 原理——以getAndIncrement()为例

AtomicIntger实现的原子操作是基于Unsafe类CAS操作实现的。CAS是指Compare And Swap即比较并交换。只有当变量的实际值和我们传入的期望值一致时,才进行交换操作(用传入的新值替换旧值)。

Unsafe类提供了三种基本的CAS操作。分别是CAS更新int,CAS更新long,CAS更新Object(引用类型)。


/**
*解释一下参数含义,其它两个类似
*@param var1 要更新的变量,一个Integer变量
*@param var2 Integer对象中实际存储int值位置的偏移量(个人理解,欢迎指正)
*@param var4 期望值
*@param var5 更新为这个值
*方法语义:
*    如果var1这个Integer对象在其偏移var2位置的值等于var4,则将该值改为var5,否则不改。
*    以上为原子操作,不会被打断,要么和期望相同修改成功,要么不同修改失败。
*/
public final native boolean compareAndSwapInt(Object var1, long var2, int var4, int var5);

public final native boolean compareAndSwapLong(Object var1, long var2, long var4, long var6);

public final native boolean compareAndSwapObject(Object var1, long var2, Object var4, Object var5);

Unsafe类中基于基本的CAS操作又实现了getAndInt()等方法,可以无阻塞地实现原子更新操作,不需加锁,不涉及线程阻塞等。源码如下:

//Unsafe中getAndAddInt的定义,语义是在原变量上加上var4
public final int getAndAddInt(Object var1, long var2, int var4) {
    int var5;
    do {
        //取出现有的值
        var5 = this.getIntVolatile(var1, var2);  
        //计算处相加后的值,并进行CAS操作,如果失败说明现有值被修改,需要重新计算相加后的值。
        //如果成功循环结束,返回相加操作之前的值
    } while(!this.compareAndSwapInt(var1, var2, var5, var5 + var4));

    return var5;
}

而AtomicInteger中的getAndAdd、getAndIncrement等方法就是直接调用了Unsafe.getAndAddInt方法:

//AtomicInteger中getAndAdd的定义
public final int getAndAdd(int delta) {
        return unsafe.getAndAddInt(this, valueOffset, delta);
}

//AtomicInteger getAndIncrement定义
public final int getAndIncrement() {
        return unsafe.getAndAddInt(this, valueOffset, 1);
}

看到这里就可以明白用AtomicInteger实现“i++”是如何做到线程安全的了。上文说道i++分为三步:取值,计算,写回。但若是写回时,原值已经被其它线程改变了,那这次的计算就不正确了。unsafe.getAndAddInt中,通过CAS操作避免了这种情况,如果写回时值改变了就重新取值,重新计算并写回,一直循环直到成功。

三.其它原子类的使用

java.util.concurrent.atomic包中除了AtomicIntger,还有很多其它原子类,它们的实现原理都可以追根溯源到unsafe的三个CAS操作,这里不做详细记录了,只类别各举一个例子简单记录一下它们的使用,详细的使用规则,可在用到时参考JDK API文档和JDK源码

3.1 原子更新基本类型

AtomicBoolean
AtomicInteger
AtomicLong

使用AtomicBoolean实现多线程下某一段代码只执行一次。

AtomicBoolean isHappen= new AtomicBoolean(false);

if(isHappen.compareAndSet(false,true)) {
  //多线程下这一段代码也只会执行一次
  do something、、

}

3.2原子更新数组

AtomicIntegerArray
AtomicLongArray
AtomicReferenceArray

AtomicIntegerArray API举例:

//初始化
AtomicIntegerArray ai = new AtomicIntegerArray(new int[]{1,2,3,4,5});

//以原子方式将输入值与数组中索引i的元素相加
int addAndGet(int i,int delta); 
//如果当前值等于预期值,则以原子方式将数组位置i的元素设置成update值
boolean compareAndSet(int i,int expect,int update);

3.3原子更新引用类型

AtomicReference
AtomicStampedRference
AtomicMarkableReference

ABA问题

ABA问题是指在进行CAS操作时可能变量已经变了,而且变了多次,但变化的结果和原来的一致,即又变回初始值了。这时其实已经变过了,但CAS认为它和旧值一样,就认为没有变,其实已经变过了。

AtomicStampedReference解决ABA问题

解决的办法就是记录版本号。AtomicReference只需记录一个T reference代表要操作的引用。而AtomicStampedReference使用一个Pair泛型类同时记录reference和stamp。进行CAS操作时比较的是这个pair。Pair中的reference和stamp都是final的,一旦初始化就不会被修改了。

使用AtomicStampedReference主要使用的是其compareAndSet方法.

    public boolean compareAndSet(V   expectedReference,
                                 V   newReference,
                                 int expectedStamp,
                                 int newStamp) {
        Pair<V> current = pair;
        return
//先判断期望值和当前值是否相同,不同也没有必要进行CAS了
            expectedReference == current.reference &&  
            expectedStamp == current.stamp &&
//如果新值和当前值已经一样了,也不需更新了。
            ((newReference == current.reference &&
              newStamp == current.stamp) ||
//进行CAS:用新值生成新的Pair,如果实际值和传入的当前值pair是同一个,则进行替换。
//底层使用unsafe的compareAndSwapObject方法比较引用类型
             casPair(current, Pair.of(newReference, newStamp)));
    }

3.4原子更新字段类

AtomicIntegerFieldUpdater
AtomicLongFieldUpdater
AtomicReferenceFieldUpdater

//可用于原子地更新User类的old字段。old字段必须是public volatile修饰的。
AtomicIntegerFieldUpdater<User> a = 
                      AtomicIntegerFieldUpdater.newUpdater(User.class,"old");

User conan = new User("conan",10);
a.getAndIncrement(conan);

相关文章

网友评论

    本文标题:Java并发之原子类

    本文链接:https://www.haomeiwen.com/subject/mavohqtx.html