/
CooperativeCancellationViewModel.kt
73 lines (59 loc) 路 2.38 KB
/
CooperativeCancellationViewModel.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package com.lukaslechner.coroutineusecasesonandroid.usecases.coroutines.usecase11
import androidx.lifecycle.LiveData
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.ViewModel
import androidx.lifecycle.viewModelScope
import kotlinx.coroutines.*
import java.math.BigInteger
import kotlin.system.measureTimeMillis
class CooperativeCancellationViewModel(
private val defaultDispatcher: CoroutineDispatcher = Dispatchers.Default
) : ViewModel() {
private var calculationJob: Job? = null
fun performCalculation(factorialOf: Int) {
uiState.value = UiState.Loading
calculationJob = viewModelScope.launch {
try {
var result: BigInteger = BigInteger.ZERO
val computationDuration = measureTimeMillis {
result = calculateFactorialOf(factorialOf)
}
var resultString = ""
val stringConversionDuration = measureTimeMillis {
resultString = convertToString(result)
}
uiState.value =
UiState.Success(resultString, computationDuration, stringConversionDuration)
} catch (exception: Exception) {
uiState.value = if (exception is CancellationException) {
UiState.Error("Calculation was cancelled")
} else {
UiState.Error("Error while calculating result")
}
}
}
}
// factorial of n (n!) = 1 * 2 * 3 * 4 * ... * n
private suspend fun calculateFactorialOf(number: Int): BigInteger =
withContext(defaultDispatcher) {
var factorial = BigInteger.ONE
for (i in 1..number) {
// yield enables cooperative cancellations
// alternatives:
// - ensureActive()
// - isActive() - possible to do clean up tasks with
yield()
factorial = factorial.multiply(BigInteger.valueOf(i.toLong()))
}
factorial
}
private suspend fun convertToString(number: BigInteger): String =
withContext(defaultDispatcher) {
number.toString()
}
fun uiState(): LiveData<UiState> = uiState
fun cancelCalculation() {
calculationJob?.cancel()
}
private val uiState: MutableLiveData<UiState> = MutableLiveData()
}