ViewModelScope 避免内存泄漏的原理

避免的是什么？

避免的是协程的内存泄漏。

如何避免？

总体逻辑

通过 lifecycle 监听 Activity 的生命周期，在 Activity 销毁时对协程进行 cancel。

监听状态变化

下图代码是注册监听的地方。可以看到在 Activity 销毁时调用了 getViewModelStore().clear()
从字面意思我们可以看出这是做了清理的操作。

如何清理？

我们继续追踪 **getViewModelStore().clear()**发现这个方法实际上遍历所有的 ViewModel 实例，依次执行 ViewModel 的清理方法。

继续看下 ViewModel 的清理方法。

我们发现 ViewModel 的清理方法的主要操作对象是 mBagOfTags，看下它的定义。

也就是一个存储 Object 对象的 map，我们可以猜测这个 map 就是用于存储协程相关的数据，然后用于清理操作。
我们看下这个 map 是在哪里 put 的，来验证我们的想法，我们看下 viewmodelscope 的源码。

我们发现这个 setTagIfAbsent 可能就是 put 的地方，我们继续看下它的定义。

果然如此，每次获取 ViewModelScope 时拿到的都是一个 CloseableCoroutineScope 而这个类自己自带一个清理方法 close，猜测应该会在 viewmodel 的 clear 方法调用到。

我们验证下前面的猜测。

可以看到在 closeWithRuntimeException 中确实调用了 close 方法。

总结

至此，大体上理清楚了 ViewModelScope 避免内存泄漏逻辑。就是通过一个 map 存储每次获取 viewmodelsope 返回的 CloseableCoroutineScope ，然后在 Activity 销毁时，调用CloseableCoroutineScope 的 close 方法。