Kotlin 协程版的 AutoDispose

深藏阁楼爱情的钟 2022-04-25 07:12 211阅读 0赞

![640?wx\_fmt=png][640_wx_fmt_png]

大家一定用过 RxJava，也一定知道用 RxJava 发了个任务，任务还没结束页面就被关闭了，如果任务迟迟不回来，页面就会被泄露；如果任务后面回来了，执行回调更新 UI 的时候也会大概率空指针。

因此大家一定会用到 Uber 的开源框架 AutoDispose。

什么？你说你没用？好吧，那就没用吧。。我是不会介绍它的。⊙﹏⊙|||。怎么可能。(～￣▽￣)～。。它其实就是利用 `View` 的 `OnAttachStateChangeListener` ，当 `View` 被拿下的时候，我们就取消所有之前用 RxJava 发出去的请求。

1.  ` static final class Listener extends MainThreadDisposable implements View.OnAttachStateChangeListener {`
2.  `   private final View view;`
3.  `   private final CompletableObserver observer;`
4.    
    
5.  `   Listener(View view, CompletableObserver observer) {`
6.  `     this.view = view;`
7.  `     this.observer = observer;`
8.  `   }`
9.    
    
10. `   @Override public void onViewAttachedToWindow(View v) { }`
11.   
    
12. `   @Override public void onViewDetachedFromWindow(View v) {`
13. `     if (!isDisposed()) {`
14. `     //看到没看到没看到没？`
15. `       observer.onComplete();`
16. `     }`
17. `   }`
18.   
    
19. `   @Override protected void onDispose() {`
20. `     view.removeOnAttachStateChangeListener(this);`
21. `   }`
22. ` }`

好了，我最近在想我们用协程其实也会有这样的问题呀：

1.  `button.onClick {`
2.  `   try {`
3.  `       val req = Request()`
4.  `       val resp = async { sendRequest(req) }.await()`
5.  `       updateUI(resp)`
6.  `   } catch (e: Exception) {`
7.  `       e.printStackTrace()`
8.  `   }`
9.  `}`

如果 `await` 返回结果之前我们就退出了当前的 `Activity` 那么，后面 `updateUI` 就要凉凉。这就尴尬了。不过问题不大，照猫画虎谁不会，我们也可以搞一个 `onClickAutoDisposable` 嘛。

1.  `fun View.onClickAutoDisposable (`
2.  `       context: CoroutineContext = Dispatchers.Main,`
3.  `       handler: suspend CoroutineScope.(v: android.view.View?) -> Unit`
4.  `) {`
5.  `   setOnClickListener { v ->`
6.  `       GlobalScope.launch(context, CoroutineStart.DEFAULT) {`
7.  `           handler(v)`
8.  `       }.asAutoDisposable(v)`
9.  `   }`
10. `}`

第一步，不要脸的先抄 Anko 的 `onClick`，不同之处在于我们改了个名 XD。啊，还有我们加了个 `.asAutoDisposable(v)`，大家就假装有这个方法吧。。。

> (╬￣皿￣)=○ 假装个头啊，假装就完成功能的话还要程序员干什么。。让产品假装一下不就行了。。

OK OK，咱们下面来实现它。。想想， `GlobalScope.launch` 其实返回的是一个 `Job`，所以嘛，我们给 `Job` 搞一个扩展方法不就得了。

1.  `fun Job.asAutoDisposable(view: View) = AutoDisposableJob(view, this)`

第二步，我们再偷偷的创建一个类，叫 `AutoDisposableJob`，抄一下前面的 `Listener`：

1.  `class AutoDisposableJob(private val view: View, private val wrapped: Job)`
2.  `   //我们实现了 Job 这个接口，但没有直接实现它的方法，而是用 wrapped 这个成员去代理这个接口`
3.  `    : Job by wrapped, OnAttachStateChangeListener {`
4.  `   override fun onViewAttachedToWindow(v: View?) = Unit`
5.    
    
6.  `   override fun onViewDetachedFromWindow(v: View?) {`
7.  `       //当 View 被移除的时候，取消协程`
8.  `       cancel()`
9.  `       view.removeOnAttachStateChangeListener(this)`
10. `   }`
11.   
    
12. `   private fun isViewAttached() =`
13. `           Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT && view.isAttachedToWindow || view.windowToken != null`
14.   
    
15. `   init {`
16. `       if(isViewAttached()) {`
17. `           view.addOnAttachStateChangeListener(this)`
18. `       } else {`
19. `           cancel()`
20. `       }`
21.   
    
22. `       //协程执行完毕时要及时移除 listener 免得造成泄露`
23. `       invokeOnCompletion() {`
24. `           view.removeOnAttachStateChangeListener(this)`
25. `       }`
26. `   }`
27. `}`

这样的话，我们就可以使用这个扩展了：

1.  `button.onClickAutoDisposable{`
2.  `   try {`
3.  `       val req = Request()`
4.  `       val resp = async { sendRequest(req) }.await()`
5.  `       updateUI(resp)`
6.  `   } catch (e: Exception) {`
7.  `       e.printStackTrace()`
8.  `   }`
9.  `}`

当 `button` 这个对象从 window 上撤下来的时候，我们的协程就会收到 cancel 的指令。

好的，我又凭实力忽悠了本周的文章。。。

--------------------

转载请注明出处：微信公众号 Kotlin

![640?wx\_fmt=jpeg][640_wx_fmt_jpeg]

[640_wx_fmt_png]: /images/20220125/8ce6bb20da1641458b06da795500456d.png
[640_wx_fmt_jpeg]: /images/20220125/3e10714a7bc149c39c35c192527ebcc2.png