SwiftData自定义JSON DataStore添加关联关系后加载失败求助

你的问题出在WWDC提供的基础JSON DataStore实现没有处理关联属性的标识符映射，导致保存的关联引用还是临时ID，和实体实际的永久ID不匹配，最终加载时SwiftData无法正确解析关联关系。

问题根源 #

当你插入带有关联的实体时：

1. SwiftData会先给所有新实体分配临时PersistentIdentifier
2. 示例代码仅为每个插入的快照生成了永久ID，但没有更新快照内部的关联属性值——Settings的hack字段依然保留着Hack的临时ID
3. 保存的JSON中，Settings关联的临时ID和Hack实际的永久ID不一致，加载时SwiftData找不到对应实体，或者混淆了实体类型，引发致命错误

解决方案 #

需要修改JSONStore的save方法，手动处理关联属性的ID映射：先为所有插入的实体建立临时ID到永久ID的映射表，再遍历快照的所有属性，把关联的临时ID替换成对应的永久ID。

步骤1：添加ID映射辅助函数 #

在JSONStore类中添加一个递归处理属性值的辅助函数，用于替换嵌套或集合中的临时ID：

private func remapIdentifiers(in value: Any, using map: [PersistentIdentifier: PersistentIdentifier]) -> Any {
    // 处理单个PersistentIdentifier
    if let tempId = value as? PersistentIdentifier {
        return map[tempId] ?? tempId
    }
    // 处理数组类型的关联（比如@Relationship(.cascade) var hacks: [Hack]）
    else if let array = value as? [Any] {
        return array.map { remapIdentifiers(in: $0, using: map) }
    }
    // 处理嵌套字典（如果有复杂关联结构）
    else if let dict = value as? [String: Any] {
        var newDict = [String: Any]()
        for (key, val) in dict {
            newDict[key] = remapIdentifiers(in: val, using: map)
        }
        return newDict
    }
    // 其他类型直接返回
    return value
}

步骤2：修改save方法逻辑 #

调整save方法的执行顺序，先建立所有插入实体的ID映射，再替换快照中的关联ID：

func save(_ request: DataStoreSaveChangesRequest<DefaultSnapshot>) throws -> DataStoreSaveChangesResult<DefaultSnapshot> {
    var remappedIdentifiers = [PersistentIdentifier: PersistentIdentifier]()
    var serializedData = try read()
    
    // 第一步：先为所有插入的实体生成永久ID，建立临时ID→永久ID的映射表
    for snapshot in request.inserted {
        let permanentIdentifier = try PersistentIdentifier.identifier(for: identifier,
                                                                      entityName: snapshot.persistentIdentifier.entityName,
                                                                      primaryKey: UUID())
        remappedIdentifiers[snapshot.persistentIdentifier] = permanentIdentifier
    }
    
    // 第二步：处理插入的实体，替换关联属性中的临时ID
    for snapshot in request.inserted {
        guard let permanentId = remappedIdentifiers[snapshot.persistentIdentifier] else { continue }
        // 映射所有属性值中的临时ID
        let remappedValues = snapshot.values.mapValues { remapIdentifiers(in: $0, using: remappedIdentifiers) }
        // 生成替换后的永久快照
        let permanentSnapshot = snapshot.copy(values: remappedValues, persistentIdentifier: permanentId)
        serializedData[permanentId] = permanentSnapshot
    }
    
    // 处理更新的实体（如果更新中包含新的关联，同样需要映射ID）
    for snapshot in request.updated {
        let remappedValues = snapshot.values.mapValues { remapIdentifiers(in: $0, using: remappedIdentifiers) }
        let updatedSnapshot = snapshot.copy(values: remappedValues, persistentIdentifier: snapshot.persistentIdentifier)
        serializedData[snapshot.persistentIdentifier] = updatedSnapshot
    }
    
    // 处理删除的实体（原逻辑不变）
    for snapshot in request.deleted {
        serializedData[snapshot.persistentIdentifier] = nil
    }
    
    try write(serializedData)
    
    return DataStoreSaveChangesResult<DefaultSnapshot>(for: self.identifier, remappedIdentifiers: remappedIdentifiers)
}

修改后的效果 #

保存的JSON中，Settings的hack字段会指向Hack的永久ID，加载时SwiftData能正确匹配关联实体，不会再出现崩溃。

这不是Apple的Bug，也不是自定义DataStore不支持关联——WWDC示例只是最基础的实现，需要开发者自己扩展关联关系、复杂查询等进阶功能。

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