NotNegativeLong

nytta · Nov 20, 2018 · 7071be1 · 7071be1
1 parent e298683
commit 7071be1
Showing 1 changed file with 30 additions and 13 deletions.
diff --git a/cobia-core/src/main/java/lam/cobia/core/NotNegativeLong.java b/cobia-core/src/main/java/lam/cobia/core/NotNegativeLong.java
@@ -20,37 +20,54 @@ public NotNegativeLong() {
     }
 
     public NotNegativeLong(long initialValue) {
-        checkValue(initialValue, 0);
-        value = new AtomicLong(initialValue);
-    }
-
-    private void checkValue(long value, long leastValue) {
-        if (value < leastValue) {
-            throw new IllegalStateException(value + " < " + leastValue);
+        if (initialValue < 0) {
+            throw new IllegalArgumentException("initialValue(" + initialValue + ") < 0");
         }
+        value = new AtomicLong(initialValue);
     }
 
     public long getAndIncrement() {
-        return value.getAndIncrement();
+        long oldValue = value.get();
+        long newValue = oldValue == Long.MAX_VALUE ? 0 : oldValue + 1;
+        while (!value.compareAndSet(oldValue, newValue)) {
+            oldValue = value.get();
+            newValue = oldValue == Long.MAX_VALUE ? 0 : oldValue + 1;
+        }
+        return oldValue;
     }
 
     public long incrementAndGet() {
-        return value.incrementAndGet();
+        long oldValue = value.get();
+        long newValue = oldValue == Long.MAX_VALUE ? 0 : oldValue + 1;
+        while (!value.compareAndSet(oldValue, newValue)) {
+            oldValue = value.get();
+            newValue = oldValue == Long.MAX_VALUE ? 0 : oldValue + 1;
+        }
+        return newValue;
     }
 
     public long getAndDecrement() {
-        long oldValue = value.getAndDecrement();
-        checkValue(oldValue, 1);
+        long oldValue = value.get();
+        long newValue = oldValue == 0 ? 0 : oldValue - 1;
+        while (!value.compareAndSet(oldValue, newValue)) {
+            oldValue = value.get();
+            newValue = oldValue == 0 ? 0 : oldValue - 1;
+        }
         return oldValue;
     }
 
     public long decrementAndGet() {
-        long newValue = value.decrementAndGet();
-        checkValue(newValue, 0);
+        long oldValue = value.get();
+        long newValue = oldValue == 0 ? 0 : oldValue - 1;
+        while (!value.compareAndSet(oldValue, newValue)) {
+            oldValue = value.get();
+            newValue = oldValue == 0 ? 0 : oldValue - 1;
+        }
         return newValue;
     }
 
     public long get() {
         return value.get();
     }
+
 }