Как изменить источник данных kotlin StateFlow

Question

Я переношу этот образец кода на 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.