一文吃透 Java 中的并發(fā)原子類(lèi)!
一、簡(jiǎn)介
在 Java 的java.util.concurrent包中,除了提供底層鎖、并發(fā)同步等工具類(lèi)以外,還提供了一組原子操作類(lèi),大多以Atomic開(kāi)頭,他們位于java.util.concurrent.atomic包下。
所謂原子類(lèi)操作,顧名思義,就是這個(gè)操作要么全部執(zhí)行成功,要么全部執(zhí)行失敗,是保證并發(fā)編程安全的重要一環(huán)。
相比通過(guò)synchronized和lock等方式實(shí)現(xiàn)的線程安全同步操作,原子類(lèi)的實(shí)現(xiàn)機(jī)制則完全不同。它采用的是通過(guò)無(wú)鎖(lock-free)的方式來(lái)實(shí)現(xiàn)線程安全(thread-safe)訪問(wèn),底層原理主要基于CAS操作來(lái)實(shí)現(xiàn)。
某些業(yè)務(wù)場(chǎng)景下,通過(guò)原子類(lèi)來(lái)操作,既可以實(shí)現(xiàn)線程安全的要求,又可以實(shí)現(xiàn)高效的并發(fā)性能,同時(shí)編程方面更加簡(jiǎn)單。
下面我們一起來(lái)看看它的具體玩法!
二、常用原子操作類(lèi)
在java.util.concurrent.atomic包中,因?yàn)樵宇?lèi)眾多,如果按照類(lèi)型進(jìn)行劃分,可以分為五大類(lèi),每個(gè)類(lèi)型下的原子類(lèi)可以用如下圖來(lái)概括(不同 JDK 版本,可能略有不同,本文主要基于 JDK 1.8 進(jìn)行采樣)。
圖片
雖然原子操作類(lèi)很多,但是大體的用法基本類(lèi)似,只是針對(duì)不同的數(shù)據(jù)類(lèi)型進(jìn)行了單獨(dú)適配,這些原子類(lèi)都可以保證多線程下數(shù)據(jù)的安全性,使用起來(lái)也比較簡(jiǎn)單。
2.1、基本類(lèi)型
基本類(lèi)型的原子類(lèi),也是最常用的原子操作類(lèi),JDK為開(kāi)發(fā)者提供了三個(gè)基礎(chǔ)類(lèi)型的原子類(lèi),內(nèi)容如下:
- AtomicBoolean:布爾類(lèi)型的原子操作類(lèi)
- AtomicInteger:整數(shù)類(lèi)型的原子操作類(lèi)
- AtomicLong:長(zhǎng)整數(shù)類(lèi)型的原子操作類(lèi)
以AtomicInteger為例,常用的操作方法如下:
方法 | 描述 |
| 獲取當(dāng)前值 |
| 設(shè)置 value 值 |
| 先取得舊值,然后加1,最后返回舊值 |
| 先取得舊值,然后減1,最后返回舊值 |
| 加1,然后返回新值 |
| 減1,然后返回新值 |
| 先取得舊值,然后增加指定值,最后返回舊值 |
| 增加指定值,然后返回新值 |
| 直接使用CAS方式,將【舊值】更新成【新值】,核心方法 |
AtomicInteger的使用方式非常簡(jiǎn)單,使用示例如下:
AtomicInteger atomicInteger = new AtomicInteger();
// 先獲取值,再自增,默認(rèn)初始值為0
int v1 = atomicInteger.getAndIncrement();
System.out.println("v1:" + v1);
// 獲取自增后的ID值
int v2 = atomicInteger.incrementAndGet();
System.out.println("v2:" + v2);
// 獲取自減后的ID值
int v3 = atomicInteger.decrementAndGet();
System.out.println("v3:" + v3);
// 使用CAS方式,將就舊值更新成 10
boolean v4 = atomicInteger.compareAndSet(v3,10);
System.out.println("v4:" + v4);
// 獲取最新值
int v5 = atomicInteger.get();
System.out.println("v5:" + v5);
輸出結(jié)果:
v1:0
v2:2
v3:1
v4:true
v5:10
下面我們以對(duì)某個(gè)變量累加 10000 次為例,采用 10 個(gè)線程,每個(gè)線程累加 1000 次來(lái)實(shí)現(xiàn),對(duì)比不同的實(shí)現(xiàn)方式執(zhí)行結(jié)果的區(qū)別(預(yù)期結(jié)果值為 10000)。
方式一:線程不安全操作實(shí)現(xiàn)
public class Demo1 {
/**
* 初始化一個(gè)變量
*/
private static volatile int a = 0;
public static void main(String[] args) throws InterruptedException {
final int threads = 10;
CountDownLatch countDownLatch = new CountDownLatch(threads);
for (int i = 0; i < threads; i++) {
new Thread(new Runnable() {
@Override
public void run() {
for (int j = 0; j < 1000; j++) {
a++;
}
countDownLatch.countDown();
}
}).start();
}
// 阻塞等待10個(gè)線程執(zhí)行完畢
countDownLatch.await();
// 輸出結(jié)果值
System.out.println("結(jié)果值:" + a);
}
}
輸出結(jié)果:
結(jié)果值:9527
從日志上可以很清晰的看到,實(shí)際結(jié)果值與預(yù)期不符,即使變量a加了volatile關(guān)鍵字,也無(wú)法保證累加結(jié)果的正確性。
針對(duì)volatile關(guān)鍵字,在之前的文章中我們有所介紹,它只能保證變量的可見(jiàn)性和程序的有序性,無(wú)法保證程序操作的原子性,導(dǎo)致運(yùn)行結(jié)果與預(yù)期不符。
方式二:線程同步安全操作實(shí)現(xiàn)
public class Demo2 {
/**
* 初始化一個(gè)變量
*/
private static int a = 0;
public static void main(String[] args) throws InterruptedException {
final int threads = 10;
CountDownLatch countDownLatch = new CountDownLatch(threads);
for (int i = 0; i < threads; i++) {
new Thread(new Runnable() {
@Override
public void run() {
synchronized (Demo2.class){
for (int j = 0; j < 1000; j++) {
a++;
}
}
countDownLatch.countDown();
}
}).start();
}
// 阻塞等待10個(gè)線程執(zhí)行完畢
countDownLatch.await();
// 輸出結(jié)果值
System.out.println("結(jié)果值:" + a);
}
}
輸出結(jié)果:
結(jié)果值:10000
當(dāng)多個(gè)線程操作同一個(gè)變量或者方法的時(shí)候,可以在方法上加synchronized關(guān)鍵字,可以同時(shí)實(shí)現(xiàn)變量的可見(jiàn)性、程序的有序性、操作的原子性,達(dá)到運(yùn)行結(jié)果與預(yù)期一致的效果。
同時(shí)也可以采用Lock鎖來(lái)實(shí)現(xiàn)多線程操作安全的效果,執(zhí)行結(jié)果也會(huì)與預(yù)期一致。
方式三:原子類(lèi)操作實(shí)現(xiàn)
public class Demo3 {
/**
* 初始化一個(gè)原子操作類(lèi)
*/
private static AtomicInteger a = new AtomicInteger();
public static void main(String[] args) throws InterruptedException {
final int threads = 10;
CountDownLatch countDownLatch = new CountDownLatch(threads);
for (int i = 0; i < threads; i++) {
new Thread(new Runnable() {
@Override
public void run() {
for (int j = 0; j < 1000; j++) {
// 采用原子性操作累加
a.incrementAndGet();
}
countDownLatch.countDown();
}
}).start();
}
// 阻塞等待10個(gè)線程執(zhí)行完畢
countDownLatch.await();
// 輸出結(jié)果值
System.out.println("結(jié)果值:" + a.get());
}
}
輸出結(jié)果:
結(jié)果值:10000
從日志結(jié)果上可見(jiàn),原子操作類(lèi)也可以實(shí)現(xiàn)在多線程環(huán)境下執(zhí)行結(jié)果與預(yù)期一致的效果,關(guān)于底層實(shí)現(xiàn)原理,我們等會(huì)在后文中進(jìn)行介紹。
與synchronized和Lock等實(shí)現(xiàn)方式相比,原子操作類(lèi)因?yàn)椴捎脽o(wú)鎖的方式實(shí)現(xiàn),因此某些場(chǎng)景下可以帶來(lái)更高的執(zhí)行效率。
2.2、對(duì)象引用類(lèi)型
上文提到的基本類(lèi)型的原子類(lèi),只能更新一個(gè)變量,如果需要原子性更新多個(gè)變量,這個(gè)時(shí)候可以采用對(duì)象引用類(lèi)型的原子操作類(lèi),將多個(gè)變量封裝到一個(gè)對(duì)象中,JDK為開(kāi)發(fā)者提供了三個(gè)對(duì)象引用類(lèi)型的原子類(lèi),內(nèi)容如下:
- AtomicReference:對(duì)象引用類(lèi)型的原子操作類(lèi)
- AtomicStampedReference:帶有版本號(hào)的對(duì)象引用類(lèi)型的原子操作類(lèi),可以解決 ABA 問(wèn)題
- AtomicMarkableReference:帶有標(biāo)記的對(duì)象引用類(lèi)型的原子操作類(lèi)
以AtomicReference為例,構(gòu)造一個(gè)對(duì)象引用,具體用法如下:
public class User {
private String name;
private Integer age;
public User(String name, Integer age) {
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "User{" +
"name='" + name + '\'' +
", age=" + age +
'}';
}
}
AtomicReference<User> atomicReference = new AtomicReference<>();
// 設(shè)置原始值
User user1 = new User("張三", 20);
atomicReference.set(user1);
// 采用CAS方式,將user1更新成user2
User user2 = new User("李四", 21);
atomicReference.compareAndSet(user1, user2);
System.out.println("更新后的對(duì)象:" + atomicReference.get().toString());
輸出結(jié)果:
更新后的對(duì)象:User{name='李四', age=21}
2.3、對(duì)象屬性類(lèi)型
在某項(xiàng)場(chǎng)景下,可能你只想原子性更新對(duì)象中的某個(gè)屬性值,此時(shí)可以采用對(duì)象屬性類(lèi)型的原子操作類(lèi),JDK為開(kāi)發(fā)者提供了三個(gè)對(duì)象屬性類(lèi)型的原子類(lèi),內(nèi)容如下:
- AtomicIntegerFieldUpdater:屬性為整數(shù)類(lèi)型的原子操作類(lèi)
- AtomicLongFieldUpdater:屬性為長(zhǎng)整數(shù)類(lèi)型的原子操作類(lèi)
- AtomicReferenceFieldUpdater:屬性為對(duì)象類(lèi)型的原子操作類(lèi)
需要注意的是,這些原子操作類(lèi)需要滿足以下條件才可以使用。
- 1.被操作的字段不能是 static 類(lèi)型
- 2.被操縱的字段不能是 final 類(lèi)型
- 3.被操作的字段必須是 volatile 修飾的
- 4.屬性必須對(duì)于當(dāng)前的 Updater 所在區(qū)域是可見(jiàn)的,簡(jiǎn)單的說(shuō)就是盡量使用public修飾字段
以AtomicIntegerFieldUpdater為例,構(gòu)造一個(gè)整數(shù)類(lèi)型的屬性引用,具體用法如下:
public class User {
private String name;
/**
* age 字段加上 volatile 關(guān)鍵字,并且改成 public 修飾
*/
public volatile int age;
public User(String name, int age) {
this.name = name;
this.age = age;
}
}
User user = new User("張三", 20);
AtomicIntegerFieldUpdater<User> fieldUpdater = AtomicIntegerFieldUpdater.newUpdater(User.class, "age");
// 將 age 的年齡原子性操作加 1
fieldUpdater.getAndIncrement(user);
System.out.println("更新后的屬性值:" + fieldUpdater.get(user));
輸出結(jié)果:
更新后的屬性值:21
2.4、數(shù)組類(lèi)型
數(shù)組類(lèi)型的原子操作類(lèi),并不是指對(duì)數(shù)組本身的原子操作,而是對(duì)數(shù)組中的元素進(jìn)行原子性操作,這一點(diǎn)需要特別注意,如果要針對(duì)整個(gè)數(shù)組進(jìn)行更新,可以采用對(duì)象引入類(lèi)型的原子操作類(lèi)進(jìn)行處理。
JDK為開(kāi)發(fā)者提供了三個(gè)數(shù)組類(lèi)型的原子類(lèi),內(nèi)容如下:
- AtomicIntegerArray:數(shù)組為整數(shù)類(lèi)型的原子操作類(lèi)
- AtomicLongArray:數(shù)組為長(zhǎng)整數(shù)類(lèi)型的原子操作類(lèi)
- AtomicReferenceArray:數(shù)組為對(duì)象類(lèi)型的原子操作類(lèi)
以AtomicIntegerArray為例,具體用法如下:
int[] value = new int[]{0, 3, 5};
AtomicIntegerArray array = new AtomicIntegerArray(value);
// 將下標(biāo)為[0]的元素,原子性操作加 1
array.getAndIncrement(0);
System.out.println("下標(biāo)為[0]的元素,更新后的值:" + array.get(0));
輸出結(jié)果:
下標(biāo)為[0]的元素,更新后的值:1
2.5、累加器類(lèi)型
累加器類(lèi)型的原子操作類(lèi),是從 jdk 1.8 開(kāi)始加入的,專(zhuān)門(mén)用來(lái)執(zhí)行數(shù)值類(lèi)型的數(shù)據(jù)累加操作,性能更好。
它的實(shí)現(xiàn)原理與基本數(shù)據(jù)類(lèi)型的原子類(lèi)略有不同,當(dāng)多線程競(jìng)爭(zhēng)時(shí)采用分段累加的思路來(lái)實(shí)現(xiàn)目標(biāo)值,在多線程環(huán)境中,它比AtomicLong性能要高出不少,特別是寫(xiě)多的場(chǎng)景。
JDK為開(kāi)發(fā)者提供了四個(gè)累加器類(lèi)型的原子類(lèi),內(nèi)容如下:
- LongAdder:長(zhǎng)整數(shù)類(lèi)型的原子累加操作類(lèi)
- LongAccumulator:LongAdder的功能增強(qiáng)版,它支持自定義的函數(shù)操作
- DoubleAdder:浮點(diǎn)數(shù)類(lèi)型的原子累加操作類(lèi)
- DoubleAccumulator:同樣的,也是DoubleAdder的功能增強(qiáng)版,支持自定義的函數(shù)操作
以LongAdder為例,具體用法如下:
LongAdder adder = new LongAdder();
// 自增加 1,默認(rèn)初始值為0
adder.increment();
adder.increment();
adder.increment();
System.out.println("最新值:" + adder.longValue());
輸出結(jié)果:
最新值:3
三、小結(jié)
本文主要圍繞AtomicInteger的用法進(jìn)行一次知識(shí)總結(jié),JUC包下的原子操作類(lèi)非常的多,但是大體用法基本相似,只是針對(duì)不同的數(shù)據(jù)類(lèi)型做了細(xì)分處理。
在實(shí)際業(yè)務(wù)開(kāi)發(fā)中,原子操作類(lèi)通常用于計(jì)數(shù)器,累加器等場(chǎng)景,比如編寫(xiě)一個(gè)多線程安全的全局唯一 ID 生成器。
public class IdGenerator {
private static AtomicLong atomic = new AtomicLong(0);
public long getNextId() {
return atomic.incrementAndGet();
}
}