SwiftUI接收NSManagedObjectContextDidSaveObjectIDs通知时线程警告处理方案

问题 #

在SwiftData项目中，因加载大量对象会导致旧设备性能问题，放弃使用@Query自动更新视图，转而监听NSManagedObjectContextDidSaveObjectIDs通知，待数据处理完成后通过FetchDescriptor手动更新数据集以避免视图冻结。该通知由处理ModelContext的@ModelActor发送。

当前代码实现：

.onReceive(NotificationCenter.default.publisher(for: Notification.Name.NSManagedObjectContextDidSaveObjectIDs)) { notification in
    //print("Receive, on thread: \(Thread.current) is Main \(Thread.isMainThread)")
    Task {
        await MainActor.run {
            refreshData()
        }
    }
}

func refreshData() {
    var isProcessing = false
    Task {
        isProcessing = await DataActor.shared.isProcessingData
    }
    if !waiting && isProcessing {
        waiting = true
        Task {
            try await Task.sleep(nanoseconds: 500_000_000)
            waiting = false
            filterData()
        }
    }
}

func filterData(){
    let descriptor = FetchDescriptor<Event>(
        sortBy: [
            .init(\.startTime)
        ]
    )
    let events = try! modelContext.fetch(descriptor)
    data = filterManager.sortedEventsByDay(events: filterManager.filteredEvents(events: events))
}

执行时每次更新都会收到警告：

不允许从后台线程发布变更；请确保在模型更新时通过receive(on:)等操作符从主线程发布值。

由于SwiftUI的.onReceive无法直接设置receive(on:)指定主线程接收通知，询问最佳处理方式或更优实现思路。

解决方案 #

1. 修正通知接收的线程问题 #

问题根源在于通知可能在后台线程发布，.onReceive默认在通知发布线程执行闭包。可以给通知发布者添加receive(on:)操作符，强制切换到主线程接收：

.onReceive(NotificationCenter.default.publisher(for: .NSManagedObjectContextDidSaveObjectIDs)
    .receive(on: DispatchQueue.main)) { notification in
    refreshData()
}

这样无需在闭包内嵌套Task和MainActor.run，直接在主线程执行逻辑，彻底消除线程警告。

2. 修复refreshData的逻辑漏洞 #

原代码中异步赋值isProcessing后直接同步判断，会导致isProcessing永远为初始的false，逻辑完全失效。修正为异步等待后再执行判断：

func refreshData() {
    guard !waiting else { return }
    Task {
        let isProcessing = await DataActor.shared.isProcessingData
        if isProcessing {
            waiting = true
            try await Task.sleep(nanoseconds: 500_000_000)
            waiting = false
            await MainActor.run {
                filterData()
            }
        }
    }
}

同时filterData中使用try!存在崩溃风险，建议替换为do-catch处理错误：

func filterData(){
    let descriptor = FetchDescriptor<Event>(
        sortBy: [.init(\.startTime)]
    )
    do {
        let events = try modelContext.fetch(descriptor)
        data = filterManager.sortedEventsByDay(events: filterManager.filteredEvents(events: events))
    } catch {
        print("Fetch events failed: \(error)")
        // 可添加用户提示等错误处理逻辑
    }
}

3. 更优实现：用ModelActor封装异步查询 #

既然使用@ModelActor处理数据，可直接让其负责查询和过滤逻辑，再将结果同步到主线程更新视图，无需依赖通知：

// 定义ModelActor
@ModelActor
actor DataActor {
    static let shared = DataActor()
    
    func fetchAndFilterEvents() async throws -> [[Event]] {
        let descriptor = FetchDescriptor<Event>(sortBy: [.init(\.startTime)])
        let events = try modelContext.fetch(descriptor)
        let filtered = filterManager.filteredEvents(events: events)
        return filterManager.sortedEventsByDay(events: filtered)
    }
}

// View中调用刷新逻辑
func refreshData() {
    guard !waiting else { return }
    waiting = true
    Task {
        do {
            let result = try await DataActor.shared.fetchAndFilterEvents()
            await MainActor.run {
                data = result
                waiting = false
            }
        } catch {
            print("Fetch failed: \(error)")
            await MainActor.run {
                waiting = false
            }
        }
    }
}

这种方式符合SwiftData的Actor安全模型，所有数据操作在ModelActor上下文执行，线程问题更可控，逻辑也更清晰。可在ModelActor完成数据修改后主动调用refreshData，替代通知监听。

内容的提问来源于stack exchange，提问作者Luuk D. Jansen