如何高效获取SwiftData对象，避免内存与主线程性能问题

开发音频播放器时，使用SwiftData的@Query获取Song模型列表，发现只要UI发生变化（比如弹出sheet），SwiftData就会多次全量查询所有Song对象。小数据量下影响不明显，但数据规模扩大后，主线程会因重复查询出现性能压力。

相关代码 #

App入口代码

struct FrPlayerApp: App {
    @AppStorage("isDarkMode") private var isDarkMode = false
    @StateObject private var timerManager = SleepTimerManager()

    var body: some Scene {
        WindowGroup {
            TabControl()
                .preferredColorScheme(isDarkMode ? .dark: .light)
        }
        .environmentObject(timerManager)
        .modelContainer(
            for: [Song.self, Library.self, Album.self], isAutosaveEnabled: false)
}

LibraryView代码

struct LibraryView: View {

    @Environment(.modelContext) private var context

    @Query(sort: \Song.title, order: .forward) var songs: [Song]

    @State private var isPresentingUser: Song? = nil

    var body: some View {

        ForEach(songs, id: .id) { song in

            VStack {

                HStack(spacing: 5) {

                    HStack {

                        Text(song.title)
                            .frame(height: 40)
                            .lineLimit(2)
                            .multilineTextAlignment(.leading)
                            .font(.subheadline)

                        Spacer()

                    }

                    // ... (其他代码)

                    HStack {

                        Image(systemName: "ellipsis")

                    }
                    .contentShape(Rectangle())
                    .frame(width: 50, height: 50)
                    .onTapGesture {
                        isPresentingUser = song
                    }

                }
                .padding(.horizontal)
                .frame(height: 60)

            }

            Divider()

        }
        .sheet(item: $isPresentingUser) { song in
            librarySongSheet(song: song)
                .presentationDetents([.height(CGFloat(310))])
        }

    }

}

问题触发场景 #

点击歌曲条目里的省略号弹出sheet时，会触发多次全量查询，UI其他变化时也会出现相同问题：

解决思路与方案 #

1. 限制@Query的不必要刷新

默认@Query会在视图层级刷新、上下文变更时重新执行查询。可以通过指定animation参数为nil，减少因动画触发的额外查询：

@Query(sort: \Song.title, order: .forward, animation: nil) var songs: [Song]

2. 分页加载数据

针对大数据量场景，避免全量查询，改用分页加载：

@State private var currentPage = 0
private let pageSize = 50

@Query(sort: \Song.title, order: .forward, limit: pageSize) var songs: [Song]

// 在视图滚动到底部时加载下一页
.onAppear {
    context.fetch(songs, offset: currentPage * pageSize)
}

3. 封装查询逻辑到ViewModel

将查询逻辑抽离到单独的ObservableObject中，避免视图重建时重复触发查询：

class SongLibraryViewModel: ObservableObject {
    @Published var songs: [Song] = []
    private var cancellables = Set<AnyCancellable>()

    init(context: ModelContext) {
        let fetchDescriptor = FetchDescriptor<Song>(sortBy: [SortDescriptor(\.title)])
        context.fetch(fetchDescriptor)
            .receive(on: DispatchQueue.main)
            .assign(to: \.songs, on: self)
            .store(in: &cancellables)
    }
}

// 在LibraryView中使用
struct LibraryView: View {
    @Environment(.modelContext) private var context
    @StateObject private var viewModel = SongLibraryViewModel(context: context)

    var body: some View {
        ForEach(viewModel.songs, id: .id) { song in
            // ... 原有视图代码
        }
    }
}

4. 为Song模型实现Equatable

如果Song未正确实现Equatable，ForEach可能误判元素变化导致视图全量重建，间接触发查询。添加Equatable实现：

@Model
class Song: Equatable {
    var title: String
    // 其他属性

    static func == (lhs: Song, rhs: Song) -> Bool {
        lhs.id == rhs.id // 用模型唯一ID判断相等
    }
}

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