Swift中如何让DataStore实现直接使用BaseEntity的UUID类型ID？

我想要构建如下代码结构：

protocol BaseEntity: Identifiable {
    var id: UUID { get set }
}

protocol DataStore {
    associatedtype T: BaseEntity
    
    typealias CompleteHandler<T> = ((Result<T, Error>) -> Void)

    func read(with id: T.ID, completion: CompleteHandler<T>)
}

但在具体实现类Storage中，我希望直接使用BaseEntity协议中定义的UUID类型的id，不想采用类型转换的方式：

func read(with id: Entity.ID, completion: CompleteHandler<Entity>) {
    guard let uuid = id as? UUID else {
        return
    }
        
    // 后续代码...
}

请问应该如何实现？

核心思路是明确约束BaseEntity的ID关联类型必须为UUID，让编译器直接推断Entity.ID的类型为UUID，无需转换。

步骤1：修改BaseEntity协议，显式约束ID类型 #

通过where子句强制所有遵循BaseEntity的类型，其Identifiable的ID必须是UUID：

protocol BaseEntity: Identifiable where ID == UUID {
    var id: UUID { get set }
}

步骤2：调整DataStore协议（可选优化） #

可以简化CompleteHandler的定义，利用协议的关联类型T，无需重复声明泛型：

protocol DataStore {
    associatedtype T: BaseEntity
    
    typealias CompleteHandler = ((Result<T, Error>) -> Void)

    func read(with id: T.ID, completion: CompleteHandler)
}

步骤3：实现Storage类 #

此时Entity.ID的类型已被编译器明确为UUID，可以直接使用：

class Storage<Entity: BaseEntity>: DataStore {
    func read(with id: Entity.ID, completion: CompleteHandler) {
        // id直接是UUID类型，可直接用于逻辑处理
        print("执行查询，ID：\(id.uuidString)")
        
        // 示例：返回模拟结果
        let mockEntity = MockEntity(id: id)
        completion(.success(mockEntity as! Entity))
    }
}

// 示例遵循BaseEntity的类型
class MockEntity: BaseEntity {
    var id: UUID
    init(id: UUID) {
        self.id = id
    }
}

为什么这个方案可行？ #

通过在BaseEntity的协议定义中添加where ID == UUID，我们从根源上保证了所有符合该协议的类型，其标识ID的类型只能是UUID。在泛型类Storage中，Entity遵循BaseEntity，因此Entity.ID的类型会被编译器直接推断为UUID，无需任何类型转换操作，同时也避免了类型转换可能带来的运行时错误。

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