如何优化SwiftData @Query结果按月分组的SwiftUI List性能

问题根源 #

原方案在内存中对所有Item实例做分组和排序，当数据量达数千条时，会在主线程执行大量计算，同时一次性加载所有数据到内存，导致初始加载卡顿甚至无响应。

优化方案 #

1. 先通过SwiftData查询排序，再在List中动态分组（推荐） #

利用SwiftUI的Section结合已排序的数据，实现懒加载式分组，避免提前在内存中构建完整分组字典：

首先，修改@Query让数据按日期倒序排列：

struct ContentView: View {
    // 先按date倒序查询，确保同月份条目连续
    @Query(sort: \Item.date, order: .reverse) var items: [Item]

    var body: some View {
        List {
            // 遍历懒加载的分组序列
            ForEach(Array(items.groupedByMonth()), id: \.key) { monthKey, itemsInMonth in
                Section(header: Text(monthKey)) {
                    ForEach(itemsInMonth) { item in
                        Text(item.name ?? "")
                    }
                }
            }
        }
    }
}

// 给[Item]扩展懒加载分组方法
extension Array where Element == Item {
    func groupedByMonth() -> LazyMapSequence<[String], (key: String, value: [Element])> {
        let calendar = Calendar.current
        // 用lazy分组，避免一次性处理所有数据
        let grouped = self.lazy.grouping { item -> String in
            guard let date = item.date else { return "无日期" }
            return DateFormatter.monthFormatter.string(from: date)
        }
        // 对分组key按日期倒序排序
        let sortedKeys = grouped.keys.sorted { key1, key2 in
            guard let date1 = DateFormatter.monthFormatter.date(from: key1),
                  let date2 = DateFormatter.monthFormatter.date(from: key2) else {
                return false
            }
            return date1 > date2
        }
        // 返回懒加载的键值对序列
        return sortedKeys.lazy.map { ($0, grouped[$0]!) }
    }
}

// 全局缓存日期格式化器，避免重复创建
extension DateFormatter {
    static let monthFormatter: DateFormatter = {
        let formatter = DateFormatter()
        formatter.dateFormat = "yyyy年MM月"
        formatter.locale = Locale.current
        return formatter
    }()
}

核心是用lazy修饰分组和排序操作，确保只有当List需要渲染对应Section时才处理该分组数据，减少初始加载的计算量；同时@Query提前按日期排序，保证同月份条目连续，提升分组效率。

2. 利用SwiftData的FetchDescriptor做数据库层面分组（进阶） #

如果需要更底层优化，可让数据库完成分组和排序，减少内存加载的数据量。这种方式适合先获取分组key，再按需加载对应条目：

// 示例：获取所有存在的月份分组
func fetchMonthGroups() async throws -> [String] {
    let descriptor = FetchDescriptor<Item>(
        predicate: #Predicate { $0.date != nil },
        sortBy: [SortDescriptor(\Item.date, order: .reverse)],
        groupingBy: \Item.monthKey
    )
    let results = try modelContext.fetch(descriptor)
    return results.map { $0.key }
}

// 给Item添加分组用的计算属性
extension Item {
    var monthKey: String {
        guard let date = date else { return "无日期" }
        return DateFormatter.monthFormatter.string(from: date)
    }
}

3. 辅助优化点 #

• 避免强制解包：原代码中element.date!容易引发崩溃，建议添加空值处理，将无日期条目归到单独分组。
• 复用列表行：确保ForEach使用SwiftData模型默认的id属性，让SwiftUI可以复用列表行，提升滚动性能。

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