如何实现Swift枚举与SwiftData查找表的同步？

针对你的问题，核心痛点是既要用SwiftData的多Model关联实现可查询的关系，又要避免手动维护枚举和数据库查找表的一致性。下面是一套完整的强绑定+自动同步的解决方案，完全贴合你的需求：

1. 核心思路 #

我们要让枚举和SwiftData Model实现双向强绑定：

• 用枚举的rawValue作为Model的唯一ID，确保一一对应
• 把枚举的所有属性（包括计算属性）同步到Model中
• 实现自动同步逻辑，App启动时自动让数据库中的查找表和枚举的所有case保持一致
• 给上层代码提供便捷API，直接用枚举操作，无需关心Model细节

2. 完整代码实现 #

2.1 改造枚举与定义绑定的SwiftData Model #

首先保留原枚举的核心逻辑，增加和Model的关联；同时定义和枚举强绑定的DieModel（避免和枚举重名，更清晰）：

import SwiftData

// 原枚举保持核心逻辑，增加CaseIterable用于遍历所有case，以及和Model的关联
enum Die: Int, CaseIterable {
    case blue, yellow, violet, orange, green
    
    var numSides: Int {
        switch self {
        case .blue, .yellow: return 10
        case .violet, .orange: return 12
        case .green: return 20
        }
    }
    
    // 快速生成对应的SwiftData Model
    var model: DieModel {
        let model = DieModel()
        model.id = self.rawValue
        model.numSides = self.numSides
        return model
    }
}

// SwiftData查找表Model，和枚举强绑定
@Model final class DieModel {
    // 直接用枚举的rawValue作为唯一ID，确保和枚举一一对应
    var id: Int
    // 同步枚举的计算属性，实现数据库可查询
    var numSides: Int
    // 和Card的关联关系
    @Relationship(inverse: \Card.dice) var cards: [Card] = []
    
    // SwiftData要求的空初始化器
    init() {
        self.id = 0
        self.numSides = 0
    }
    
    // 从枚举快速初始化Model的便捷构造器
    init(die: Die) {
        self.id = die.rawValue
        self.numSides = die.numSides
    }
    
    // 从Model转回原枚举的便捷属性
    var asEnum: Die? {
        Die(rawValue: id)
    }
}

// 关联DieModel的Card Model
@Model final class Card {
    var id: String
    @Relationship(inverse: \DieModel.cards) var dice: [DieModel] = []
    
    init(id: String) {
        self.id = id
    }
    
    // 便捷API：直接用原枚举数组设置关联的骰子
    func setDice(_ dice: [Die], using context: ModelContext) {
        // 先确保数据库中已存在所有枚举对应的Model
        syncDieModels(to: context)
        // 查找并关联对应的Model
        self.dice = dice.compactMap { die in
            try? context.fetch(FetchDescriptor<DieModel>(predicate: #Predicate { $0.id == die.rawValue })).first
        }
    }
    
    // 便捷API：获取关联的原枚举数组
    var diceEnums: [Die] {
        dice.compactMap { $0.asEnum }
    }
}

2.2 实现枚举与查找表的自动同步 #

这个工具方法会在App启动时执行，确保数据库中的DieModel记录和枚举的所有case完全一致：

/// 同步枚举到SwiftData数据库，确保查找表和枚举完全一致
func syncDieModels(to context: ModelContext) {
    // 1. 获取数据库中已有的所有DieModel的ID
    let existingIds = (try? context.fetch(FetchDescriptor<DieModel>()))?.map { $0.id } ?? []
    // 2. 获取枚举所有case的rawValue
    let enumRawValues = Die.allCases.map { $0.rawValue }
    
    // 3. 插入枚举新增的case对应的Model
    let missingRawValues = enumRawValues.filter { !existingIds.contains($0) }
    for rawValue in missingRawValues {
        guard let die = Die(rawValue: rawValue) else { continue }
        context.insert(die.model)
    }
    
    // 4. 移除枚举已删除的case对应的Model（可选，根据业务需求决定是否保留历史数据）
    let extraIds = existingIds.filter { !enumRawValues.contains($0) }
    for id in extraIds {
        if let extraModel = try? context.fetch(FetchDescriptor<DieModel>(predicate: #Predicate { $0.id == id })).first {
            context.delete(extraModel)
        }
    }
    
    // 5. 保存同步结果
    try? context.save()
}

2.3 如何使用 #

在App初始化、创建ModelContainer后，立即执行同步：

// 初始化SwiftData容器
let container = try ModelContainer(for: Card.self, DieModel.self)
// 同步枚举与查找表
syncDieModels(to: container.mainContext)

// 示例：创建卡片并关联骰子
let newCard = Card(id: "card_001")
// 直接用原枚举设置骰子
newCard.setDice([.blue, .green], using: container.mainContext)
// 保存卡片
container.mainContext.insert(newCard)
try? container.mainContext.save()

// 示例：查询所有关联20面骰子的卡片
let predicate = #Predicate<Card> { $0.dice.contains(where: { $0.numSides == 20 }) }
let cardsWith20SidedDice = try container.mainContext.fetch(FetchDescriptor<Card>(predicate: predicate))

3. 方案优势 #

1. 完全自动同步：无需手动维护枚举和数据库的一致性，新增/删除枚举case后，App启动时自动同步
2. 可查询性：现在可以对DieModel的任何属性（id/numSides）进行过滤、排序，解决了原方案中枚举数组无法查询的问题
3. 代码友好：上层代码可以直接用原枚举操作，无需关心SwiftData Model的细节
4. 数据一致性：用枚举rawValue作为Model的ID，确保枚举和Model的一一对应，不会出现数据不匹配

4. 注意事项 #

• 枚举必须实现CaseIterable，这样才能遍历所有case进行同步
• 如果你的业务需要保留历史数据（比如枚举删除case后，数据库中对应的记录不能删除），可以移除同步逻辑中"移除多余Model"的代码块
• 枚举的rawValue一旦确定，不要随意修改，否则会导致数据库中已有的DieModelID和枚举不匹配

内容来源于stack exchange