增加原子类更新字段的方法

byhieg · byhieg · commit 4d5f713eb52b · 2017-05-07T10:28:38.000+08:00
diff --git a/src/main/java/cn/byhieg/threadtutorial/concurrent/atom/AtomFactory.java b/src/main/java/cn/byhieg/threadtutorial/concurrent/atom/AtomFactory.java
@@ -1,9 +1,6 @@
 package cn.byhieg.threadtutorial.concurrent.atom;
 
-import java.util.concurrent.atomic.AtomicInteger;
-import java.util.concurrent.atomic.AtomicIntegerArray;
-import java.util.concurrent.atomic.AtomicLong;
-import java.util.concurrent.atomic.AtomicReference;
+import java.util.concurrent.atomic.*;
 
 /**
  * Created by shiqifeng on 2017/5/5.
@@ -30,7 +27,10 @@ public AtomicIntegerArray createAtomArray(int[] a) {
     }
 
     public AtomicReference<MyObject> createAtomReference(MyObject object){
-        AtomicReference<MyObject> objectAtomicReference = new AtomicReference<>();
-        return objectAtomicReference;
+        return new AtomicReference<>();
+    }
+
+    public AtomicIntegerFieldUpdater<MyObject> createAtomIntegerUpdate(String fieldName) {
+        return  AtomicIntegerFieldUpdater.newUpdater(MyObject.class, fieldName);
     }
 }
diff --git a/src/test/java/cn/byhieg/threadtutorialtest/concurrenttest/atomtest/AtomFactoryTest.java b/src/test/java/cn/byhieg/threadtutorialtest/concurrenttest/atomtest/AtomFactoryTest.java
@@ -7,6 +7,7 @@
 
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicIntegerArray;
+import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;
 import java.util.concurrent.atomic.AtomicReference;
 
 /**
@@ -17,13 +18,15 @@ public class AtomFactoryTest extends TestCase {
     AtomicInteger integer;
     AtomicIntegerArray array;
     AtomicReference<MyObject> reference;
+    AtomicIntegerFieldUpdater<MyObject> updater;
 
     public void setUp() throws Exception {
         super.setUp();
         integer = AtomFactory.getInstance().createAtomInt(1);
     }
 
     public void testAtomInt() throws Exception {
+        System.out.println("int原子类");
         new Thread(()->{
             for (int i = 0; i < 10; i++) {
                 integer.getAndIncrement();
@@ -42,6 +45,7 @@ public void testAtomInt() throws Exception {
     }
 
     public void testAtomArray() throws Exception {
+        System.out.println("原子类数组");
         int [] value = new int[]{1,2,3,4};
         array = AtomFactory.getInstance().createAtomArray(value);
         array.getAndSet(1,10);
@@ -50,14 +54,23 @@ public void testAtomArray() throws Exception {
     }
 
     public void testAtomRef()throws Exception {
+        System.out.println("原子类");
         MyObject object = new MyObject();
         reference = AtomFactory.getInstance().createAtomReference(object);
         reference.set(object);
         MyObject newObject = new MyObject();
         newObject.name = "xiaoli";
         reference.compareAndSet(object, newObject);
         System.out.println(reference.get().name);
+    }
+
 
+    public void testUpdater() throws Exception {
+        System.out.println("原子类更新字段");
+        updater = AtomFactory.getInstance().createAtomIntegerUpdate("id");
+        MyObject object = new MyObject();
+        System.out.println(updater.getAndIncrement(object));
+        System.out.println(updater.get(object));
     }
 
 
