onSuccess.kt

fun <T> Resource<T>.onSuccess(
    onSuccess: (data: T) -> Unit
): Resource<T> {
    if (this is Resource.Success<T>)
        onSuccess(data)

    return this
}

fun <T> Resource<T>.onLoading(
    onLoading: (partialData: T?) -> Unit
): Resource<T> {
    if (this is Resource.Loading)
        onLoading(partialData)

    return this
}

fun <T> Resource<T>.onFailure(
    onFailure: (throwable: Throwable?) -> Unit
): Resource<T> {
    if (this is Resource.Failure)
        onFailure(throwable)

    return this
}