使用async/await批量保存Core Data数据时遇随机错误求助

核心问题分析 #

你的问题主要来自三个方面：

• 上下文线程安全违规：即使使用async/await，Core Data上下文仍不是线程安全的，跨线程访问或错误的上下文调度会引发随机崩溃。
• 内存过载：循环创建上万条Category对象并堆积在同一个上下文里，会导致内存占用激增，触发系统的随机内存警告或崩溃。
• 强制解包风险：代码中多处使用!强制解包，在资源紧张的场景下（比如批量处理）容易出现nil值，导致断点错误。

具体修复方案 #

1. 使用批量插入API替代循环创建对象 #

Core Data提供了NSBatchInsertRequest，专门用于批量插入大量数据，性能远高于循环创建对象，还能避免上下文内存过载。

示例代码：

class func batchAddCategories(_ categoriesData: [(name: String, icon: NSImage?, countries: [Country], persons: [Person]?)]) async throws {
    let container = await (NSApplication.shared.delegate as! AppDelegate).persistentContainer
    
    // 使用后台执行的上下文，避免阻塞主线程
    let context = container.newBackgroundContext()
    context.performAndWait {
        let batchInsert = NSBatchInsertRequest(entityName: "Category", objects: categoriesData.map { data in
            var properties: [String: Any] = [
                "categoryName": data.name,
                "country": NSSet(array: data.countries)
            ]
            
            // 处理可选字段
            if let persons = data.persons {
                properties["person"] = NSSet(array: persons)
            }
            if let icon = data.icon, let tiffData = icon.tiffRepresentation {
                properties["categoryIcon"] = tiffData
            }
            
            // 生成搜索键
            var searchComponents = [data.name, data.countries.map { $0.countryName }.joined(separator: ", ")]
            if let persons = data.persons {
                searchComponents.append(persons.map { $0.personName }.joined(separator: ", "))
            }
            properties["categorySearchKey"] = searchComponents.joined(separator: ", ")
            
            return properties
        })
        
        do {
            try context.execute(batchInsert)
            try context.save()
        } catch {
            print("批量插入失败: \(error)")
            throw error
        }
    }
}

2. 严格遵守上下文线程安全规则 #

• 每个上下文只能在创建它的线程/队列中使用，使用perform或performAndWait确保代码在上下文的正确线程执行。
• 后台上下文不要直接访问UI相关对象（比如NSImage），如果需要处理图片，先在主线程转换为Data再传入后台上下文。

3. 移除强制解包，处理可选值 #

把代码中的!替换为可选绑定或默认值，避免nil导致的崩溃：

// 原代码：
let entity = NSEntityDescription.entity(forEntityName: "Category", in: context)!
// 改为：
guard let entity = NSEntityDescription.entity(forEntityName: "Category", in: context) else {
    print("无法获取Category实体")
    return
}

// 原代码：
CD.categoryIcon = categoryIcon.tiffRepresentation!
// 改为：
if let tiffData = categoryIcon?.tiffRepresentation {
    CD.categoryIcon = tiffData
}

4. 分批次保存，避免一次性写入大量数据 #

如果必须使用循环创建对象，分批次保存上下文，减少内存压力：

class func addCategoriesInBatches(_ categoriesData: [(name: String, icon: NSImage?, countries: [Country], persons: [Person]?)], batchSize: Int = 1000) async throws {
    let container = await (NSApplication.shared.delegate as! AppDelegate).persistentContainer
    let context = container.newBackgroundContext()
    
    for i in stride(from: 0, to: categoriesData.count, by: batchSize) {
        let batch = Array(categoriesData[i..<min(i+batchSize, categoriesData.count)])
        
        try await context.perform {
            for data in batch {
                guard let entity = NSEntityDescription.entity(forEntityName: "Category", in: context) else {
                    throw NSError(domain: "CoreDataError", code: -1, userInfo: [NSLocalizedDescriptionKey: "无法获取Category实体"])
                }
                let cdCategory = Category(entity: entity, insertInto: context)
                cdCategory.categoryName = data.name
                cdCategory.country = NSSet(array: data.countries)
                
                if let persons = data.persons {
                    cdCategory.person = NSSet(array: persons)
                }
                cdCategory.categoryIcon = data.icon?.tiffRepresentation
                
                var searchComponents = [data.name, data.countries.map { $0.countryName }.joined(separator: ", ")]
                if let persons = data.persons {
                    searchComponents.append(persons.map { $0.personName }.joined(separator: ", "))
                }
                cdCategory.categorySearchKey = searchComponents.joined(separator: ", ")
            }
            
            try context.save()
            // 重置上下文，释放内存
            context.reset()
        }
    }
}

5. 修复保存函数的线程问题 #

保存操作必须在上下文的所属线程执行，不要用MainActor.run包裹后台上下文的保存：

func saveCoreData(context: NSManagedObjectContext) async throws {
    try await context.perform {
        if context.hasChanges {
            try context.save()
        }
    }
}

额外排查点 #

• 检查Country和Person对象是否属于同一个上下文：如果这些对象是从主上下文获取的，传入后台上下文会引发线程违规，需要先转为ID再在后台上下文获取，或者确保所有相关对象都在同一个后台上下文创建。
• 启用Core Data的并发调试：在Scheme的Arguments中添加-com.apple.CoreData.ConcurrencyDebug 1，这样线程违规会直接抛出明确错误，方便定位问题。

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