SwiftUI中ObservableObject如何监听响应Core Data数据变更

在SwiftUI开发中，当你不使用@FetchRequest，而是在遵循ObservableObject的数据源类中手动执行NSFetchRequest拉取Core Data数据、封装为自定义结构（比如供Charts组件使用的统计字典）时，会出现初始加载正常、但Core Data数据变更后UI无法自动刷新的问题。
初始问题代码如下：
数据源类：

class CJChartsDataDataSource: ObservableObject {

    @Published var cjCountItemsForDates = [Date: NSNumber]()        
    
    init(context: NSManagedObjectContext) {
        
        if let countForDates = CJDataItem.countForItemsByDate(with: context, with: Date(), andNumberOfDates: 7) {
            cjCountItemsForDates = countForDates
                            
        }
    }
}

对应视图代码：

struct CJReportsViewController: View {
    
    @Environment(\.managedObjectContext) var managedObjectContext
    
    @ObservedObject var dataSource: CJChartsDataDataSource
    
    var body: some View {
        /// ...
    }
}

核心逻辑是监听NSManagedObjectContext的变更通知，当检测到CJDataItem实体发生增删改时，重新执行查询逻辑更新@Published属性，即可自动驱动SwiftUI视图刷新。
具体实现步骤：

• 在数据源类中持有NSManagedObjectContext的引用，同时持有通知监听的取消令牌，避免内存泄漏
• 注册Core Data上下文的保存完成通知，过滤出和CJDataItem相关的变更事件，避免无关数据变更触发无效刷新
• 收到有效变更事件后，在主线程重新执行数据查询，更新对外暴露的统计数据属性
• 数据源类销毁时自动取消通知监听

修改后的可直接使用的数据源代码：

import CoreData
import Combine

class CJChartsDataDataSource: ObservableObject {
    @Published var cjCountItemsForDates = [Date: NSNumber]()
    
    private let context: NSManagedObjectContext
    private var changeObserver: AnyCancellable?
    
    init(context: NSManagedObjectContext) {
        self.context = context
        // 首次进入加载数据
        reloadChartData()
        
        // 监听上下文数据保存事件
        changeObserver = NotificationCenter.default.publisher(
            for: NSManagedObjectContext.didSaveObjectsNotification,
            object: context
        )
        .receive(on: DispatchQueue.main)
        .sink { [weak self] notification in
            guard let self = self else { return }
            // 校验本次变更是否涉及CJDataItem实体
            let hasItemChange = self.checkIfCJItemChanged(in: notification)
            guard hasItemChange else { return }
            // 变更涉及目标实体，重新拉取统计数据
            self.reloadChartData()
        }
    }
    
    /// 统一的数据重载方法，首次加载和后续刷新复用
    private func reloadChartData() {
        guard let countResult = CJDataItem.countForItemsByDate(
            with: context,
            with: Date(),
            andNumberOfDates: 7
        ) else { return }
        cjCountItemsForDates = countResult
    }
    
    /// 检查通知中是否包含CJDataItem的增删改记录
    private func checkIfCJItemChanged(in notification: Notification) -> Bool {
        let checkEntity: (Any?) -> Bool = { objects in
            guard let managedObjects = objects as? Set<NSManagedObject> else { return false }
            return managedObjects.contains { $0.entity.name == "CJDataItem" }
        }
        let userInfo = notification.userInfo
        return checkEntity(userInfo?[NSInsertedObjectsKey])
            || checkEntity(userInfo?[NSUpdatedObjectsKey])
            || checkEntity(userInfo?[NSDeletedObjectsKey])
    }
    
    deinit {
        changeObserver?.cancel()
    }
}

注意事项 #

• 如果项目中使用后台上下文写入数据，需要将监听对象调整为NSPersistentStoreCoordinator，同时确保后台上下文保存后将变更合并到视图层使用的viewContext
• 所有@Published属性的更新必须放在主线程执行，避免触发SwiftUI的线程断言警告
• 增加实体类型校验逻辑可以过滤大量无关刷新，降低列表滚动、其他实体数据变更时的性能损耗

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