Compose中如何在导航图内外共享同一ViewModel实例？

我的Compose应用顶级可组合项结构如下：

ModalBottomSheetLayout(/*...*/) {
    Scaffold(
        topBar = {
            when (currentScreen) {
                /*...*/
            }
        },
        content = {
            AppNavigation(navController)
        },
        bottomBar = {
            // Bottom navigation
        }
    )
}

由于同时使用BottomSheet和BottomNavigation，无法将BottomSheet的处理委托给AppNavigation中的页面（否则BottomSheet会显示在BottomNavigation上方，违反Material Design规范）。页面顶部栏也与页面本身分离，需要让TopBar、BottomSheet和对应页面共享同一个ViewModel。

但该ViewModel不能放在NavHost作用域内（因为TopBar和BottomSheet在NavHost外部），若放在Activity的ViewModelStore中，单Activity模式下它永远不会被清空，导致ViewModel变成单例，引发内存泄漏和状态更新异常问题。我期望用户离开绑定页面时，共享ViewModel能被销毁，但目前似乎无法实现。

疑问 #

1. 是否可以在不更改UI结构的情况下解决这个问题？
2. 这个问题是否需要解决？
3. 如果修复了状态相关Bug，单例ViewModel是否可行？

一、不修改UI结构的可行方案 #

1. 绑定ViewModel到当前页面的NavBackStackEntry #

这是最可靠的方案，利用Jetpack Navigation的NavBackStackEntry管理ViewModel生命周期：

• 在顶级可组合项中监听navController.currentBackStackEntryFlow，获取当前活跃的页面BackStackEntry
• 在TopBar、BottomSheet中，通过该BackStackEntry获取ViewModel，ViewModel的生命周期会和页面的BackStackEntry绑定，页面出栈时自动销毁

示例代码：

// 顶级可组合项中
val currentBackStackEntry by navController.currentBackStackEntryFlow.collectAsStateWithLifecycle()

// TopBar中
currentBackStackEntry?.let { entry ->
    val viewModel = viewModel<YourPageViewModel>(entry)
    // 使用viewModel更新TopBar状态
}

// ModalBottomSheetLayout中
currentBackStackEntry?.let { entry ->
    val viewModel = viewModel<YourPageViewModel>(entry)
    // 根据viewModel状态控制BottomSheet显示/隐藏及内容
}

如果使用Hilt，替换为hiltViewModel<YourPageViewModel>(entry)即可。这种方案完全不需要修改现有UI结构，完美匹配需求。

2. 自定义ViewModelStore（不推荐） #

可以手动创建ViewModelStore，在页面切换时跟踪当前页面，当页面离开时清空对应Store中的ViewModel。但这种方式需要自己处理生命周期回调、页面切换监听，容易出现边界错误，维护成本高，不如第一种方案简洁可靠。

二、这个问题必须解决 #

• 内存泄漏：Activity级别的ViewModel会长期持有页面相关的资源、回调或数据引用，即使页面已销毁，这些引用无法被GC回收，长期运行会导致内存占用过高，甚至触发OOM。
• 状态污染：单例ViewModel的状态会在页面间共享，比如用户返回之前访问过的页面时，状态未重置，会显示旧数据或异常UI，严重影响用户体验，属于必须修复的问题。

三、单例ViewModel不可行 #

即使手动重置ViewModel的状态，也无法解决内存泄漏问题——ViewModel默认持有Application或Activity上下文引用，长期存在会导致内存泄漏。此外，手动重置状态容易遗漏边界场景，增加代码复杂度和维护成本，不符合Jetpack组件的设计原则。

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