Support crossinline suspend lambda as parameter of inline function

Use fake continuation instead of ALOAD 0 while inlining Do not generate state machine for inner lambdas and inner objects, which capture crossinline suspend lambda. #KT-19159: Fixed
2018-01-18 15:21:19 +03:00
parent 042ca55be7
commit 6854135077
80 changed files with 5214 additions and 144 deletions
@@ -0,0 +1,84 @@
+// FILE: inlined.kt
+// WITH_RUNTIME
+// NO_CHECK_LAMBDA_INLINING
+
+suspend inline fun inlineMe(c: suspend (String) -> String, d: suspend () -> String): String {
+    return c(try { d() } catch (e: Exception) { "Exception 1 ${e.message}" })
+}
+
+suspend inline fun noinlineMe(noinline c: suspend (String) -> String, noinline d: suspend () -> String): String {
+    return c(try { d() } catch (e: Exception) { "Exception 1 ${e.message}" })
+}
+
+suspend inline fun crossinlineMe(crossinline c: suspend (String) -> String, crossinline d: suspend () -> String): String {
+    return c(try { d() } catch (e: Exception) { "Exception 1 ${e.message}" })
+}
+
+// FILE: inlineSite.kt
+
+import kotlin.coroutines.experimental.*
+
+fun builder(c: suspend () -> Unit) {
+    c.startCoroutine(object: Continuation<Unit> {
+        override val context: CoroutineContext
+            get() = EmptyCoroutineContext
+
+        override fun resume(value: Unit) {
+        }
+
+        override fun resumeWithException(exception: Throwable) {
+            throw exception
+        }
+    })
+}
+
+suspend fun yieldString(s: String) = s
+
+suspend fun throwException(s: String): String {
+    throw RuntimeException(s)
+}
+
+suspend fun inlineSite() {
+    var res = inlineMe({ it }) { yieldString("OK") }
+    if (res != "OK") throw RuntimeException("FAIL 1: $res")
+
+    res = inlineMe({ it }) { throwException("OK") }
+    if (res != "Exception 1 OK") throw RuntimeException("FAIL 2: $res")
+
+    res = noinlineMe({ it }) { yieldString("OK") }
+    if (res != "OK") throw RuntimeException("FAIL 3: $res")
+
+    res = noinlineMe({ it }) { throwException("OK") }
+    if (res != "Exception 1 OK") throw RuntimeException("FAIL 4: $res")
+
+    res = crossinlineMe({ it }) { yieldString("OK") }
+    if (res != "OK") throw RuntimeException("FAIL 5: $res")
+
+    res = crossinlineMe({ it }) { throwException("OK") }
+    if (res !=  "Exception 1 OK") throw RuntimeException("FAIL 6: $res")
+
+    res = inlineMe({ try {yieldString(it) } catch (e: Exception) { yieldString("Exception 2 $it") } }) { yieldString("OK") }
+    if (res != "OK") throw RuntimeException("FAIL 7: $res")
+
+    res = inlineMe({ try {throwException(it) } catch (e: Exception) { yieldString("Exception 2 $it") } }) { yieldString("OK") }
+    if (res != "Exception 2 OK") throw RuntimeException("FAIL 8: $res")
+
+    res = noinlineMe({ try {yieldString(it) } catch (e: Exception) { yieldString("Exception 2 $it") } }) { yieldString("OK") }
+    if (res != "OK") throw RuntimeException("FAIL 9: $res")
+
+    res = noinlineMe({ try {throwException(it) } catch (e: Exception) { yieldString("Exception 2 $it") } }) { yieldString("OK") }
+    if (res != "Exception 2 OK") throw RuntimeException("FAIL 10: $res")
+
+    res = crossinlineMe({ try {yieldString(it) } catch (e: Exception) { yieldString("Exception 2 $it") } }) { yieldString("OK") }
+    if (res != "OK") throw RuntimeException("FAIL 11: $res")
+
+    res = crossinlineMe({ try {throwException(it) } catch (e: Exception) { yieldString("Exception 2 $it") } }) { yieldString("OK") }
+    if (res !=  "Exception 2 OK") throw RuntimeException("FAIL 12: $res")
+}
+
+fun box(): String {
+    builder {
+        inlineSite()
+    }
+    return "OK"
+}