Retrofit源码分析笔记(二)Kotlin协程的实现

667 阅读2分钟

retrofit源码分析笔记(一)

上一篇我们分析了最简单的retrofit用法,其实只需要添加合适的CallAdapter就可以结合LiveData,和Rxjava。 这篇文章主要是介绍Retrofit是怎么适配Kotlin协程的。 首先我们看一下Retrofit适配协程的写法

image.png

经过上述的配置,Retrofit已经可以适配协程的写法。

那么Retrofit是怎么适配协程的呢。在上一篇文章中,我们了解到retrofit的入参解析,在入参解析的时候就kotlin就已经确定是否需要使用协程,下面的代码中,p == lastParameter如果为true则方法使用了协程。 因为呗suspend关键字修饰的方法,在编译后会在入参中多一个Continuation关键字。这时候实际参数,会比用retrofit注解修饰的参数个数多一。

image.png

并且在parseParameter方法中继续验证最后一个参数是否是Continuation

image.png

在来看返回值,和CallAdapter的选择,上一篇我们跳过了协程的case。 第一个红框中的代码是在区分协程中,我们使用的是什么写法的返回值,直接返回java对象,还是返回Response对象且泛型类型为java对象。将标志位 continuationWantsResponse置为true。 在区分完对象后,统一将返回值包装成Call对象。 之后选择CallAdapter就显而易见,和上一篇一样,会选择DefaultCallAdapterFactory。 原因是我们协程的返回值被我们封装程Call.class对象了。 第二个框中的代码就是将收集的参数全部组合起来,根据标志位创建不同的javabean。

image.png image.png

我们先来看SuspendForResponse,在调用retrofit的invoke方法后,最后会调用的SuspendForResponse的adapt方法,上面我们已经说了CallAdapter会选择DefaultCallAdapterFactory,调用 KotlinExtensions.awaitResponse(call, continuation);将选择的call对象,和continuation传入方法。这是java调用Kotlin的扩展方法,不懂的同学可以看看。

image.png

在awaitResponse方法中,调用了Call的enqueue方法,获取了请求返回值,DefaultCallAdapterFactory上一篇我们已经说过了,这里就不在叙述。

image.png

至此协程的流程结束。 SuspendForBody中调用流程是一样的,只是会在请求完成后多了一步解析的工作

image.png