Files
kotlin/compiler/testData/codegen/bytecodeListing/tcoContinuation.kt
2019-05-14 16:00:39 +03:00

41 lines
1.1 KiB
Kotlin
Vendored

// COMMON_COROUTINES_TEST
// WITH_RUNTIME
// WITH_COROUTINES
// There should be no $foo$$inlined$map$1$1 class
interface FlowCollector<T> {
suspend fun emit(value: T)
}
interface Flow<T> {
suspend fun collect(collector: FlowCollector<T>)
}
public inline fun <T> flow(crossinline block: suspend FlowCollector<T>.() -> Unit) = object : Flow<T> {
override suspend fun collect(collector: FlowCollector<T>) = collector.block()
}
suspend inline fun <T> Flow<T>.collect(crossinline action: suspend (T) -> Unit): Unit =
collect(object : FlowCollector<T> {
override suspend fun emit(value: T) = action(value)
})
public inline fun <T, R> Flow<T>.transform(crossinline transformer: suspend FlowCollector<R>.(value: T) -> Unit): Flow<R> {
return flow {
return@flow collect { value ->
return@collect transformer(value)
}
}
}
public inline fun <T, R> Flow<T>.map(crossinline transformer: suspend (value: T) -> R): Flow<R> = transform { value -> return@transform emit(transformer(value)) }
suspend fun foo() {
flow<Int> {
emit(1)
}.map { it + 1 }
.collect {
}
}