为何ViewModel中StateFlow的collect不会随按钮点击重复累积？

MainActivity.kt #

示例1 #

override fun onCreate(savedInstanceState: Bundle?) {
    super.onCreate(savedInstanceState)
    val activityMainBinding: ActivityMainBinding = ActivityMainBinding.inflate(layoutInflater)
    setContentView(activityMainBinding.root)

    val settingsViewModel: SettingsViewModel by viewModels()
    activityMainBinding.loadSettingsBtn.setOnClickListener {
        lifecycleScope.launch {
            settingsViewModel.getSettings().collect {
                Log.w("A8452", "MainActivity -> ${it.javaClass.name}")
            }
        }
    }
}

示例2 #

private var isCollectFired = false

override fun onCreate(savedInstanceState: Bundle?) {
    super.onCreate(savedInstanceState)
    val activityMainBinding: ActivityMainBinding = ActivityMainBinding.inflate(layoutInflater)
    setContentView(activityMainBinding.root)

    val settingsViewModel: SettingsViewModel by viewModels()
    activityMainBinding.loadSettingsBtn.setOnClickListener {
        lifecycleScope.launch {
            if (!isCollectFired) {
                isCollectFired = true
                settingsViewModel.getSettings().collect {
                    Log.w("A8452", "MainActivity -> ${it.javaClass.name}")
                }
            } else
                settingsViewModel.getSettings()
        }
    }
}

SettingsViewModel.kt #

两个示例均使用该类中的相同代码：

class SettingsViewModel : ViewModel() {

    private val mutableStateFlow: MutableStateFlow<ConnectionStatus> = MutableStateFlow(ConnectionStatus.Loading)
    private lateinit var settingsRepository: SettingsRepository

    fun getSettings(): StateFlow<ConnectionStatus> {
        if (!::settingsRepository.isInitialized)
            settingsRepository = SettingsRepository()

        viewModelScope.launch {
            settingsRepository.getSettings().collect {
                Log.w("A8452", "SettingsViewModel -> ${it.javaClass.name}")
                mutableStateFlow.emit(it)
            }
        }

        return mutableStateFlow
    }

}

问题说明 #

• 示例1中，多次点击按钮会触发多次collect，每次点击后collect的执行次数像滚雪球一样增多。
• 示例2中用isCollectFired布尔变量控制只执行一次collect，解决了问题，但不确定是否为最优方案。
• 疑问：SettingsViewModel的getSettings方法里也有未加检查的collect，但多次点击按钮时不会出现滚雪球式的增多，原因是什么？

原因解析 #

核心区别在于仓库实例的初始化逻辑和流的订阅机制：

1. SettingsViewModel里的settingsRepository使用lateinit声明，且只有在未初始化时才会创建实例：if (!::settingsRepository.isInitialized)。也就是说，不管点击多少次按钮，settingsRepository只会被初始化一次。
2. 假设settingsRepository.getSettings()返回的是单次请求的冷流，第一次collect就会完成整个流的收集；如果是热流，也只会有一个settingsRepository实例对应的流在被收集，后续调用getSettings不会重新创建仓库实例，自然不会新增collect任务。
3. 而示例1中，每次点击按钮都会启动新协程去订阅同一个StateFlow（settingsViewModel.getSettings()返回的mutableStateFlow）。StateFlow是热流，每个新的collect都会成为一个新订阅者，所以每次点击都会新增一个订阅，导致日志输出次数越来越多。

示例2的优化建议 #

示例2的方案能解决问题，但更优雅的方式是在Activity生命周期内仅订阅一次，同时避免内存泄漏：

override fun onCreate(savedInstanceState: Bundle?) {
    super.onCreate(savedInstanceState)
    val activityMainBinding: ActivityMainBinding = ActivityMainBinding.inflate(layoutInflater)
    setContentView(activityMainBinding.root)

    val settingsViewModel: SettingsViewModel by viewModels()
    // 在STARTED状态下订阅，生命周期销毁时自动取消
    lifecycleScope.launch {
        repeatOnLifecycle(Lifecycle.State.STARTED) {
            settingsViewModel.getSettings().collect {
                Log.w("A8452", "MainActivity -> ${it.javaClass.name}")
            }
        }
    }

    activityMainBinding.loadSettingsBtn.setOnClickListener {
        // 点击仅触发数据加载，无需处理订阅
        settingsViewModel.getSettings()
    }
}

内容的提问来源于stack exchange，提问作者Ahlam saleh