[Native] Move isRestrictedSuspendFunction from IrUtils2

This function is also helpful for JVM backend.
This commit is contained in:
Ivan Kylchik
2023-08-16 11:17:03 +02:00
committed by Space Team
parent 2657adc6dc
commit 5cd96831dc
4 changed files with 11 additions and 9 deletions
@@ -9,6 +9,7 @@ import org.jetbrains.kotlin.descriptors.Modality
import org.jetbrains.kotlin.descriptors.DescriptorVisibilities
import org.jetbrains.kotlin.ir.IrElement
import org.jetbrains.kotlin.ir.IrStatement
import org.jetbrains.kotlin.ir.ObsoleteDescriptorBasedAPI
import org.jetbrains.kotlin.ir.builders.*
import org.jetbrains.kotlin.ir.builders.irCall
import org.jetbrains.kotlin.ir.declarations.*
@@ -26,6 +27,8 @@ import org.jetbrains.kotlin.ir.types.makeNotNull
import org.jetbrains.kotlin.ir.util.*
import org.jetbrains.kotlin.ir.visitors.*
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.resolve.calls.checkers.isRestrictedSuspendFunction
import org.jetbrains.kotlin.resolve.calls.checkers.isRestrictsSuspensionReceiver
internal class NativeSuspendFunctionsLowering(
generationState: NativeGenerationState
@@ -35,12 +38,13 @@ internal class NativeSuspendFunctionsLowering(
override val stateMachineMethodName = Name.identifier("invokeSuspend")
@OptIn(ObsoleteDescriptorBasedAPI::class)
override fun getCoroutineBaseClass(function: IrFunction): IrClassSymbol =
(if (function.isRestrictedSuspendFunction()) {
if (function.descriptor.isRestrictedSuspendFunction()) {
symbols.restrictedContinuationImpl
} else {
symbols.continuationImpl
})
}
override fun nameForCoroutineClass(function: IrFunction) =
fileLowerState.getCoroutineImplUniqueName(function).synthesizedName
@@ -75,10 +75,6 @@ fun IrBuilderWithScope.irCatch() =
)
)
@OptIn(ObsoleteDescriptorBasedAPI::class)
fun IrFunction.isRestrictedSuspendFunction(): Boolean =
this.descriptor.extensionReceiverParameter?.type?.isRestrictsSuspensionReceiver() == true
fun IrBuilderWithScope.irByte(value: Byte) =
IrConstImpl.byte(startOffset, endOffset, context.irBuiltIns.byteType, value)