SwiftUI Table切换数据源时选中状态无法恢复的优化方案咨询

不用依赖Task.sleep这种临时方案，核心是利用SwiftUI的状态管理和视图标识机制，确保选中状态和数据源切换同步：

关键思路 #

1. 用唯一ID保存选中状态：别用数组索引，切换数据源后索引对应的条目会变化。给每个数据项添加唯一标识（比如遵循Identifiable协议），保存选中项的ID而非位置。
2. 监听数据源切换事件：切换数据源时，先保存当前选中状态，再加载新数据源，最后恢复对应数据源的历史选中ID。
3. 强制Table视图重建：给Table添加.id()修饰符，绑定到当前数据源的标识，让SwiftUI在数据源切换时重新创建Table，确保新的选中状态被正确识别。

代码示例 #

import SwiftUI

// 数据模型必须有唯一ID，遵循Identifiable
struct DataItem: Identifiable {
    let id: String
    let title: String
}

struct TableSelectionSyncView: View {
    // 当前展示的数据源
    @State private var currentItems: [DataItem] = []
    // 当前选中项的ID
    @State private var selectedItemId: String?
    // 保存不同数据源的历史选中状态（key:数据源标识，value:选中ID）
    private var savedSelections: [String: String] = [:]
    
    // 数据源选项
    private let sourceOptions = ["用户列表", "订单列表", "商品列表"]
    @State private var selectedSource = "用户列表"
    
    var body: some View {
        VStack(spacing: 16) {
            // 数据源切换器
            Picker("切换数据源", selection: $selectedSource) {
                ForEach(sourceOptions, id: \.self) { source in
                    Text(source)
                }
            }
            .pickerStyle(.segmented)
            .onChange(of: selectedSource) { newSource in
                // 1. 保存当前数据源的选中状态
                if let selectedId = selectedItemId {
                    savedSelections[selectedSource] = selectedId
                }
                // 2. 加载新数据源
                currentItems = loadItems(for: newSource)
                // 3. 恢复新数据源的历史选中状态
                selectedItemId = savedSelections[newSource]
            }
            
            // 带选中状态的Table
            Table(currentItems, selection: $selectedItemId) {
                TableColumn("标题", value: \.title)
            }
            // 核心：用数据源标识作为Table的ID，切换时重建视图
            .id(selectedSource)
            .frame(minHeight: 300)
        }
        .padding()
        .onAppear {
            // 初始化加载默认数据源
            currentItems = loadItems(for: selectedSource)
        }
    }
    
    // 模拟加载不同数据源的逻辑
    private func loadItems(for source: String) -> [DataItem] {
        switch source {
        case "用户列表":
            return [
                DataItem(id: "user_1", title: "张三"),
                DataItem(id: "user_2", title: "李四"),
                DataItem(id: "user_3", title: "王五")
            ]
        case "订单列表":
            return [
                DataItem(id: "order_1", title: "20240501001"),
                DataItem(id: "order_2", title: "20240501002")
            ]
        case "商品列表":
            return [
                DataItem(id: "goods_1", title: "iPhone 15"),
                DataItem(id: "goods_2", title: "MacBook Pro"),
                DataItem(id: "goods_3", title: "AirPods Pro")
            ]
        default:
            return []
        }
    }
}

为什么这个方案有效 #

• 唯一ID保障选中准确性：不管数据源怎么切换，选中的都是对应ID的条目，不会因为索引错位导致错误。
• .id()修饰符触发视图重建：当数据源切换时，Table的id值变化，SwiftUI会销毁旧的Table视图并创建新的，新视图会直接读取最新的selectedItemId绑定值，避免旧视图状态的残留。
• 同步状态更新：所有操作都在主线程的状态更新周期内完成，不需要等待延迟，UI会和状态同步刷新。

如果遇到数据源异步加载的情况，只需在异步加载完成后再设置selectedItemId即可，确保回调在主线程执行就行，同样不需要依赖延迟。

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