-
Notifications
You must be signed in to change notification settings - Fork 22
Using many coroutines to pass on the work
Devrath edited this page Jun 15, 2021
·
1 revision
class CalculationInSeveralCoroutinesViewModel(
private val factorialCalculator: FactorialCalculator = FactorialCalculator(),
private val defaultDispatcher: CoroutineDispatcher = Dispatchers.Default
) : BaseViewModel<UiState>() {
fun performCalculation(
factorialOf: Int,
numberOfCoroutines: Int
) {
uiState.value = UiState.Loading
viewModelScope.launch {
var factorialResult = BigInteger.ZERO
val computationDuration = measureTimeMillis {
factorialResult =
factorialCalculator.calculateFactorial(
factorialOf,
numberOfCoroutines
)
}
var resultString = ""
val stringConversionDuration = measureTimeMillis {
resultString = convertToString(factorialResult)
}
uiState.value =
UiState.Success(resultString, computationDuration, stringConversionDuration)
}
}
private suspend fun convertToString(
number: BigInteger
): String =
withContext(defaultDispatcher) {
number.toString()
}
}
class FactorialCalculator(
private val defaultDispatcher: CoroutineDispatcher = Dispatchers.Default
) {
suspend fun calculateFactorial(
factorialOf: Int,
numberOfCoroutines: Int
): BigInteger {
return withContext(defaultDispatcher) {
val subRanges = createSubRangeList(factorialOf, numberOfCoroutines)
subRanges.map { subRange ->
async {
calculateFactorialOfSubRange(subRange)
}
}.awaitAll()
.fold(BigInteger.ONE, { acc, element ->
ensureActive()
acc.multiply(element)
})
}
}
suspend fun calculateFactorialOfSubRange(
subRange: SubRange
): BigInteger {
return withContext(defaultDispatcher) {
Timber.d(addCoroutineDebugInfo("Calculate factorial of $subRange"))
var factorial = BigInteger.ONE
for (i in subRange.start..subRange.end) {
ensureActive()
factorial = factorial.multiply(BigInteger.valueOf(i.toLong()))
}
factorial
}
}
fun createSubRangeList(
factorialOf: Int,
numberOfSubRanges: Int
): List<SubRange> {
val quotient = factorialOf.div(numberOfSubRanges)
val rangesList = mutableListOf<SubRange>()
var curStartIndex = 1
repeat(numberOfSubRanges - 1) {
rangesList.add(
SubRange(
curStartIndex,
curStartIndex + (quotient - 1)
)
)
curStartIndex += quotient
}
rangesList.add(SubRange(curStartIndex, factorialOf))
return rangesList
}
}
data class SubRange(val start: Int, val end: Int)