flatMapLatest

kt

fun <T, R> Flow<T>.flatMapLatest(transform: suspend (T) -> Flow<R>): Flow<R> = flow {
    // This coroutineScope is key to managing cancellation of the inner flow
    coroutineScope {
        var previousJob: Job? = null

        collect { value ->
            // Cancel any ongoing inner flow collection
            previousJob?.cancelAndJoin()

            // Launch new inner flow collector in a new coroutine
            previousJob = launch {
                transform(value).collect { innerValue ->
                    emit(innerValue)  // Emit downstream
                }
            }
        }
    }
}