JS: fix noInline version of inline suspend fun (KT-27611 fixed)

2018-10-19 14:56:55 +03:00
parent 455db32199
commit 829bdf6061
11 changed files with 145 additions and 66 deletions
@@ -21,14 +21,8 @@ import org.jetbrains.kotlin.descriptors.Modality
 import org.jetbrains.kotlin.descriptors.PropertyDescriptor
 import org.jetbrains.kotlin.descriptors.isOverridable
 import org.jetbrains.kotlin.js.backend.ast.JsExpression
-import org.jetbrains.kotlin.js.backend.ast.JsFunction
 import org.jetbrains.kotlin.js.backend.ast.JsName
-import org.jetbrains.kotlin.js.backend.ast.metadata.coroutineMetadata
-import org.jetbrains.kotlin.js.backend.ast.metadata.isInlineableCoroutineBody
-import org.jetbrains.kotlin.js.descriptorUtils.shouldBeExported
-import org.jetbrains.kotlin.js.translate.callTranslator.CallTranslator
 import org.jetbrains.kotlin.js.translate.context.TranslationContext
-import org.jetbrains.kotlin.js.translate.expression.InlineMetadata
 import org.jetbrains.kotlin.js.translate.expression.translateAndAliasParameters
 import org.jetbrains.kotlin.js.translate.expression.translateFunction
 import org.jetbrains.kotlin.js.translate.expression.wrapWithInlineMetadata
@@ -36,7 +30,6 @@ import org.jetbrains.kotlin.js.translate.general.TranslatorVisitor
 import org.jetbrains.kotlin.js.translate.utils.*
 import org.jetbrains.kotlin.psi.*
 import org.jetbrains.kotlin.resolve.descriptorUtil.isExtensionProperty
-import org.jetbrains.kotlin.resolve.source.getPsi

 abstract class AbstractDeclarationVisitor : TranslatorVisitor<Unit>()  {
    override fun emptyResult(context: TranslationContext) { }
@@ -102,20 +95,7 @@ abstract class AbstractDeclarationVisitor : TranslatorVisitor<Unit>()  {
            null
        }

-        if (descriptor.isSuspend && descriptor.isInline && descriptor.shouldBeExported(context.config) && functionAndContext != null) {
-            val exportFunction = functionAndContext.first as JsFunction
-            exportFunction.fillCoroutineMetadata(context, descriptor, exportFunction.coroutineMetadata!!.hasController)
-            addFunction(descriptor, exportFunction, expression)
-
-            val innerContext = functionAndContext.second
-            val inlineFunction = functionAndContext.first.deepCopy() as JsFunction
-            inlineFunction.name = null
-            inlineFunction.coroutineMetadata = null
-            inlineFunction.isInlineableCoroutineBody = true
-            context.addDeclarationStatement(innerContext.wrapWithInlineMetadata(context, inlineFunction, descriptor).makeStmt())
-        } else {
-            addFunction(descriptor, functionAndContext?.first, expression)
-        }
+        addFunction(descriptor, functionAndContext?.first, expression)
    }

    override fun visitTypeAlias(typeAlias: KtTypeAlias, data: TranslationContext?) {}
@@ -137,14 +117,8 @@ abstract class AbstractDeclarationVisitor : TranslatorVisitor<Unit>()  {
            function.body.statements += FunctionBodyTranslator.setDefaultValueForArguments(descriptor, innerContext)
        }
        innerContext.translateFunction(expression, function)
-        val result = if (descriptor.isSuspend && descriptor.shouldBeExported(context.config)) {
-            function
-        }
-        else {
-            innerContext.wrapWithInlineMetadata(context, function, descriptor)
-        }

-        return Pair(result, innerContext)
+        return Pair(innerContext.wrapWithInlineMetadata(context, function, descriptor), innerContext)
    }

    // used from kotlinx.serialization
@@ -21,10 +21,7 @@ import com.intellij.psi.PsiElement
 import org.jetbrains.kotlin.descriptors.*
 import org.jetbrains.kotlin.incremental.js.IncrementalResultsConsumer
 import org.jetbrains.kotlin.js.backend.ast.*
-import org.jetbrains.kotlin.js.backend.ast.metadata.descriptor
-import org.jetbrains.kotlin.js.backend.ast.metadata.functionDescriptor
-import org.jetbrains.kotlin.js.backend.ast.metadata.hasDefaultValue
-import org.jetbrains.kotlin.js.backend.ast.metadata.type
+import org.jetbrains.kotlin.js.backend.ast.metadata.*
 import org.jetbrains.kotlin.js.config.JSConfigurationKeys
 import org.jetbrains.kotlin.js.descriptorUtils.shouldBeExported
 import org.jetbrains.kotlin.js.inline.util.FunctionWithWrapper
@@ -0,0 +1,61 @@
+// IGNORE_BACKEND: JS_IR
+// EXPECTED_REACHABLE_NODES: 1280
+
+// MODULE: lib
+// FILE: lib.kt
+import kotlin.coroutines.*
+
+var resume: () -> Unit = {}
+
+suspend fun suspendAndReturn(msg: String): String {
+    return suspendCoroutine<String> { cont ->
+        resume = {
+            cont.resume(msg)
+        }
+    }
+}
+
+suspend fun <T> runSus(block: suspend () -> T): T = block()
+
+abstract class A {
+    fun f() = o()
+
+    abstract fun o(): String
+}
+
+// MODULE: libInline(lib)
+// FILE: libInline.kt
+
+suspend inline fun foo(): String {
+    val a = object : A() {
+        override fun o(): String = "O"
+    }
+    val k = "K"
+
+    return suspendAndReturn(a.f()) + runSus {
+        val b = object : A() {
+            override fun o(): String = k
+        }
+        runSus { suspendAndReturn(b.f()) }
+    }
+}
+
+// MODULE: main(libInline, lib)
+// FILE: main.kt
+
+import kotlin.coroutines.*
+
+fun box(): String {
+    var testResult: String = "fail"
+
+    val continuation = Continuation<String>(EmptyCoroutineContext) { result ->
+        testResult = result.getOrThrow()
+    }
+
+    // Test the noinline version of public inline suspend functions
+    js("libInline").foo(continuation)
+    resume()
+    resume()
+
+    return testResult
+}
@@ -14,4 +14,4 @@ suspend fun bar(): Unit {
    println(a + b)
 }

-// LINES: 39 39 4 4 4 5 5 44 44 5 88 44 5 5 6 4 4 4 15 9 9 9 9 15 9 9 9 * 15 10 10 11 11 11 2 11 11 * 11 12 12 13 13 13 2 13 13 13 13 14 14
+// LINES: 39 39 4 4 4 5 5 44 44 5 88 44 5 5 6 4 4 4 15 9 9 9 * 15 10 10 11 11 11 2 11 11 * 11 12 12 13 13 13 2 13 13 13 13 14 14 * 15 9 9 9 9
@@ -8,4 +8,4 @@ suspend fun foo() {
 suspend fun delay() {
 }

-// LINES: 6 1 1 1 1 6 1 1 * 6 2 2 2 2 2 * 3 3 4 4 4 2 4 4 5 5 * 9
+// LINES: 6 1 1 * 6 2 2 2 2 2 * 3 3 4 4 4 2 4 4 5 5 * 6 1 1 1 1 9