Support repeated 'invoke' calls on coroutines

#KT-12782 In Progress
2016-06-14 12:55:13 +03:00
parent a25602c709
commit c2ac49ec96
4 changed files with 149 additions and 7 deletions
@@ -0,0 +1,99 @@
+class Controller {
+    var lastSuspension: Continuation<String>? = null
+    var result = "fail"
+    suspend fun suspendHere(x: Continuation<String>) {
+        lastSuspension = x
+    }
+
+    fun hasNext() = lastSuspension != null
+    fun next() {
+        val x = lastSuspension!!
+        lastSuspension = null
+        x.resume("56")
+    }
+}
+
+fun builder(coroutine c: Controller.() -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1).resume(Unit)
+    c(controller2).resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+fun builder2(coroutine c: Controller.(Long, String) -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1, 1234567890123456789L, "Q").resume(Unit)
+    c(controller2, 1234567890123456789L, "Q").resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+
+private fun runControllers(controller1: Controller, controller2: Controller) {
+    while (controller1.hasNext()) {
+        if (!controller2.hasNext()) throw RuntimeException("fail 1")
+
+        if (controller1.lastSuspension === controller2.lastSuspension) throw RuntimeException("equal references")
+
+        controller1.next()
+        controller2.next()
+    }
+
+    if (controller2.hasNext()) throw RuntimeException("fail 2")
+
+    if (controller1.result != "OK") throw RuntimeException("fail 3")
+    if (controller2.result != "OK") throw RuntimeException("fail 4")
+}
+
+fun box(): String {
+    // no suspension
+    builder {
+        result = "OK"
+    }
+
+    // 1 suspension
+    builder {
+        if (suspendHere() != "56") return@builder
+        result = "OK"
+    }
+
+    // 2 suspensions
+    builder {
+        if (suspendHere() != "56") return@builder
+        suspendHere()
+        result = "OK"
+    }
+
+    // with capture and params
+
+    var x = "O"
+    var y = "K"
+
+    // no suspension
+    builder2 { a, b ->
+        if (a != 1234567890123456789L || b != "Q" ) return@builder2
+        result = x + y
+    }
+
+    // 1 suspension
+    builder2 { a, b ->
+        if (a != 1234567890123456789L || b != "Q" ) return@builder2
+        if (suspendHere() != "56") return@builder2
+        result = x + y
+    }
+
+    // 2 suspensions
+    builder2 { a, b ->
+        if (a != 1234567890123456789L || b != "Q" ) return@builder2
+        if (suspendHere() != "56") return@builder2
+        suspendHere()
+        result = x + y
+    }
+
+    return "OK"
+}