SwiftUI中NSManagedObjectContext注入Environment遇异常求助

你猜的完全没错——问题根源就是loadPersistentStores是异步执行的，但你在它完成前就把viewContext同步注入到了SwiftUI环境里。当应用启动时，CategoriesView立刻通过@FetchRequest访问上下文，这时候持久化存储还没加载完毕，就会弹出警告；而模态视图触发时，上下文的状态更不稳定，直接导致崩溃。

下面是两种可行的解决方案，确保只有在持久化存储完全加载完成后，才把上下文注入到SwiftUI环境中：

方案一：用Combine监听存储加载完成 #

我们可以给PersistentContainer添加一个Combine发布者，用来通知外界存储加载完成的事件：

import Combine

class PersistentContainer: NSPersistentContainer {
    static let shared = PersistentContainer()
    private let loadedSubject = PassthroughSubject<Void, Error>()
    var loaded: AnyPublisher<Void, Error> {
        loadedSubject.eraseToAnyPublisher()
    }

    private override init(name: String) {
        super.init(name: name)
        viewContext.mergePolicy = NSMergePolicy(merge: .mergeByPropertyStoreTrumpMergePolicyType)
        viewContext.automaticallyMergesChangesFromParent = true
        loadPersistentStores { [weak self] description, error in
            guard let self = self else { return }
            if let error = error {
                self.loadedSubject.send(completion: .failure(error))
                fatalError("Unable to load persistent stores: \(error)")
            } else {
                self.loadedSubject.send(())
                self.loadedSubject.send(completion: .finished)
            }
        }
    }
}

然后修改SceneDelegate，等待发布者发出完成信号后再初始化SwiftUI视图：

extension SceneDelegate: UIWindowSceneDelegate {
    func scene(_ scene: UIScene, willConnectTo session: UISceneSession, options connectionOptions: UIScene.ConnectionOptions) {
        guard let windowScene = scene as? UIWindowScene else { return }
        let window = UIWindow(windowScene: windowScene)
        self.window = window
        
        // 等待存储加载完成后再设置根视图
        PersistentContainer.shared.loaded
            .receive(on: DispatchQueue.main)
            .sink(receiveCompletion: { completion in
                if case .failure(let error) = completion {
                    print("加载持久化存储失败: \(error)")
                }
            }, receiveValue: { [weak self] in
                guard let self = self else { return }
                let context = PersistentContainer.shared.viewContext
                let rootView = TabBarView().environment(\.managedObjectContext, context)
                window.rootViewController = UIHostingController(rootView: rootView)
                window.makeKeyAndVisible()
            })
            .store(in: &cancellables)
    }
    
    // 保存Combine订阅，防止被提前释放
    private var cancellables = Set<AnyCancellable>()
}

方案二：用闭包回调实现（无需Combine） #

如果你不想引入Combine，可以用简单的闭包回调来实现同样的效果：

class PersistentContainer: NSPersistentContainer {
    static let shared = PersistentContainer()
    private var onLoadedCallback: (() -> Void)?

    private override init(name: String) {
        super.init(name: name)
        viewContext.mergePolicy = NSMergePolicy(merge: .mergeByPropertyStoreTrumpMergePolicyType)
        viewContext.automaticallyMergesChangesFromParent = true
        loadPersistentStores { description, error in
            if let error = error {
                fatalError("Unable to load persistent stores: \(error)")
            }
            // 存储加载完成后触发回调
            self.onLoadedCallback?()
        }
    }
    
    func whenLoaded(completion: @escaping () -> Void) {
        // 如果存储已经加载完成，直接执行回调
        if !persistentStoreCoordinator.persistentStores.isEmpty {
            completion()
        } else {
            onLoadedCallback = completion
        }
    }
}

对应的SceneDelegate修改：

extension SceneDelegate: UIWindowSceneDelegate {
    func scene(_ scene: UIScene, willConnectTo session: UISceneSession, options connectionOptions: UIScene.ConnectionOptions) {
        guard let windowScene = scene as? UIWindowScene else { return }
        let window = UIWindow(windowScene: windowScene)
        self.window = window
        
        // 等待存储加载完成
        PersistentContainer.shared.whenLoaded { [weak self] in
            guard let self = self else { return }
            let context = PersistentContainer.shared.viewContext
            let rootView = TabBarView().environment(\.managedObjectContext, context)
            window.rootViewController = UIHostingController(rootView: rootView)
            window.makeKeyAndVisible()
        }
    }
}

额外注意事项 #

• 永远不要在持久化存储加载完成前访问viewContext，此时它还没有关联到持久化存储协调器，必然会出问题
• 如果你使用的是iOS 14+的@main App结构，也可以用类似的思路，在App初始化时等待存储加载完成后再返回根视图
• 模态视图崩溃的原因和启动时的警告一致：访问了未绑定到存储的上下文，所以确保所有视图都在存储加载完成后再展示即可

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