UseCase.kt

UseCase.kt

inline fun <T> ViewModel.useCase(
    noinline call: suspend () -> Result<T>,
    noinline onError: (suspend (exception: Throwable) -> Unit)? = null,
    noinline onSuccess: (suspend (data: T) -> Unit)? = null
) {
    flow {
        when (val response = call()) {
            is Result.Success -> emit(response.data)
            is Result.Error -> throw response.exception
        }
    }.onEach {
        onSuccess?.invoke(it)
    }.catch { throwable ->
        onError?.invoke(throwable)
    }.launchIn(viewModelScope)
}