commonMain.dev.inmo.tgbotapi.extensions.utils.FlowsAggregation.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tgbotapi.utils Show documentation

Additional extensions for core part of tgbotapi

There is a newer version: 20.0.1

package dev.inmo.tgbotapi.extensions.utils

import dev.inmo.micro_utils.coroutines.safely
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.flow.*

/**
 * Analog of [merge] function for [Flow]s. The difference is in the usage of [MutableSharedFlow] in this case
 */
fun  aggregateFlows(
    withScope: CoroutineScope,
    vararg flows: Flow,
    internalBufferSize: Int = 64
): Flow {
    val sharedFlow = MutableSharedFlow(extraBufferCapacity = internalBufferSize)
    flows.forEach {
        it.onEach {
            safely { sharedFlow.emit(it) }
        }.launchIn(withScope)
    }
    return sharedFlow
}

fun  Flow>.flatten(): Flow = flow {
    collect {
        it.forEach {
            emit(it)
        }
    }
}

fun  Flow.flatMap(mapper: suspend (T) -> Iterable): Flow = flow {
    collect {
        mapper(it).forEach {
            emit(it)
        }
    }
}