android源码宇宙-StateFlow

用法

image.png

源码

collect添加观察者

1. MutableStateFlow.collect

collect是一个扩展方法，是唯一入口

下一步就是找到真实Flow实现类里面的collect方法

2. 查看MutableStateFlow的实现，MutableStateFlow的真实实现类是StateFlowImpl

image.png

3. StateFlowImpl.collect方法

下图中的freeSlot(slot)我们需要说明一下，它实际上就是用来释放当前位置的观察者的，当while结束的时候就会调用freeSlot来释放当前观察者对应的状态。 下面的slot.awaitPending()方法也需要说明一下，他实际上就是在我们更新完一次数据后重新将我们的协程挂起，等待下一次数据更新到来

image.png

StateFlow.value更新数据

1. StateFlowImpl.setValue

image.png

2. updateState方法

其它

StateFlow中的状态介绍

image.png

StateFlow是如何管理多个观察者的

StateFlow使用StateFlowSlot来管理状态，因为StateFlow有多个观察者，所以会使用数组来管理StateFlowSlot

这个StateFlowSlot内部管理了当前观察者的状态，每当我们添加观察者、遍历观察者给每个观察者回调数据、取消观察者的时候都会更改这个StateFlowSlot的状态。

1. 首先来到StateFlowImpl.collect方法

image.png

2. 看一下allocateSlot方法

因为我们的StateFlowImpl实现了AbstractSharedFlow，所以该方法在AbstractSharedFlow中

3. StateFlowImpl.createSlot方法

image.png

StateFlow.collect如何实现多次挂起，多次收到回调数据

实际上就是每一个collect方法内部都维护了一个死循环，每次到来新数据的时候都会调用emit更新新数据，同时将协程重新挂起，等待下一次数据到来的时候再结束挂起。

1. StateFlowImpl.collect

image.png

2. awaitPending

image.png

StateFlow.value设置新值的时候是如何结束所有观察者的挂起状态的

结论：我们知道，每次给StateFlow更新完数据后，都会将协程重新挂起，挂起的过程中会顺便把协程的Continuation赋值给StateFlowSlot对应的_state。当我们新数据到来的时候会调用_state(Continuation).resume来恢复协程，协程结束挂起状态。

1. StateFlowImpl.updateState

起点是这个方法

重点StateFlowSlot.makePending

image.png

2. StateFlowSolt.makePending

进一步我们可能会想这个_state为什么会是一个协程挂起对象，他是在哪里设置的，见下图：

总结

相较于LiveData，StateFlow的实现更加简练，它不依赖于我们的生命周期，因此控制StateFlow的状态只需要管理好包裹它的父协程状态即可。

工作原理：

StateFlow使用collect方法注册观察者，每一个观察者内部都维护了一个死循环，当数据更新完成的时候会将协程挂起保证死循环不退出。

每一个StateFlow都会维护一个StateFlowSlot的数组(可以称之为状态槽slots)用来管理每个观察者的状态。当我们调用collect方法注册了一个观察者的时候，首先会将StateFlow中当前的值回调给观察者，然后StateFlow会将观察者的协程挂起，等待下一次数据更新到来。

当下一次数据到来的时候，StateFlow会将数据更新到最新值，然后会遍历slots，将所有观察者对应的协程结束挂起，这样所有的观察者，会继续在死循环中去读取StateFlow中的最新值，并回调。