android源码宇宙-StateFlow

1,079 阅读3分钟

用法

image.png

image.png

源码

collect添加观察者

  1. MutableStateFlow.collect

collect是一个扩展方法,是唯一入口
下一步就是找到真实Flow实现类里面的collect方法
image.png

  1. 查看MutableStateFlow的实现,MutableStateFlow的真实实现类是StateFlowImpl

image.png

image.png

  1. StateFlowImpl.collect方法

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

image.png

image.png

StateFlow.value更新数据

  1. StateFlowImpl.setValue

image.png

image.png

  1. updateState方法

image.png
image.png

其它

StateFlow中的状态介绍

image.png

image.png

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

StateFlow使用StateFlowSlot来管理状态,因为StateFlow有多个观察者,所以会使用数组来管理StateFlowSlot
这个StateFlowSlot内部管理了当前观察者的状态,每当我们添加观察者、遍历观察者给每个观察者回调数据、取消观察者的时候都会更改这个StateFlowSlot的状态。

  1. 首先来到StateFlowImpl.collect方法

image.png

image.png

  1. 看一下allocateSlot方法

因为我们的StateFlowImpl实现了AbstractSharedFlow,所以该方法在AbstractSharedFlow中
image.png

  1. StateFlowImpl.createSlot方法

image.png

image.png

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

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

  1. StateFlowImpl.collect

image.png

image.png

  1. awaitPending

image.png

image.png

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

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

  1. StateFlowImpl.updateState

起点是这个方法

重点StateFlowSlot.makePending

image.png

image.png

  1. StateFlowSolt.makePending

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

总结

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

工作原理:
StateFlow使用collect方法注册观察者,每一个观察者内部都维护了一个死循环,当数据更新完成的时候会将协程挂起保证死循环不退出。
每一个StateFlow都会维护一个StateFlowSlot的数组(可以称之为状态槽slots)用来管理每个观察者的状态。当我们调用collect方法注册了一个观察者的时候,首先会将StateFlow中当前的值回调给观察者,然后StateFlow会将观察者的协程挂起,等待下一次数据更新到来。
当下一次数据到来的时候,StateFlow会将数据更新到最新值,然后会遍历slots,将所有观察者对应的协程结束挂起,这样所有的观察者,会继续在死循环中去读取StateFlow中的最新值,并回调。