你好,我是朱涛。今天这节实战课,我们接着前面第12讲里实现的网络请求框架,来进一步完善这个KtHttp,让它支持挂起函数。

在上一次实战课当中,我们已经开发出了两个版本的KtHttp,1.0版本的是基于命令式风格的,2.0版本的是基于函数式风格的。其中2.0版本的代码风格,跟我们平时工作写的代码风格很不一样,之前我也说了,这主要是因为业界对Kotlin函数式编程接纳度并不高,所以这节课的代码,我们将基于1.0版本的代码继续改造。这样,也能让课程的内容更接地气一些,甚至你都可以借鉴今天写代码的思路,复用到实际的Android或者后端开发中去。

跟往常一样,这节课的代码还是会分为两个版本:

好,接下来就正式开始吧!

3.0 版本:支持异步(Call)

有了上一次实战课的基础,这节课就会轻松一些了。关于动态代理、注解、反射之类的知识不会牵涉太多,我们今天主要把精力都集中在协程上来。不过,在正式开始写协程代码之前,我们需要先让KtHttp支持异步请求,也就是Callback请求。

这是为什么呢?别忘了第15讲的内容:挂起函数本质就是Callback!所以,为了让KtHttp支持挂起函数,我们可以采用迂回的策略,让它先支持Callback。在之前1.0、2.0版本的代码中,KtHttp是只支持同步请求的,你可能对异步同步还有些懵,我带你来看个例子吧。

首先,这个是同步代码:

fun main() {
    // 同步代码
    val api: ApiService = KtHttpV1.create(ApiService::class.java)
    val data: RepoList = api.repos(lang = "Kotlin", since = "weekly")
    println(data)
}

可以看到,在main函数当中,我们调用了KtHttp 1.0的代码,其中3行代码的运行顺序是1、2、3,这就是典型的同步代码。它的另一个特点就是:所有代码都会在一个线程中执行,因此这样的代码如果运行在Android、Swing之类的UI编程平台上,是会导致主线程卡死的。

那么,异步代码又是长什么样的呢?

private fun testAsync() {
    // 异步代码
    KtHttpV3.create(ApiServiceV3::class.java).repos(
        lang = "Kotlin",
        since = "weekly"
    ).call(object : Callback<RepoList> {
        override fun onSuccess(data: RepoList) {
            println(data)
        }

        override fun onFail(throwable: Throwable) {
            println(throwable)
        }
    })
}

上面的testAsync()方法当中的代码,就是典型的异步代码,它跟同步代码最大的差异就是,有了一个Callback,而且代码不再是按照顺序执行的了。你可以参考下面这个动图:

图片

所以,在3.0版本的开发中,我们就是要实现类似上面testAsync()的请求方式。为此,我们首先需要创建一个Callback接口,在这个Callback当中,我们可以拿到API请求的结果。

interface Callback<T: Any> {
    fun onSuccess(data: T)
    fun onFail(throwable: Throwable)
}

在Callback这个接口里,有一个泛型参数T,还有两个回调,分别是onSuccess代表接口请求成功、onFail代表接口请求失败。需要特别注意的是,这里我们运用了空安全思维当中的泛型边界“T: Any”,这样一来,我们就可以保证T类型一定是非空的。

除此之外,我们还需要一个KtCall类,它的作用是承载Callback,或者说,它是用来调用Callback的。

class KtCall<T: Any>(
    private val call: Call,
    private val gson: Gson,
    private val type: Type
) {
    fun call(callback: Callback<T>): Call {
        // TODO
    }
}

KtCall这个类仍然使用了泛型边界“T: Any”,另外,它还有几个关键的成员分别是:OkHttp的Call对象、JSON解析的Gson对象,以及反射类型Type。然后还有一个call()方法,它接收的是前面我们定义的Callback对象,返回的是OkHttp的Call对象。所以总的来说,call()方法当中的逻辑会分为三个步骤。

class KtCall<T: Any>(
    private val call: Call,
    private val gson: Gson,
    private val type: Type
) {
    fun call(callback: Callback<T>): Call {
        // 步骤1, 使用call请求API
        // 步骤2, 根据请求结果,调用callback.onSuccess()或者是callback.onFail()
        // 步骤3, 返回OkHttp的Call对象
    }
}

我们一步步来分析这三个步骤:

接下来,我们看看具体代码是怎么样的:

class KtCall<T: Any>(
    private val call: Call,
    private val gson: Gson,
    private val type: Type
) {
    fun call(callback: Callback<T>): Call {
        call.enqueue(object : okhttp3.Callback {
            override fun onFailure(call: Call, e: IOException) {
                callback.onFail(e)
            }

            override fun onResponse(call: Call, response: Response) {
                try { // ①
                    val t = gson.fromJson<T>(response.body?.string(), type)
                    callback.onSuccess(t)
                } catch (e: Exception) {
                    callback.onFail(e)
                }
            }
        })
        return call
    }
}

经过前面的解释,这段代码就很好理解了,唯一需要注意的是注释①处,由于API返回的结果并不可靠,即使请求成功了,其中的JSON数据也不一定合法,所以这里我们一般还需要进行额外的判断。在实际的商业项目当中,我们可能还需要根据当中的状态码,进行进一步区分和封装,这里为了便于理解,我就简单处理了。

那么在实现了KtCall以后,我们就只差ApiService这个接口了,这里我们定义ApiServiceV3,以作区分。

interface ApiServiceV3 {
    @GET("/repo")
    fun repos(
        @Field("lang") lang: String,
        @Field("since") since: String
    ): KtCall<RepoList> // ①
}

我们需要格外留意以上代码中的注释①,这其实就是3.0和1.0之间的最大区别。由于repo()方法的返回值类型是KtCall,为了支持这种写法,我们的invoke方法就需要跟着做一些小的改动:

// 这里也同样使用了泛型边界
private fun <T: Any> invoke(path: String, method: Method, args: Array<Any>): Any? {
    if (method.parameterAnnotations.size != args.size) return null

    var url = path
    val parameterAnnotations = method.parameterAnnotations
    for (i in parameterAnnotations.indices) {
        for (parameterAnnotation in parameterAnnotations[i]) {
            if (parameterAnnotation is Field) {
                val key = parameterAnnotation.value
                val value = args[i].toString()
                if (!url.contains("?")) {
                    url += "?$key=$value"
                } else {
                    url += "&$key=$value"
                }

            }
        }
    }

    val request = Request.Builder()
        .url(url)
        .build()

    val call = okHttpClient.newCall(request)
    val genericReturnType = getTypeArgument(method)
    
    // 变化在这里
    return KtCall<T>(call, gson, genericReturnType)
}

// 拿到 KtCall<RepoList> 当中的 RepoList类型
private fun getTypeArgument(method: Method) =
    (method.genericReturnType as ParameterizedType).actualTypeArguments[0]

在上面的代码中,大部分代码和1.0版本的一样的,只是在最后封装了一个KtCall对象,直接返回。所以在后续调用它的时候,我们就可以这么写了:ktCall.call()。

private fun testAsync() {
    // 创建api对象
    val api: ApiServiceV3 = KtHttpV3.create(ApiServiceV3::class.java)

    // 获取ktCall
    val ktCall: KtCall<RepoList> = api.repos(
        lang = "Kotlin",
        since = "weekly"
    )

    // 发起call异步请求
    ktCall.call(object : Callback<RepoList> {
        override fun onSuccess(data: RepoList) {
            println(data)
        }

        override fun onFail(throwable: Throwable) {
            println(throwable)
        }
    })
}

以上代码很好理解,我们一步步创建API对象、ktCall对象,最后发起请求。不过,在工作中一般是不会这么写代码的,因为创建太多一次性临时对象了。我们完全可以用链式调用的方式来做:

private fun testAsync() {
    KtHttpV3.create(ApiServiceV3::class.java)
    .repos(
        lang = "Kotlin",
        since = "weekly"
    ).call(object : Callback<RepoList> {
        override fun onSuccess(data: RepoList) {
            println(data)
        }

        override fun onFail(throwable: Throwable) {
            println(throwable)
        }
    })
}

如果你没有很多编程经验,那你可能会对这种方式不太适应,但在实际写代码的过程中,你会发现这种模式写起来会比上一种舒服很多,因为你再也不用为临时变量取名字伤脑筋了

总的来说,到这里的话,我们的异步请求接口就已经完成了。而且,由于我们的实际请求已经通过OkHttp派发(enqueue)到统一的线程池当中去了,并不会阻塞主线程,所以这样的代码模式执行在Android、Swing之类的UI编程平台,也不会引起UI界面卡死的问题。

那么,3.0版本是不是到这里就结束了呢?其实并没有,因为我们还有一种情况没有考虑。我们来看看下面这段代码示例:

interface ApiServiceV3 {
    @GET("/repo")
    fun repos(
        @Field("lang") lang: String,
        @Field("since") since: String
    ): KtCall<RepoList>

    @GET("/repo")
    fun reposSync(
        @Field("lang") lang: String,
        @Field("since") since: String
    ): RepoList // 注意这里
}

private fun testSync() {
    val api: ApiServiceV3 = KtHttpV3.create(ApiServiceV3::class.java)
    val data: RepoList = api.reposSync(lang = "Kotlin", since = "weekly")
    println(data)
}

请留意注释的地方,repoSync()的返回值类型是RepoList,而不是KtCall类型,这其实是我们1.0版本的写法。看到这,你是不是发现问题了?虽然KtHttp支持了异步请求,但原本的同步请求反而不支持了。

所以,为了让KtHttp同时支持两种请求方式,我们只需要增加一个 if判断即可:

private fun <T: Any> invoke(path: String, method: Method, args: Array<Any>): Any? {
    // 省略其他代码

    return if (isKtCallReturn(method)) {
        val genericReturnType = getTypeArgument(method)
        KtCall<T>(call, gson, genericReturnType)
    } else {
        // 注意这里
        val response = okHttpClient.newCall(request).execute()

        val genericReturnType = method.genericReturnType
        val json = response.body?.string()
        gson.fromJson<Any?>(json, genericReturnType)
    }
}

// 判断当前接口的返回值类型是不是KtCall
private fun isKtCallReturn(method: Method) =
    getRawType(method.genericReturnType) == KtCall::class.java

在上面的代码中,我们定义了一个方法isKtCallReturn(),它的作用是判断当前接口方法的返回值类型是不是KtCall,如果是的话,我们就认为它是一个异步接口,这时候返回KtCall对象;如果不是,我们就认为它是同步接口。这样我们只需要将1.0的逻辑挪到else分支,就可以实现兼容了。

那么到这里,我们3.0版本的开发就算是完成了。接下来,我们进入4.0版本的开发。

4.0 版本:支持挂起函数

终于来到协程实战的部分了。在日常的开发工作当中,你也许经常会面临这样的一个问题:虽然很想用Kotlin的协程来简化异步开发,但公司的底层框架全部都是Callback写的,根本不支持挂起函数,我一个上层的业务开发工程师,能有什么办法呢?

其实,我们当前的KtHttp就面临着类似的问题:3.0版本只支持Callback异步调用,现在我们想要扩展出挂起函数的功能。这其实就是大部分Kotlin开发者会遇到的场景。

就我这几年架构迁移的实践经验来看,针对这个问题,我们主要有两种解法:

下面我们先来看看第一种解法。至于第二种解法,其实还可以细分出好几种思路,由于它涉及到挂起函数更底层的一些知识,具体方案我会在源码篇的第27讲介绍。

解法一:扩展KtCall

这种方式有一个优势,那就是我们不需要改动3.0版本的任何代码。这种场景在工作中也是十分常见的,比如说,项目中用到的SDK是开源的,或者SDK是公司其他部门开发的,我们无法改动SDK。

具体的做法,就是为KtCall这个类扩展出一个挂起函数。

/*
注意这里                   函数名称
   ↓                        ↓        */
suspend fun <T: Any> KtCall<T>.await(): T = TODO()

在上面的代码中,我们定义了一个扩展函数await()。首先,它是一个挂起函数,其次,它的扩展接收者类型是KtCall,其中带着一个泛型T,挂起函数的返回值也是泛型T。

而由于它是一个挂起函数,所以,我们的代码就可以换成这样的方式来写了。

fun main() = runBlocking {
    val ktCall = KtHttpV3.create(ApiServiceV3::class.java)
        .repos(lang = "Kotlin", since = "weekly")

    val result = ktCall.await() // 调用挂起函数
    println(result)
}

那么,现在我们就只剩下一个问题了:await()具体该如何实现?

在这里,我们需要用到Kotlin官方提供的一个顶层函数:suspendCoroutine{},它的函数签名是这样的:

public suspend inline fun <T> suspendCoroutine(crossinline block: (Continuation<T>) -> Unit): T {
    // 省略细节
}

从它的函数签名,我们可以发现,它是一个挂起函数,也是一个高阶函数,参数类型是“(Continuation) -> Unit”,如果你还记得第15讲当中的内容,你应该就已经发现了,它其实就等价于挂起函数类型!

所以,我们可以使用suspendCoroutine{} 来实现await()方法:

/*
注意这里                   
   ↓                                */
suspend fun <T: Any> KtCall<T>.await(): T = suspendCoroutine{
    continuation ->
    //   ↑
    // 注意这里 
}

如果你仔细分析这段代码的话,会发现suspendCoroutine{} 的作用,其实就是将挂起函数当中的continuation暴露出来

那么,suspendCoroutine{} 当中的代码具体该怎么写呢?答案应该也很明显了,当然是要用这个被暴露出来的continuation来做文章啦!

这里我们再来回顾一下Continuation这个接口:

public interface Continuation<in T> {
    public val context: CoroutineContext
    // 关键在于这个方法
    public fun resumeWith(result: Result<T>)
}

通过定义可以看到,整个Continuation只有一个方法,那就是resumeWith(),根据它的名字我们就可以推测出,它是用于“恢复”的,参数类型是Result。所以很明显,这就是一个带有泛型的“结果”,它的作用就是承载协程执行的结果。

所以,综合来看,我们就可以进一步写出这样的代码了:

suspend fun <T: Any> KtCall<T>.await(): T =
    suspendCoroutine { continuation ->
        call(object : Callback<T> {
            override fun onSuccess(data: T) {
                continuation.resumeWith(Result.success(data))
            }

            override fun onFail(throwable: Throwable) {
                continuation.resumeWith(Result.failure(throwable))
            }
        })
    }

以上代码也很容易理解,当网络请求执行成功以后,我们就调用resumeWith(),同时传入Result.success(data);如果请求失败,我们就传入Result.failure(throwable),将对应的异常信息传进去。

不过,也许你会觉得创建Result的写法太繁琐了,没关系,你可以借助Kotlin官方提供的扩展函数提升代码可读性。

suspend fun <T : Any> KtCall<T>.await(): T =
    suspendCoroutine { continuation ->
        call(object : Callback<T> {
            override fun onSuccess(data: T) {
                continuation.resume(data)
            }

            override fun onFail(throwable: Throwable) {
                continuation.resumeWithException(throwable)
            }
        })
    }

到目前为止,await()这个扩展函数其实就已经实现了。这时候,如果我们在协程当中调用await()方法的话,代码是可以正常执行的。不过,这种做法其实还有一点瑕疵,那就是不支持取消

让我们来写一个简单的例子:

fun main() = runBlocking {
    val start = System.currentTimeMillis()
    val deferred = async {
        KtHttpV3.create(ApiServiceV3::class.java)
            .repos(lang = "Kotlin", since = "weekly")
            .await()
    }

    deferred.invokeOnCompletion {
        println("invokeOnCompletion!")
    }
    delay(50L)

    deferred.cancel()
    println("Time cancel: ${System.currentTimeMillis() - start}")

    try {
        println(deferred.await())
    } catch (e: Exception) {
        println("Time exception: ${System.currentTimeMillis() - start}")
        println("Catch exception:$e")
    } finally {
        println("Time total: ${System.currentTimeMillis() - start}")
    }
}

suspend fun <T : Any> KtCall<T>.await(): T =
    suspendCoroutine { continuation ->
        call(object : Callback<T> {
            override fun onSuccess(data: T) {
                println("Request success!") // ①
                continuation.resume(data)
            }

            override fun onFail(throwable: Throwable) {
                println("Request fail!:$throwable")
                continuation.resumeWithException(throwable)
            }
        })
    }

/*
输出结果:
Time cancel: 536   // ②
Request success!   // ③
invokeOnCompletion!
Time exception: 3612  // ④
Catch exception:kotlinx.coroutines.JobCancellationException: DeferredCoroutine was cancelled; job=DeferredCoroutine{Cancelled}@6043cd28
Time total: 3612
*/

在main函数当中,我们在async里调用了挂起函数,接着50ms过去后,我们就去尝试取消协程。这段代码中一共有三处地方需要注意,我们来分析一下:

综上所述,当我们使用suspendCoroutine{} 来实现挂起函数的时候,默认情况下是不支持取消的。那么,具体该怎么做呢?其实也很简单,就是使用Kotlin官方提供的另一个API:suspendCancellableCoroutine{}

suspend fun <T : Any> KtCall<T>.await(): T =
//            变化1
//              ↓
    suspendCancellableCoroutine { continuation ->
        val call = call(object : Callback<T> {
            override fun onSuccess(data: T) {
                println("Request success!")
                continuation.resume(data)
            }

            override fun onFail(throwable: Throwable) {
                println("Request fail!:$throwable")
                continuation.resumeWithException(throwable)
            }
        })

//            变化2
//              ↓
        continuation.invokeOnCancellation {
            println("Call cancelled!")
            call.cancel()
        }
    }

当我们使用suspendCancellableCoroutine{} 的时候,可以往continuation对象上面设置一个监听:invokeOnCancellation{},它代表当前的协程被取消了,这时候,我们只需要将OkHttp的call取消即可。

这样一来,main()函数就能保持不变,得到的输出结果却大不相同。

/*
suspendCoroutine结果:

Time cancel: 536   
Request success!   
invokeOnCompletion!
Time exception: 3612  // ①
Catch exception:kotlinx.coroutines.JobCancellationException: DeferredCoroutine was cancelled; job=DeferredCoroutine{Cancelled}@6043cd28
Time total: 3612
*/

/*
suspendCancellableCoroutine结果:

Call cancelled!
Time cancel: 464
invokeOnCompletion!
Time exception: 466  // ②
Catch exception:kotlinx.coroutines.JobCancellationException: DeferredCoroutine was cancelled; job=DeferredCoroutine{Cancelled}@6043cd28
Time total: 466
Request fail!:java.io.IOException: Canceled  // ③
*/

对比注释①、②,可以发现,后者是会立即响应协程取消事件的,所以当代码执行到deferred.await()的时候,会立即抛出异常,而不会挂起很长时间。另外,通过注释③这里的结果,我们也可以发现,OkHttp的网络请求确实被取消了。

所以,我们可以得出一个结论,使用suspendCancellableCoroutine{},我们可以避免不必要的挂起,比如例子中的deferred.await();另外也可以节省计算机资源,因为这样可以避免不必要的协程任务,比如这里被成功取消的网络请求。

到这里,我们的解法一就已经完成了。这种方式并没有改动KtHttp的源代码,而是以扩展函数来实现的。所以,从严格意义上来讲,KtHttp 4.0版本并没有开发完毕,等到第27讲我们深入理解了挂起函数的底层原理后,我们再来完成解法二的代码。

小结

这节课,我们在KtHttp 1.0版本的基础上,扩展出了异步请求的功能,完成了3.0版本的开发;接着,我们又在3.0版本的基础上,让KtHttp支持了挂起函数,这里我们是用的外部扩展的思路,并没有碰KtHttp内部的代码。

这里主要涉及以下几个知识点:

思考题

你能分析出下面的代码执行结果吗?为什么会是这样的结果?它能给你带来什么启发?欢迎给我留言,也欢迎你把今天的内容分享给更多的朋友。

fun main() = runBlocking {
    val start = System.currentTimeMillis()
    val deferred = async {
        KtHttpV3.create(ApiServiceV3::class.java)
            .repos(lang = "Kotlin", since = "weekly")
            .await()
    }

    deferred.invokeOnCompletion {
        println("invokeOnCompletion!")
    }
    delay(50L)

    deferred.cancel()
    println("Time cancel: ${System.currentTimeMillis() - start}")

    try {
        println(deferred.await())
    } catch (e: Exception) {
        println("Time exception: ${System.currentTimeMillis() - start}")
        println("Catch exception:$e")
    } finally {
        println("Time total: ${System.currentTimeMillis() - start}")
    }
}

suspend fun <T : Any> KtCall<T>.await(): T =
    suspendCancellableCoroutine { continuation ->
        val call = call(object : Callback<T> {
            override fun onSuccess(data: T) {
                println("Request success!")
                continuation.resume(data)
            }

            override fun onFail(throwable: Throwable) {
                println("Request fail!:$throwable")
                continuation.resumeWithException(throwable)
            }
        })

// 注意这里
//        continuation.invokeOnCancellation {
//            println("Call cancelled!")
//            call.cancel()
//        }
    }

评论