【Android Kotlin协程】利用CoroutineContext实现网络请求失败后重试逻辑

最新推荐文章于 2023-12-27 22:13:58 发布

程序员小何SS

最新推荐文章于 2023-12-27 22:13:58 发布

阅读量685

点赞数

分类专栏： Android Kotlin 文章标签： android kotlin java

本文链接：https://blog.csdn.net/xiaokangss/article/details/125613893

版权

Android 同时被 2 个专栏收录

70 篇文章 3 订阅

订阅专栏

Kotlin

27 篇文章 0 订阅

订阅专栏

前言

在Android开发中有一个典型场景：网络请求失败后重试：一般的逻辑是弹出一个Dialog提醒用户“网络请求失败”，并提供重试的按钮。

如果当前页面只有一个网络请求，那么逻辑就很简单了：只需要再调用一下发起这个网络请求的方法就可以了。而当一个页面有多个网络请求时，我常用的办法为失败回调加状态，根据不同的状态调用不同的方法。但是这个方法不免有些繁琐，也有点不安全。首先，你要额外的增加状态，并将它传来传去。有些情况下，你甚至还需要重新初始化网络请求参数。更要命的是：你还要管理这个状态，一旦管理不善，就会导致调用了不该调用的方法，引入严重的BUG。

直到有一天我看到CoroutineExceptionHandler，灵光突现——可以使用协程上下文来保存将来可能需要重试的网络请求和Request数据，这样就能解决上面的问题了。

由于我所开发的大多数项目都是采用ViewModel实现网络请求逻辑和UI层的解耦，而网络请求基本上是采用Coroutine+Retrofit的方式实现的，基本上都是使用viewModelScope。

viewModelScope.launch() { 
	request()
}

viewModelScope本质上是一个ViewModel的扩展函数，利用它可以便捷的在ViewModel创建协程，具体的代码就不展开了。默认情况下，它的CoroutineContext由Job和CoroutineDispatcher组成。而协程的上下文本质上就是一个实现了key-value访问的方式的链表结构。我们可以通过继承AbstractCoroutineContextElement的方式实现自定义的CoroutineContext上下文：

class RetryCallback(val callback: () -> Unit) : AbstractCoroutineContextElement(RetryCallback) {
    companion object Key : CoroutineContext.Key<RetryCallback>
}

紧接着，当网络请求发生异常时借助CoroutineExceptionHandler获取到我们需要重新执行的操作：

val coroutineExceptionHandler = CoroutineExceptionHandler { coroutineContext, throwable ->
     val callback = coroutineContext[RetryCallback]
        ?.callback
}

紧接着，要将coroutineExceptionHandler添加到发起网络请求的协程上下文里：

viewModelScope.launch(exceptionHandler
      + RetryCallback { request() }) { 
	request()
}

此时，只要在发起网络请求的页面里获取到callback，并在点击重试按钮的时候调用它，就能实现重试的逻辑。

进一步对它进行封装并增加失败后自动重试逻辑，创建供ViewModel使用的接口，用来处理网络请求错误的后续逻辑：

interface ViewModelFailed {
    /**
     * @param throwable:异常信息
     * @param callback:需要重试的函数
     * */
    fun requestFailed(throwable: Throwable, callback: () -> Unit)
}

为它创建扩展函数，用来创建CoroutineExceptionHandler和RetryCallback上下文实例：

/**
 * @param autoReTry:是否自动重试
 * @param callback:需要重试的函数
 * */
fun ViewModelFailed.initRetry(autoReTry: Boolean = false, callback: () -> Unit) =
    CoroutineExceptionHandler { coroutineContext, throwable ->
        val retryCallBack = {
            coroutineContext[RetryCallback]
                ?.callback?.invoke()
        }
        if (autoReTry) {
            //自动开始重试逻辑
            onRetry()
            retryCallBack.invoke()
        } else {
            //不自动开始重试，后续操作交给用户决定
            requestFailed(throwable) {
                retryCallBack
            }
        }
    } + RetryCallback(callback)

ViewModel需要实现ViewModelFailed接口，并在发起网络请求的协程中调用initRetry方法添加异常处理上下文：

 class MainViewViewModel : ViewModel(), ViewModelFailed {
    val liveData: MutableLiveData<BaseData> = MutableLiveData()


      /**
     * @param num:用来演示Request请求数据
     * @param repeat:失败后自动重试的次数
     * */
    fun request(num: Int, repeat: Int = 0) {
        liveData.value = BaseData.loading()
        viewModelScope.launch(initRetry(repeat > 0) {
            request(num,repeat - 1)
        }) {
            liveData.value = BaseData.success(simulateHttp(num))
        }
    }

    private suspend fun simulateHttp(num: Int) = withContext(Dispatchers.IO) {
        //模拟网络请求
        ...
    }

    override fun requestFailed(throwable: Throwable, callback: () -> Unit) {
        //处理失败逻辑
        dialog()
        //重试
        callback.invoke()
    }
    
    override fun onRetry() {
        
    }

}