Jetpack Compose返回时Screen A重构原因及避免方法咨询

Jetpack Compose Navigation的默认行为是：当从Screen A导航到Screen B时，Screen A对应的Composable会被从Composition中移除（执行Dispose逻辑）。从Screen B返回时，导航组件会重新调用Screen A的Composable函数，导致整个页面“重构”——本质是Composable被重新实例化，之前的临时状态和非持久化数据会丢失。

这是因为导航组件默认仅在回退栈中保存目的地的路由信息和参数，不保留前一个目的地的Composition状态。而这里的“重构”是整个Composable的重新创建，区别于Compose基于状态变化的普通重组。

以下按场景给出具体实现方式：

1. 用rememberSaveable保存简单状态 #

如果Screen A只有简单的可序列化状态（比如输入框文本、选中状态、计数器值），用rememberSaveable替代普通的remember。rememberSaveable会将状态持久化到Android的Bundle中，即使Composable被Dispose再重建，状态也能自动恢复。

示例代码：

@Composable
fun ScreenA() {
    // 用rememberSaveable保存输入框状态
    val inputText = rememberSaveable { mutableStateOf("") }
    
    TextField(
        value = inputText.value,
        onValueChange = { inputText.value = it },
        label = { Text("输入内容") }
    )
    // 其他UI元素
}

2. 使用ViewModel管理复杂状态 #

如果Screen A涉及业务逻辑或复杂状态（比如网络请求数据、列表状态），推荐将状态和逻辑放到ViewModel中。ViewModel的生命周期独立于Composable的Composition，即使页面被重构，ViewModel实例会被保留，状态不会丢失。

示例代码：

// 定义ViewModel
class ScreenAViewModel : ViewModel() {
    // 保存列表数据
    val itemList = mutableStateListOf<String>()
    
    // 模拟网络请求加载数据
    fun loadData() {
        viewModelScope.launch {
            // 模拟异步请求
            delay(1000)
            itemList.addAll(listOf("Item 1", "Item 2", "Item 3"))
        }
    }
}

// Screen A的Composable
@Composable
fun ScreenA(viewModel: ScreenAViewModel = viewModel()) {
    // 首次进入时加载数据
    LaunchedEffect(Unit) {
        viewModel.loadData()
    }
    
    LazyColumn {
        items(viewModel.itemList) { item ->
            Text(text = item, modifier = Modifier.padding(16.dp))
        }
    }
}

这里即使返回Screen A时Composable重构，ViewModel还是原来的实例，itemList的数据会被保留，不需要重新加载。

3. 配置导航目的地的状态恢复 #

在导航图中给Screen A的目的地设置restoreState = true，让导航组件自动恢复该目的地的状态。这个配置需要配合rememberSaveable或ViewModel一起使用，才能完整保留状态。

示例代码：

NavHost(navController = navController, startDestination = "screen_a") {
    composable(
        route = "screen_a",
        // 启用状态恢复
        restoreState = true
    ) {
        ScreenA()
    }
    composable(route = "screen_b") {
        ScreenB(navController = navController)
    }
}

4. 用SavedStateHandle保存跨页面状态 #

如果需要在Screen A和Screen B之间传递或保存状态，可以使用导航回退栈条目的SavedStateHandle。它属于导航组件的状态容器，即使页面被Dispose，状态也会被保留在回退栈中。

示例代码：

// 在Screen A中获取SavedStateHandle
@Composable
fun ScreenA(navController: NavHostController) {
    val savedStateHandle = remember {
        navController.currentBackStackEntry!!.savedStateHandle
    }
    // 从SavedStateHandle获取状态，默认值为0
    val count = savedStateHandle.getStateFlow("count", 0).collectAsState()
    
    Button(onClick = {
        navController.navigate("screen_b")
    }) {
        Text("跳转到Screen B，当前计数：${count.value}")
    }
}

// 在Screen B中更新SavedStateHandle的状态
@Composable
fun ScreenB(navController: NavHostController) {
    val savedStateHandle = remember {
        navController.previousBackStackEntry!!.savedStateHandle
    }
    
    Button(onClick = {
        // 更新Screen A的计数状态
        savedStateHandle["count"] = savedStateHandle.get<Int>("count") ?: 0 + 1
        navController.popBackStack()
    }) {
        Text("返回并增加计数")
    }
}

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