SwiftData通用getObjectByID函数报错问题及解决咨询

问题背景 #

原本实现了按ID获取特定SwiftData类对象的函数，运行正常。尝试重构为适配不同类的泛型函数后，出现以下代码及两个错误：

func getObjectByID<T: PersistentModel>(context: ModelContext, id: Int) -> T? {
    do {
        let predicate = #Predicate<T> {
            id == $0.id
        }
        let descriptor = FetchDescriptor<T>(predicate: predicate)
        let objects = try context.fetch(descriptor)
        return objects.first
    } catch {
        return nil
    }
}

错误信息 #

1. id == $0.id 行报错：Operator function '==' requires that 'PersistentIdentifier' conform to 'BinaryInteger'
2. let objects = try context.fetch(descriptor) 行报错：Generic parameter 'T' could not be inferred

错误原因及解决方法 #

错误1：类型不匹配导致的比较失败 #

原因

SwiftData中PersistentModel协议默认的id属性类型是PersistentIdentifier（一个包装类型），而非你传入的Int类型。直接用Int和PersistentIdentifier进行==比较，编译器找不到对应的重载运算符，因此报错。

解决方法

有两种常见方案：

方案1：自定义协议约束模型类的id为Int类型

定义一个协议，要求模型类必须有Int类型的id属性，然后让泛型函数约束T遵循该协议：

protocol IntIdentifiable: PersistentModel {
    var id: Int { get }
}

// 你的SwiftData模型类需要遵循这个协议，比如：
// @Model class User: IntIdentifiable {
//     var id: Int
//     // 其他属性...
// }

func getObjectByID<T: IntIdentifiable>(context: ModelContext, id: Int) -> T? {
    do {
        let predicate = #Predicate<T> { $0.id == id }
        let descriptor = FetchDescriptor<T>(predicate: predicate)
        let objects = try context.fetch(descriptor)
        return objects.first
    } catch {
        return nil
    }
}

方案2：将参数id改为PersistentIdentifier类型

如果你的模型使用默认的PersistentIdentifier作为id，直接修改参数类型：

func getObjectByID<T: PersistentModel>(context: ModelContext, id: PersistentIdentifier) -> T? {
    do {
        let predicate = #Predicate<T> { $0.id == id }
        let descriptor = FetchDescriptor<T>(predicate: predicate)
        let objects = try context.fetch(descriptor)
        return objects.first
    } catch {
        return nil
    }
}

错误2：泛型参数推断失败 #

原因

这个错误大多是第一个错误导致的连锁反应：当Predicate中的类型不匹配时，编译器无法正确推断FetchDescriptor和fetch方法的泛型参数T。解决第一个错误后，这个问题通常会自动消失。

如果仍存在推断问题，调用函数时可以显式指定泛型类型：

// 示例调用，显式指定T为你的模型类
let user = getObjectByID<User>(context: modelContext, id: 1)

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