如何更改kotlin StateFlow的数据源

如何解决如何更改kotlin StateFlow的数据源

我正在将此代码示例迁移到 StateFlow。

class RosterMotor(private val repo: TodoRepository) : viewmodel() {
  private val _states = MediatorLiveData<RosterViewState>()
  val states: LiveData<RosterViewState> = _states
  private var lastSource: LiveData<RosterViewState>? = null

  init {
    load(FilterMode.ALL)
  }

  fun load(filterMode: FilterMode) {
    lastSource?.let { _states.removeSource(it) }

    val items =
      repo.items(filterMode).map { RosterViewState(it,filterMode) }.asLiveData()

    _states.addSource(items) { viewstate ->
      _states.value = viewstate
    }

    lastSource = items
  }
...
}

此样本取自 https://commonsware.com/AndExplore/ 本书

我能想到这个解决方案，但我不确定这是否是最好的方法

private val _states = MutableStateFlow(RosterViewState())
val states: StateFlow<RosterViewState> = _states

init {
    load(ALL)
}

fun load(filterMode: FilterMode) {
    viewmodelScope.launch {
        repository.items(filterMode).map { RosterViewState(it,filterMode) }
            .collect {
                _states.value = it
            }
    }
}

那么我们如何使用 StateFlow 来实现这个场景。