SwiftData+CloudKit中默认与自定义电影类型模型合并方案问询

第一步：重构统一的MovieGenre模型 #

先把枚举和自定义类合并成一个带持久化能力的@Model类，添加唯一约束避免重复，同时标记默认类型：

import SwiftData

@Model
class MovieGenre {
    // 给名称添加唯一约束，防止重复创建同类型
    @Attribute(.unique) var name: String
    // 标记是否为系统默认类型，方便后续区分（比如不允许删除默认类型）
    var isDefault: Bool
    // 关联该类型下的电影数组，支持持久化
    @Relationship(deleteRule: .cascade) var movies: [Movie] = []

    init(name: String, isDefault: Bool = false) {
        self.name = name
        self.isDefault = isDefault
    }

    // 定义系统默认的电影类型集合
    static let defaultGenres: [MovieGenre] = [
        .init(name: "Drama", isDefault: true),
        .init(name: "Teleplay", isDefault: true),
        .init(name: "Action", isDefault: true),
        .init(name: "Horror", isDefault: true),
        .init(name: "Thriller", isDefault: true),
        .init(name: "Musical", isDefault: true),
        .init(name: "Animated", isDefault: true)
    ]
}

// 示例关联的Movie模型
@Model
class Movie {
    var title: String
    @Relationship(inverse: \MovieGenre.movies) var genre: MovieGenre?

    init(title: String) {
        self.title = title
    }
}

方案1：利用UserDefaults标记首次初始化（简单高效） #

通过UserDefaults记录默认类型是否已经注入，只在应用首次启动时执行一次插入操作，避免每次启动都检查数据库：

@main
struct MovieApp: App {
    let container: ModelContainer

    init() {
        do {
            container = try ModelContainer(for: MovieGenre.self, Movie.self)
            let userDefaults = UserDefaults.standard
            
            // 检查是否已经初始化过默认类型
            guard !userDefaults.bool(forKey: "DefaultGenresLoaded") else {
                return
            }

            // 批量插入默认类型
            try container.mainContext.transaction {
                MovieGenre.defaultGenres.forEach { container.mainContext.insert($0) }
            }
            try container.mainContext.save()
            
            // 标记初始化完成，后续启动不再执行
            userDefaults.set(true, forKey: "DefaultGenresLoaded")
        } catch {
            fatalError("初始化模型容器失败: \(error)")
        }
    }

    var body: some Scene {
        WindowGroup {
            ContentView()
        }
        .modelContainer(container)
    }
}

优点：实现简单，性能开销极小，不需要每次查询数据库。
缺点：用户删除App重装会重新插入默认类型（属于合理行为）；手动修改UserDefaults可能导致重复插入，但实际场景概率极低。

方案2：SwiftData迁移计划（官方推荐） #

利用SwiftData的SchemaMigrationPlan，在数据库首次创建时注入默认数据，只有在数据库初始化/升级时执行，完全避免启动时的检查逻辑：

第一步：定义版本化Schema和迁移计划

import SwiftData

// 定义第一个版本的Schema
struct MovieSchemaV1: VersionedSchema {
    static var versionIdentifier: Schema.Version = .init(1, 0, 0)
    static var models: [any PersistentModel.Type] {
        [MovieGenre.self, Movie.self]
    }
}

// 定义迁移计划
enum MovieMigrationPlan: SchemaMigrationPlan {
    static var schemas: [VersionedSchema.Type] {
        [MovieSchemaV1.self]
    }

    static var stages: [MigrationStage] {
        // 从无到V1的轻量迁移，注入默认数据
        [MigrationStage.lightweight(from: nil, to: MovieSchemaV1.self) { context in
            MovieGenre.defaultGenres.forEach { context.insert($0) }
            try? context.save()
        }]
    }
}

第二步：初始化容器时指定迁移计划

@main
struct MovieApp: App {
    let container: ModelContainer

    init() {
        do {
            // 使用迁移计划初始化容器
            container = try ModelContainer(
                for: MovieGenre.self, Movie.self,
                migrationPlan: MovieMigrationPlan.self
            )
        } catch {
            fatalError("初始化模型容器失败: \(error)")
        }
    }

    var body: some Scene {
        WindowGroup {
            ContentView()
        }
        .modelContainer(container)
    }
}

优点：完全遵循SwiftData的持久化规范，只有在数据库首次创建时执行一次，后续启动无额外开销；支持后续版本的数据库迁移。
缺点：需要额外编写Schema和迁移计划代码，稍显繁琐但扩展性更好。

方案3：唯一约束+批量插入（自动去重） #

利用@Attribute(.unique)的约束特性，每次启动尝试插入默认类型，SwiftData会自动忽略重复的条目（需要捕获约束冲突错误）：

@main
struct MovieApp: App {
    let container: ModelContainer

    init() {
        do {
            container = try ModelContainer(for: MovieGenre.self, Movie.self)
            
            do {
                try container.mainContext.transaction {
                    MovieGenre.defaultGenres.forEach { container.mainContext.insert($0) }
                }
                try container.mainContext.save()
            } catch ModelError.constraintViolation {
                // 忽略重复插入的错误，说明默认类型已经存在
                print("默认类型已存在，跳过插入")
            } catch {
                throw error
            }
        } catch {
            fatalError("初始化模型容器失败: \(error)")
        }
    }

    var body: some Scene {
        WindowGroup {
            ContentView()
        }
        .modelContainer(container)
    }
}

优点：不需要额外标记，依赖数据库约束自动去重。
缺点：每次启动都会尝试插入，虽然不会实际创建重复数据，但会触发数据库的约束检查，有微小的性能开销；需要处理约束冲突错误。

后续优化建议 #

• 给默认类型添加保护逻辑，比如禁止用户删除或修改默认类型的名称；
• 如果需要支持CloudKit同步，确保MovieGenre类符合CloudKit同步要求（比如添加@CloudKit属性，或在容器配置中启用CloudKit）；
• 展示列表时，直接查询所有MovieGenre实体即可，无需区分默认和自定义，简化UI代码。

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