在併發程式設計中很容易出現併發安全問題，最簡單的例子就是多執行緒更新變數i=1，多個執行緒執行i++操作，就有可能獲取不到正確的值，而這個問題，最常用的方法是通過Synchronized進行控制來達到執行緒安全的目的。但是由於synchronized是採用的是悲觀鎖策略，並不是特別高效的一種解決方案。實際上，在J.U.C下的Atomic包提供了一系列的操作簡單，效能高效，並能保證執行緒安全的類去更新多種型別。Atomic包下的這些類都是採用樂觀鎖策略CAS來更新資料。

CAS原理與問題

CAS操作（又稱為無鎖操作）是一種樂觀鎖策略。它假設所有執行緒訪問共享資源的時候不會出現衝突，因此不會阻塞其他執行緒的操作。那麼，如果出現衝突了怎麼辦？無鎖操作是使用CAS(compare and swap)來鑑別執行緒是否出現衝突，出現衝突就重試當前操作直到沒有衝突為止。

CAS的操作過程

舉例說明：
Atomic包中的AtomicInteger類，是通過Unsafe類下的native函式compareAndSwapInt自旋來保證原子性，
其中incrementAndGet函式呼叫的getAndAddInt函式如下所示：

public final int getAndAddInt(Object var1, long var2, int var4) {
        int var5;
        do {
            var5 = this.getIntVolatile(var1, var2);
        } while(!this.compareAndSwapInt(var1, var2, var5, var5 + var4));

        return var5;
    }
 

CAS有3個運算元，記憶體值V，舊的預期值A，要修改的新值B。當且僅當預期值A和記憶體值V相同時，將記憶體值V修改為B，否則什麼都不做。
可見只有自旋實現更新資料操作之後，while迴圈才能夠結束。

CAS的問題

1. 自旋時間過長。由compareAndSwapInt函式可知，自旋時間過長會對效能是很大的消耗。
2. ABA問題。因為CAS會檢查舊值有沒有變化，這裡存在這樣一個有意思的問題。比如一箇舊值A變為了成B，然後再變成A，剛好在做CAS時檢查發現舊值並沒有變化依然為A，但是實際上的確發生了變化。解決方案可以新增一個版本號可以解決。原來的變化路徑A->B->A就變成了1A->2B->3C，或使用AtomicStampedReference工具類。

Atomic包的使用

原子更新基本型別

Atomic包中原子更新基本型別的工具類：
AtomicBoolean：以原子更新的方式更新boolean；
AtomicInteger：以原子更新的方式更新Integer;
AtomicLong：以原子更新的方式更新Long；

這幾個類的用法基本一致，這裡以AtomicInteger為例總結常用的方法

1. addAndGet(int delta)：以原子方式將輸入的數值與例項中原本的值相加，並返回最後的結果；
2. incrementAndGet() ：以原子的方式將例項中的原值進行加1操作，並返回最終相加後的結果；
3. getAndSet(int newValue)：將例項中的值更新為新值，並返回舊值；
4. getAndIncrement()：以原子的方式將例項中的原值加1，返回的是自增前的舊值；

原理不再贅述，參考上文compareAndSwapInt函式。

AtomicInteger使用示例：

public class AtomicExample {

    private static AtomicInteger atomicInteger = new AtomicInteger(2);

    public static void main(String[] args) {
        System.out.println(atomicInteger.getAndIncrement());
        System.out.println(atomicInteger.incrementAndGet());
        System.out.println(atomicInteger.get());
    }
}
// 2 4 4

LongAdder

為了解決自旋導致的效能問題，JDK8在Atomic包中推出了LongAdder類。LongAdder採用的方法是，共享熱點資料分離的計數：將一個數字的值拆分為一個數組。不同執行緒會命中到陣列的不同槽中，各個執行緒只對自己槽中的那個值進行CAS操作，這樣熱點就被分散了，衝突的概率就小很多；要得到這個數字的話，就要把這個值加起來。相比AtomicLong，併發量大大提高。

優點：有很高效能的併發寫的能力
缺點：讀取的效能不是很高效，而且如果讀取的時候出現併發寫的話，結果可能不是正確的

原子更新陣列型別

Atomic包中提供能原子更新陣列中元素的工具類：
AtomicIntegerArray：原子更新整型陣列中的元素；
AtomicLongArray：原子更新長整型陣列中的元素；
AtomicReferenceArray：原子更新引用型別陣列中的元素

這幾個類的用法一致，就以AtomicIntegerArray來總結下常用的方法：

1. addAndGet(int i, int delta)：以原子更新的方式將陣列中索引為i的元素與輸入值相加；
2. getAndIncrement(int i)：以原子更新的方式將陣列中索引為i的元素自增加1；
3. compareAndSet(int i, int expect, int update)：將陣列中索引為i的位置的元素進行更新

AtomicIntegerArray與AtomicInteger的方法基本一致，只不過在前者的方法中會多一個指定陣列索引位i。

AtomicIntegerArray使用示例：

public class AtomicExample {

    private static int[] value = new int[]{1, 2, 3};
    private static AtomicIntegerArray integerArray = new AtomicIntegerArray(value);

    public static void main(String[] args) {
        //對陣列中索引為2的位置的元素加3
        int result = integerArray.getAndAdd(2, 3);
        System.out.println(integerArray.get(2));
        System.out.println(result);
    }
}
// 6 3

原子更新引用型別

如果需要原子更新引用型別變數的話，為了保證執行緒安全，Atomic也提供了相關的類：

1. AtomicReference
2. AtomicReferenceFieldUpdater：原子更新引用型別裡的欄位；
3. AtomicMarkableReference：原子更新帶有標記位的引用型別；

AtomicReference使用示例：

public class AtomicExample {

    private static AtomicReference<User> reference = new AtomicReference<>();

    public static void main(String[] args) {
        User user1 = new User("a", 1);
        reference.set(user1);
        User user2 = new User("b",2);
        User user = reference.getAndSet(user2);
        System.out.println(user);
        System.out.println(reference.get());
    }

    static class User {
        private String userName;
        private int age;

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

        @Override
        public String toString() {
            return "User{" +
                    "userName='" + userName + '\'' +
                    ", age=" + age +
                    '}';
        }
    }
}
// User{userName='a', age=1}
// User{userName='b', age=2}

AtomicReferenceFieldUpdater使用示例：

public class AtomicExample {

    public static void main(String[] args) {
        AtomicReferenceFieldUpdater updater = AtomicReferenceFieldUpdater.newUpdater(Dog.class, String.class, "name");
        Dog dog1 = new Dog();
        updater.compareAndSet(dog1, dog1.name, "cat");
        System.out.println(dog1.name);
    }
}

class Dog {
    volatile String name = "dog1";
}

原子更新欄位型別

如果需要更新物件的某個欄位，Atomic同樣也提供了相應的原子操作類：

1. AtomicIntegeFieldUpdater：原子更新整型欄位類；
2. AtomicLongFieldUpdater：原子更新長整型欄位類；

要想使用原子更新欄位需要兩步操作：
原子更新欄位型別類都是抽象類，只能通過靜態方法newUpdater來建立一個更新器，並且需要設定想要更新的類和屬性；
更新類的屬性必須使用public volatile進行修飾；

AtomicIntegerFieldUpdater使用示例：

public class AtomicExample {

    private static AtomicIntegerFieldUpdater updater = AtomicIntegerFieldUpdater.newUpdater(User.class, "age");

    public static void main(String[] args) {
        User user = new User("a", 1);
        System.out.println(updater.getAndAdd(user, 5));
        System.out.println(updater.addAndGet(user, 1));
        System.out.println(updater.get(user));
    }

    static class User {
        private String userName;
        public volatile int age;

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

        @Override
        public String toString() {
            return "User{" +
                    "userName='" + userName + '\'' +
                    ", age=" + age +
                    '}';
        }
    }
}

解決CAS的ABA問題

AtomicStampedReference：原子更新引用型別，這種更新方式會帶有版本號，從而解決CAS的ABA問題

AtomicStampedReference使用示例：

public class AtomicExample {

    public static void main(String[] args) {
        Integer init1 = 1110;
//        Integer init2 = 126;
        AtomicStampedReference<Integer> reference = new AtomicStampedReference<>(init1, 1);
        int curent1 = reference.getReference();
//        Integer current2 = reference.getReference();
        reference.compareAndSet(reference.getReference(), reference.getReference() + 1, reference.getStamp(), reference.getStamp() + 1);//正確寫法
//        reference.compareAndSet(current2, current2+1, reference.getStamp(), reference.getStamp() + 1);//正確寫法
//        reference.compareAndSet(1110, 1111, reference.getStamp(), reference.getStamp() + 1);//錯誤寫法
//        reference.compareAndSet(curent1, curent1+1, reference.getStamp(), reference.getStamp() + 1);//錯誤寫法
//        reference.compareAndSet(current2, current2 + 1, reference.getStamp(), reference.getStamp() + 1);
        System.out.println("reference.getReference() = " + reference.getReference());
    }
}

AtomicStampedReference踩過的坑

參考上面的程式碼，分享一個筆者遇到的一次坑。AtomicStampedReference的compareAndSet函式中，前兩個引數是使用包裝類的。所以當引數超過128時，而且傳入引數並不是reference.getReference()獲取的話，會導致expectedReference == current.reference為false，則無法進行更新。

public boolean compareAndSet(V   expectedReference,
                                 V   newReference,
                                 int expectedStamp,
                                 int newStamp) {
        Pair<V> current = pair;
        return
            expectedReference == current.reference &&
            expectedStamp == current.stamp &&
            ((newReference == current.reference &&
              newStamp == current.stamp) ||
             casPair(current, Pair.of(newReference, newStamp)));
    }

最後，限於筆者經驗水平有限，歡迎讀者就文中的觀點提出寶貴的建議和意見。如果想獲得更多的學習資源或者想和更多的是技術愛好者一起交流，可以關注我的公眾號『全菜工程師小輝』後臺回覆關鍵詞領取學習資料、進入前後端技術交流群和程式設計師副業群。同時也可以加入程式設計師副業群Q群：735764906 一起交流。

相關推薦