Jetpack Compose中基于函数参数更新可变状态的实现与性能疑问

针对你提出的三个疑问，逐一分析并给出优化方案：

1. 原实现的性能负面影响 #

你当前直接在List Composable函数体中调用expandedState.keys.retainAll(items)，存在两个明显的性能问题：

• 无意义的重复执行：每次Composable重组（哪怕是无关状态触发的重组）都会执行该操作，若items数量较大，retainAll的遍历过滤会带来不必要的开销。
• 潜在的循环重组风险：在重组过程中直接修改expandedState，会触发额外的重组，极端情况下可能陷入"修改状态→触发重组→再次修改状态"的循环。

2. 是否需要在副作用中执行状态清理？ #

必须在副作用中执行。Composable函数体应保持纯函数特性，避免在重组期间直接修改状态。正确的做法是用LaunchedEffect监听items的变化，仅当items真正更新时才清理过期的展开状态：

@Composable
private fun List(items: List<String>) {
    // 用StateSet替代StateMap，更贴合仅存ID的需求
    val expandedState = remember { mutableStateSetOf<String>() }
    
    // 仅当items引用变化时，执行过期状态清理
    LaunchedEffect(items) {
        expandedState.retainAll(items.toSet())
    }

    ListStateless(
        onExpand = { id -> expandedState.add(id) },
        onCollapse = { id -> expandedState.remove(id) },
        items = items,
        expandedItems = expandedState
    )
}

3. 更优实现方式 #

除了副作用优化，还有以下几个改进方向：

• 封装状态逻辑到自定义remember函数：如果多个地方需要类似的展开状态管理，可将状态创建和清理逻辑封装，提高复用性：

  @Composable
  fun rememberExpandedState(items: List<String>): MutableStateSet<String> {
      val expandedState = remember { mutableStateSetOf<String>() }
      LaunchedEffect(items) {
          expandedState.retainAll(items.toSet())
      }
      return expandedState
  }
  
  // 使用示例
  @Composable
  private fun List(items: List<String>) {
      val expandedState = rememberExpandedState(items)
      ListStateless(
          onExpand = { id -> expandedState.add(id) },
          onCollapse = { id -> expandedState.remove(id) },
          items = items,
          expandedItems = expandedState
      )
  }
  

• 简化LazyItem的Key设置：若item本身是唯一标识，直接用items[index]作为key即可，确保LazyColumn的复用逻辑更高效：

  LazyColumn {
      items(
          items = items,
          key = { item -> item }
      ) { item ->
          val isExpanded = expandedItems.contains(item)
          Text(
              text = "$item $isExpanded",
              modifier = Modifier.clickable {
                  if (!isExpanded) onExpand(item) else onCollapse(item)
              }
          )
      }
  }
  

另外补充：你关于"展开状态属于视图状态，无需放到ViewModel"的判断是正确的，保持在Composable层级能避免ViewModel被无关的视图细节污染。

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