You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

618 lines
25 KiB
Markdown

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

# 18 | 实战让KtHttp支持挂起函数
你好,我是朱涛。今天这节实战课,我们接着前面[第12讲](https://time.geekbang.org/column/article/481787)里实现的网络请求框架来进一步完善这个KtHttp让它支持挂起函数。
在上一次实战课当中我们已经开发出了两个版本的KtHttp1.0版本的是基于命令式风格的2.0版本的是基于函数式风格的。其中2.0版本的代码风格跟我们平时工作写的代码风格很不一样之前我也说了这主要是因为业界对Kotlin函数式编程接纳度并不高所以这节课的代码我们将基于1.0版本的代码继续改造。这样也能让课程的内容更接地气一些甚至你都可以借鉴今天写代码的思路复用到实际的Android或者后端开发中去。
跟往常一样,这节课的代码还是会分为两个版本:
* 3.0 版本在之前1.0版本的基础上,扩展出**异步请求**的能力。
* 4.0 版本,进一步扩展异步请求的能力,让它**支持挂起函数**。
好,接下来就正式开始吧!
## 3.0 版本支持异步Call
有了上一次实战课的基础这节课就会轻松一些了。关于动态代理、注解、反射之类的知识不会牵涉太多我们今天主要把精力都集中在协程上来。不过在正式开始写协程代码之前我们需要先让KtHttp支持异步请求也就是Callback请求。
这是为什么呢?别忘了[第15讲](https://time.geekbang.org/column/article/487085)的内容:**挂起函数本质就是Callback**所以为了让KtHttp支持挂起函数我们可以采用迂回的策略让它先支持Callback。在之前1.0、2.0版本的代码中KtHttp是只支持同步请求的你可能对异步同步还有些懵我带你来看个例子吧。
首先,这个是同步代码:
```plain
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编程平台上是会导致主线程卡死的。
那么,异步代码又是长什么样的呢?
```plain
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而且代码不再是按照顺序执行的了。你可以参考下面这个动图
![图片](https://static001.geekbang.org/resource/image/a8/ff/a8d0c46d74a17d0ddfb683e0ac7468ff.gif?wh=1080x608)
所以在3.0版本的开发中我们就是要实现类似上面testAsync()的请求方式。为此,我们首先需要创建一个**Callback接口**在这个Callback当中我们可以拿到API请求的结果。
```plain
interface Callback<T: Any> {
fun onSuccess(data: T)
fun onFail(throwable: Throwable)
}
```
在Callback这个接口里有一个泛型参数T还有两个回调分别是onSuccess代表接口请求成功、onFail代表接口请求失败。需要特别注意的是这里我们运用了[空安全思维](https://time.geekbang.org/column/article/484921)当中的**泛型边界“T: Any”**这样一来我们就可以保证T类型一定是非空的。
除此之外,我们还需要一个**KtCall类**它的作用是承载Callback或者说它是用来调用Callback的。
```plain
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()方法当中的逻辑会分为三个步骤。
```plain
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对象
}
}
```
我们一步步来分析这三个步骤:
* 步骤1使用OkHttp的call对象请求API这里需要注意的是为了将请求任务派发到异步线程我们需要使用OkHttp的异步请求方法enqueue()。
* 步骤2根据请求结果调用callback.onSuccess()或者是callback.onFail()。如果请求成功了我们在调用onSuccess()之前还需要用Gson将请求结果进行解析然后才返回。
* 步骤3返回OkHttp的Call对象。
接下来,我们看看具体代码是怎么样的:
```plain
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以作区分。
```plain
interface ApiServiceV3 {
@GET("/repo")
fun repos(
@Field("lang") lang: String,
@Field("since") since: String
): KtCall<RepoList> // ①
}
```
我们需要格外留意以上代码中的注释①,这其实就是**3.0和1.0之间的最大区别**。由于repo()方法的返回值类型是KtCall为了支持这种写法我们的invoke方法就需要跟着做一些小的改动
```plain
// 这里也同样使用了泛型边界
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()。
```plain
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对象最后发起请求。不过在工作中一般是不会这么写代码的因为创建太多一次性临时对象了。我们完全可以用**链式调用**的方式来做:
```plain
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版本是不是到这里就结束了呢?其实并没有,因为我们还有一种情况没有考虑。我们来看看下面这段代码示例:
```plain
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判断**即可:
```plain
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开发者会遇到的场景。
就我这几年架构迁移的实践经验来看,针对这个问题,我们主要有两种解法:
* 第一种解法不改动SDK内部的实现直接在SDK的基础上扩展出协程的能力。
* 第二种解法改动SDK内部让SDK直接支持挂起函数。
下面我们先来看看第一种解法。至于第二种解法其实还可以细分出好几种思路由于它涉及到挂起函数更底层的一些知识具体方案我会在源码篇的第27讲介绍。
### 解法一扩展KtCall
这种方式有一个优势那就是我们不需要改动3.0版本的任何代码。这种场景在工作中也是十分常见的比如说项目中用到的SDK是开源的或者SDK是公司其他部门开发的我们无法改动SDK。
具体的做法就是为KtCall这个类扩展出一个挂起函数。
```plain
/*
注意这里 函数名称
↓ ↓ */
suspend fun <T: Any> KtCall<T>.await(): T = TODO()
```
在上面的代码中我们定义了一个扩展函数await()。首先它是一个挂起函数其次它的扩展接收者类型是KtCall其中带着一个泛型T挂起函数的返回值也是泛型T。
而由于它是一个挂起函数,所以,我们的代码就可以换成这样的方式来写了。
```plain
fun main() = runBlocking {
val ktCall = KtHttpV3.create(ApiServiceV3::class.java)
.repos(lang = "Kotlin", since = "weekly")
val result = ktCall.await() // 调用挂起函数
println(result)
}
```
那么,现在我们就只剩下一个问题了:**await()具体该如何实现?**
在这里我们需要用到Kotlin官方提供的一个顶层函数suspendCoroutine{},它的函数签名是这样的:
```plain
public suspend inline fun <T> suspendCoroutine(crossinline block: (Continuation<T>) -> Unit): T {
// 省略细节
}
```
从它的函数签名,我们可以发现,它是一个挂起函数,也是一个高阶函数,参数类型是“(Continuation) -> Unit”如果你还记得第15讲当中的内容你应该就已经发现了**它其实就等价于挂起函数类型!**
所以我们可以使用suspendCoroutine{} 来实现await()方法:
```plain
/*
注意这里
↓ */
suspend fun <T: Any> KtCall<T>.await(): T = suspendCoroutine{
continuation ->
// ↑
// 注意这里
}
```
如果你仔细分析这段代码的话会发现suspendCoroutine{} 的作用,其实就是**将挂起函数当中的continuation暴露出来**。
那么suspendCoroutine{} 当中的代码具体该怎么写呢答案应该也很明显了当然是要用这个被暴露出来的continuation来做文章啦
这里我们再来回顾一下Continuation这个接口
```plain
public interface Continuation<in T> {
public val context: CoroutineContext
// 关键在于这个方法
public fun resumeWith(result: Result<T>)
}
```
通过定义可以看到整个Continuation只有一个方法那就是resumeWith()根据它的名字我们就可以推测出它是用于“恢复”的参数类型是Result。所以很明显这就是一个带有泛型的“结果”它的作用就是承载协程执行的结果。
所以,综合来看,我们就可以进一步写出这样的代码了:
```plain
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官方提供的扩展函数提升代码可读性。
```plain
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()方法的话,代码是可以正常执行的。不过,这种做法其实还有一点瑕疵,那就是**不支持取消**。
让我们来写一个简单的例子:
```plain
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过去后我们就去尝试取消协程。这段代码中一共有三处地方需要注意我们来分析一下
* 结合注释①、③一起分析我们发现即使调用了deferred.cancel()网络请求仍然会继续执行。根据“Catch exception:”输出的异常信息我们也发现当deferred被取消以后我们还去调用await()的时候,会抛出异常。
* 对比注释②、④我们还能发现deferred.await()虽然会抛出异常但是它却耗时3000ms。虽然deferred被取消了但是当我们调用await()的时候,它并不会马上就抛出异常,而是会等到内部的网络请求执行结束以后,才抛出异常,在此之前都会被挂起。
综上所述当我们使用suspendCoroutine{} 来实现挂起函数的时候默认情况下是不支持取消的。那么具体该怎么做呢其实也很简单就是使用Kotlin官方提供的另一个API**suspendCancellableCoroutine{}**。
```plain
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()函数就能保持不变,得到的输出结果却大不相同。
```plain
/*
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内部的代码。
这里主要涉及以下几个知识点:
* 在3.0版本开发中我们运用了泛型边界“T: Any”落实对泛型的非空限制同时通过封装KtCall为下一个版本打下了基础。
* 接着在4.0版本中我们借助扩展函数的特性为KtCall扩展了await()方法。
* 在实现await()的过程中我们使用了两个协程API分别是suspendCoroutine{}、suspendCancellableCoroutine{}在Kotlin协程当中我们**永远都要优先使用后者**。
* suspendCancellableCoroutine{} 主要有两大优势:第一,它可以避免不必要的挂起,提升运行效率;第二,它可以避免不必要的资源浪费,改善软件的综合指标。
## 思考题
你能分析出下面的代码执行结果吗?为什么会是这样的结果?它能给你带来什么启发?欢迎给我留言,也欢迎你把今天的内容分享给更多的朋友。
```plain
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()
// }
}
```