Swift4中基于运行模式切换两个Core Data存储的技术实现问询

嘿，这个需求我很熟悉，刚好可以通过修改NSPersistentContainer的存储描述来实现，完全能满足你「首次启动切换、基于运行模式选存储」的要求。下面是具体的实现步骤和代码：

核心思路 #

NSPersistentContainer默认会根据你传入的name生成对应的SQLite文件，但我们可以通过自定义NSPersistentStoreDescription，替换默认的存储URL，来指定要使用的SQLite文件。同时结合UserDefaults可以实现「仅首次启动切换」的逻辑。

完整代码实现 #

lazy var persistentContainer: NSPersistentContainer = {
    let container = NSPersistentContainer(name: "myApp")
    
    // 1. 检查是否是首次启动（确保只切换一次）
    let hasInitializedStore = UserDefaults.standard.bool(forKey: "HasInitializedPersistentStore")
    var storeName = "myApp.sqlite" // 默认存储名
    
    if !hasInitializedStore {
        // 2. 根据运行模式选择对应的存储文件
        let mode = getCurrentRunningMode() // 替换成你获取运行模式的逻辑（比如启动参数、环境变量）
        storeName = mode == 1 ? "MyStore1.sqlite" : "MyStore2.sqlite"
        
        // 标记为已初始化，后续启动不再切换
        UserDefaults.standard.set(true, forKey: "HasInitializedPersistentStore")
        UserDefaults.standard.synchronize()
    }
    
    // 3. 自定义存储描述，指向目标SQLite文件
    guard let defaultStoreURL = container.persistentStoreDescriptions.first?.url else {
        fatalError("Failed to retrieve default persistent store URL")
    }
    let targetStoreURL = defaultStoreURL.deletingLastPathComponent().appendingPathComponent(storeName)
    
    let customStoreDescription = NSPersistentStoreDescription(url: targetStoreURL)
    // 复制默认描述的所有配置（保证和原配置一致：比如存储类型、迁移选项等）
    customStoreDescription.type = NSSQLiteStoreType
    customStoreDescription.options = container.persistentStoreDescriptions.first?.options ?? [:]
    customStoreDescription.shouldAddStoreAsynchronously = true
    
    // 替换容器的存储描述数组
    container.persistentStoreDescriptions = [customStoreDescription]
    
    // 4. 加载持久化存储
    container.loadPersistentStores(completionHandler: { (storeDescription, error) in
        if let error = error as NSError? {
            fatalError("Unresolved error \(error), \(error.userInfo)")
        }
        // 可选：打印当前使用的存储路径，验证是否正确
        print("Active persistent store path: \(storeDescription.url?.absoluteString ?? "Unknown")")
    })
    
    return container
}()

// 示例：获取运行模式的方法，根据你的实际场景修改
func getCurrentRunningMode() -> Int {
    // 比如从启动参数获取：
    if CommandLine.arguments.contains("-Mode1") {
        return 1
    } else {
        return 2
    }
    // 或者从环境变量、配置文件读取，按需调整
}

关键细节说明 #

• 仅首次启动切换：通过UserDefaults的标记，确保只有第一次启动应用时才会根据模式选择存储，后续启动会沿用首次选定的文件，避免每次启动都切换。
• 保持配置一致性：复制默认存储描述的options和type，确保自定义存储的配置和原容器默认配置一致（比如自动迁移、只读模式等）。
• 模型一致性：要保证MyStore1.sqlite和MyStore2.sqlite对应的Core Data模型完全一致，否则加载存储时会抛出模型不匹配的错误。如果模型有差异，需要额外添加迁移逻辑。

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