Java源码系列(7) -- AtomicInteger

January 17, 2018

一、前言

AtomicInteger基于CAS(Compare and Swap,比较并修改)的操作,主要实现乐观锁的思想。

对悲观锁来说,它会假设线程并发竞争非常严重,每次修改数据先100%确保自己获得锁并进入临界区,再安心修改目标值,进而出现线程在竞争锁的过程中消耗大量时间在等锁、加锁、解锁等操作上。(注:锁还会涉及锁自旋、公平锁、锁队列等知识点)

相比之下的乐观锁,会假设线程冲突不严重,先比较修改前的值是不是和自己的预期值一致,一致就修改并返回,不一致就放弃这次修改并发起下一次尝试,不需要把时间用在加锁和解锁上。

虽然乐观锁看起来比悲观锁好很多,不过乐观锁主要用在单个值(Int,Float,Double)的并发修改上,而不是悲观锁对一个对象甚至是一个代码块的操作。

而CAS也有自己最合适的应用场景:如乐观锁对目标值修改写入次数远多于读取操作,那CAS实际也会出现大量写竞争失败,造成可观的时间成本。

说到CAS,需要了解的是:这个能力并非由操作系统或JVM提供,而是CPU原生支持的。在JDK源码中,有一段代码调用处理器CAS指令"cmpxchgl %1,(%3)"。如果CPU不能保证CAS能力,那这个CPU完全没有数据修改安全可言。

二、类签名

由于继承Number,所以任何Number类型参数都能使用AtomicInteger

1
public class AtomicInteger extends Number implements java.io.Serializable

三、静态初始化

静态初始化块获取value的内存地址,而静态初始化块是类初始化中最先调用的,静态初始化线程安全由JVM来保证。

1
2
3
4
5
6
7
static {
    try {
        // 反射出value的变量并查找变量在类内存结构的偏移值
        valueOffset = unsafe.objectFieldOffset
            (AtomicInteger.class.getDeclaredField("value"));
    } catch (Exception ex) { throw new Error(ex); }
}

四、数据成员

volatile保证value值的有序性和线程可见性,而原子性需synchronizedLock来保障。

1
2
3
4
5
private volatile int value;

// setup to use Unsafe.compareAndSwapInt for updates
private static final Unsafe unsafe = Unsafe.getUnsafe();
private static final long valueOffset;

五、构造方法

用给定整形值初始化实例

1
2
3
public AtomicInteger(int initialValue) {
    value = initialValue;
}

初始值为0的实例

1
2
public AtomicInteger() {
}

六、成员方法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
// 获取当前整形值
public final int get() {
    return value;
}

// 设置新整形值
public final void set(int newValue) {
    value = newValue;
}

// 最终会把newValue设置成功
public final void lazySet(int newValue) {
    unsafe.putOrderedInt(this, valueOffset, newValue);
}

// 设置新整形值,返回上次保存的值
public final int getAndSet(int newValue) {
    return unsafe.getAndSetInt(this, valueOffset, newValue);
}

如果待修改的值和期待值相同,那就把待修改的值设置为update的值

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
public final boolean compareAndSet(int expect, int update) {
    return unsafe.compareAndSwapInt(this, valueOffset, expect, update);
}

public final boolean weakCompareAndSet(int expect, int update) {
    return unsafe.compareAndSwapInt(this, valueOffset, expect, update);
}

// 先返回上一个值,然后在原基础上自增1
public final int getAndIncrement() {
    return unsafe.getAndAddInt(this, valueOffset, 1);
}

// 先返回上一个值,然后在原基础上自减1
public final int getAndDecrement() {
    return unsafe.getAndAddInt(this, valueOffset, -1);
}

// 返回上一个值,并在原基础上加上指定值
// 伪代码: oldValue = value; value += delta; return oldValue; 
public final int getAndAdd(int delta) {
    return unsafe.getAndAddInt(this, valueOffset, delta);
}

// 先自增,然后返回自增后的值
public final int incrementAndGet() {
    return unsafe.getAndAddInt(this, valueOffset, 1) + 1;
}

// 先自减,然后返回自减后的值
public final int decrementAndGet() {
    return unsafe.getAndAddInt(this, valueOffset, -1) - 1;
}

// 先增加delta的值,然后返回增加后的值
public final int addAndGet(int delta) {
    return unsafe.getAndAddInt(this, valueOffset, delta) + delta;
}

七、参考链接