用法
image.png
源码
collect添加观察者
- MutableStateFlow.collect
collect是一个扩展方法,是唯一入口
下一步就是找到真实Flow实现类里面的collect方法
- 查看MutableStateFlow的实现,MutableStateFlow的真实实现类是StateFlowImpl
image.png
- StateFlowImpl.collect方法
下图中的freeSlot(slot)我们需要说明一下,它实际上就是用来释放当前位置的观察者的,当while结束的时候就会调用freeSlot来释放当前观察者对应的状态。 下面的slot.awaitPending()方法也需要说明一下,他实际上就是在我们更新完一次数据后重新将我们的协程挂起,等待下一次数据更新到来
image.png
StateFlow.value更新数据
- StateFlowImpl.setValue
image.png
- updateState方法
其它
StateFlow中的状态介绍
image.png
StateFlow是如何管理多个观察者的
StateFlow使用StateFlowSlot来管理状态,因为StateFlow有多个观察者,所以会使用数组来管理StateFlowSlot
这个StateFlowSlot内部管理了当前观察者的状态,每当我们添加观察者、遍历观察者给每个观察者回调数据、取消观察者的时候都会更改这个StateFlowSlot的状态。
- 首先来到StateFlowImpl.collect方法
image.png
- 看一下allocateSlot方法
因为我们的StateFlowImpl实现了AbstractSharedFlow,所以该方法在AbstractSharedFlow中
- StateFlowImpl.createSlot方法
image.png
StateFlow.collect如何实现多次挂起,多次收到回调数据
实际上就是每一个collect方法内部都维护了一个死循环,每次到来新数据的时候都会调用emit更新新数据,同时将协程重新挂起,等待下一次数据到来的时候再结束挂起。
- StateFlowImpl.collect
image.png
- awaitPending
image.png
StateFlow.value设置新值的时候是如何结束所有观察者的挂起状态的
结论:我们知道,每次给StateFlow更新完数据后,都会将协程重新挂起,挂起的过程中会顺便把协程的Continuation赋值给StateFlowSlot对应的_state。当我们新数据到来的时候会调用_state(Continuation).resume来恢复协程,协程结束挂起状态。
- StateFlowImpl.updateState
起点是这个方法
重点StateFlowSlot.makePending
image.png
- StateFlowSolt.makePending
进一步我们可能会想这个_state为什么会是一个协程挂起对象,他是在哪里设置的,见下图:
总结
相较于LiveData,StateFlow的实现更加简练,它不依赖于我们的生命周期,因此控制StateFlow的状态只需要管理好包裹它的父协程状态即可。
工作原理:
StateFlow使用collect方法注册观察者,每一个观察者内部都维护了一个死循环,当数据更新完成的时候会将协程挂起保证死循环不退出。
每一个StateFlow都会维护一个StateFlowSlot的数组(可以称之为状态槽slots)用来管理每个观察者的状态。当我们调用collect方法注册了一个观察者的时候,首先会将StateFlow中当前的值回调给观察者,然后StateFlow会将观察者的协程挂起,等待下一次数据更新到来。
当下一次数据到来的时候,StateFlow会将数据更新到最新值,然后会遍历slots,将所有观察者对应的协程结束挂起,这样所有的观察者,会继续在死循环中去读取StateFlow中的最新值,并回调。