kotlinx.atomicfu JVM/IR compiler plugin support

Merge-request: KT-MR-6541 Merged-by: Maria Sokolova <maria.sokolova@jetbrains.com>
2022-06-26 07:17:06 +00:00
parent eaad75fd52
commit 8053746ae0
62 changed files with 4249 additions and 349 deletions
@@ -0,0 +1,51 @@
+import kotlinx.atomicfu.*
+import kotlin.test.*
+
+class LoopTest {
+    val a = atomic(10)
+    val b = atomic(11)
+    val c = atomic(12)
+    val r = atomic<String>("aaa")
+    val intArr = AtomicIntArray(10)
+
+    private inline fun AtomicInt.extensionEmbeddedLoops(to: Int): Int =
+        loop { cur1 ->
+            compareAndSet(cur1, to)
+            loop { cur2 ->
+                return cur2
+            }
+        }
+
+    private inline fun embeddedLoops(to: Int): Int =
+        a.loop { aValue ->
+            b.loop { bValue ->
+                if (b.compareAndSet(bValue, to)) return aValue + bValue
+            }
+        }
+
+    private inline fun embeddedUpdate(to: Int): Int =
+        a.loop { aValue ->
+            a.compareAndSet(aValue, to)
+            return a.updateAndGet { cur -> cur + 100 }
+        }
+
+    private inline fun AtomicRef<String>.extesntionEmbeddedRefUpdate(to: String): String =
+        loop { value ->
+            compareAndSet(value, to)
+            return updateAndGet { cur -> "${cur}AAA" }
+        }
+
+    fun test() {
+        assertEquals(21, embeddedLoops(12))
+        assertEquals(77, c.extensionEmbeddedLoops(77))
+        assertEquals(66, intArr[0].extensionEmbeddedLoops(66))
+        assertEquals(166, embeddedUpdate(66))
+        assertEquals("bbbAAA", r.extesntionEmbeddedRefUpdate("bbb"))
+    }
+}
+
+fun box(): String {
+    val testClass = LoopTest()
+    testClass.test()
+    return "OK"
+}