如何从另一个类中收集ViewModel的StateFlow状态并触发对应方法？

首先是你提供的两个类代码：

SimpleViewModel 代码 #

class SimpleViewModel : ViewModel() {
    private val _state = MutableStateFlow(false)
    val state: StateFlow<Boolean> = _state
}

AnotherClass 代码 #

class AnotherClass {
    fun doWhenViewModelStateUpdateToTrue() {
        // 实现状态为true时的逻辑
    }
    fun doWhenViewModelStateUpdateToFalse() {
        // 实现状态为false时的逻辑
    }
}

实现方案 #

StateFlow的收集必须在协程作用域中进行，以下是几种常见场景的实现方式：

1. 在UI层（Activity/Fragment）中收集 #

使用lifecycleScope（自动绑定UI生命周期，避免内存泄漏）：

// 获取ViewModel实例
val viewModel = ViewModelProvider(this)[SimpleViewModel::class.java]
// 初始化AnotherClass实例
val anotherClass = AnotherClass()

// 启动协程收集状态更新
lifecycleScope.launch {
    viewModel.state.collect { isStateTrue ->
        // 根据状态值调用对应方法
        if (isStateTrue) {
            anotherClass.doWhenViewModelStateUpdateToTrue()
        } else {
            anotherClass.doWhenViewModelStateUpdateToFalse()
        }
    }
}

2. 在ViewModel内部收集 #

如果需要在ViewModel内部直接联动AnotherClass，使用viewModelScope（ViewModel销毁时自动取消协程）：

class SimpleViewModel : ViewModel() {
    private val _state = MutableStateFlow(false)
    val state: StateFlow<Boolean> = _state
    private val anotherClass = AnotherClass()

    init {
        // 启动协程监听状态变化
        viewModelScope.launch {
            state.collect { isStateTrue ->
                if (isStateTrue) {
                    anotherClass.doWhenViewModelStateUpdateToTrue()
                } else {
                    anotherClass.doWhenViewModelStateUpdateToFalse()
                }
            }
        }
    }

    // 示例：修改状态的方法
    fun toggleState() {
        _state.value = !_state.value
    }
}

3. 在普通类中收集 #

如果要在AnotherClass内部自行收集状态，需要手动管理协程作用域，记得在类不再使用时取消协程避免泄漏：

class AnotherClass {
    private val coroutineScope = CoroutineScope(Dispatchers.Default + SupervisorJob())

    fun startListeningToState(stateFlow: StateFlow<Boolean>) {
        coroutineScope.launch {
            stateFlow.collect { isStateTrue ->
                if (isStateTrue) {
                    doWhenViewModelStateUpdateToTrue()
                } else {
                    doWhenViewModelStateUpdateToFalse()
                }
            }
        }
    }

    fun doWhenViewModelStateUpdateToTrue() {
        // 实现逻辑
    }

    fun doWhenViewModelStateUpdateToFalse() {
        // 实现逻辑
    }

    // 类销毁前调用，取消协程
    fun cleanup() {
        coroutineScope.cancel()
    }
}

内容的提问来源于stack exchange，提问作者Mustafa Tatarhan