SwiftData恢复时如何处理关联关系下的数据重复问题？

问题原因 #

导出数据时同时包含了Entry（其关联的Criteria会被序列化）和独立的Criteria列表，恢复过程中未对UUID做唯一性校验，导致同一UUID的Criteria被插入两次：一次来自Entry关联字段的解析，一次来自单独导出的Criteria列表。

解决方案 #

1. 给Criteria模型添加UUID唯一约束 #

在SwiftData模型的UUID属性上标记唯一约束，从数据库层面阻止重复插入：

import SwiftData

@Model
class Criteria {
    @Attribute(.unique) let id: UUID
    var name: String // 示例属性
    
    init(id: UUID = UUID(), name: String) {
        self.id = id
        self.name = name
    }
}

@Model
class Entry {
    let id: UUID
    @Relationship var criteria: Criteria? // 或根据你的关联类型调整
    
    init(id: UUID = UUID(), criteria: Criteria?) {
        self.id = id
        self.criteria = criteria
    }
}

2. 调整恢复逻辑，先处理Criteria并去重 #

恢复时优先导入Criteria，通过UUID校验跳过已存在的记录，再导入Entry关联已存在的Criteria：

func restore(from backupData: Data, using context: ModelContext) throws {
    // 解析备份JSON（假设Backup结构包含entries和criteria数组）
    let backup = try JSONDecoder().decode(Backup.self, from: backupData)
    
    // 先处理Criteria：过滤本地已存在的记录
    let existingIds = try context.fetch(FetchDescriptor<Criteria>()).map { $0.id }
    for criteria in backup.criteria where !existingIds.contains(criteria.id) {
        context.insert(criteria)
    }
    try context.save() // 先保存Criteria，确保后续Entry能关联到
    
    // 处理Entry：关联已存在的Criteria实例
    for entry in backup.entries {
        if let criteriaId = entry.criteria?.id {
            if let existingCriteria = try context.fetch(
                FetchDescriptor<Criteria>(predicate: #Predicate { $0.id == criteriaId })
            ).first {
                entry.criteria = existingCriteria
            }
        }
        context.insert(entry)
    }
    try context.save()
}

// 备份数据结构示例（需符合Codable）
struct Backup: Codable {
    let entries: [Entry]
    let criteria: [Criteria]
}

3. 优化导出逻辑（可选） #

导出时先对Criteria去重，避免JSON中出现重复的Criteria记录，减少恢复时的处理压力：

func createBackup(context: ModelContext) throws -> Data {
    let allEntries = try context.fetch(FetchDescriptor<Entry>())
    let allCriteria = try context.fetch(FetchDescriptor<Criteria>())
    // 通过UUID去重Criteria（防止关联导出时重复）
    let uniqueCriteria = Dictionary(grouping: allCriteria, by: \.id).values.map { $0.first! }
    
    let backup = Backup(entries: allEntries, criteria: uniqueCriteria)
    return try JSONEncoder().encode(backup)
}

注意事项 #

• 确保JSON编解码器正确处理UUID，默认JSONEncoder/JSONDecoder已支持UUID的序列化，无需额外处理。
• 恢复前建议先备份当前数据库，避免操作失误导致数据丢失。
• 若为多对多关联，逻辑一致：优先处理关联模型的唯一性，再处理主模型的关联关系。

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