Swift泛型函数中如何判断类型是否符合继承层级中的协议

这个问题我之前做Swift协议相关开发时也踩过坑，确实很容易被协议元类型的检查逻辑搞晕。咱们先拆解下问题出在哪，再给出可行的解决方案。

为什么原来的实现会失败？ #

你原来的代码里用了 T.self is A.Protocol，这个判断只能识别T正好是协议A本身的情况。当传入的是继承自A的协议B时，B.self 的类型是 B.Protocol，而Swift并不会自动把协议元类型的继承关系映射到 is 操作符的检查中——也就是说，B.Protocol 并不会被认为是 A.Protocol 的子类，所以这个判断会返回false。

正确的实现方式 #

我们可以利用Swift中所有协议元类型都遵循的 Protocol 协议，调用它的 conforms(to:) 方法来检查协议之间的一致性。同时还要兼顾类类型的情况，因为如果传入的是遵循A的类，直接用 is 检查就可以。

修改后的函数如下：

func method<T>(_ type: T.Type) -> Bool {
    // 先判断是否是协议类型，用conforms(to:)检查是否遵循A
    if let protocolType = type as? Protocol {
        return protocolType.conforms(to: A.self)
    }
    // 如果是类类型，直接检查是否遵循A
    return T.self is A.Type
}

测试验证 #

我们用你定义的协议和一个遵循B的类来测试：

protocol A: AnyObject {}
protocol B: A {}
class C: B {}

print(method(A.self)) // 输出 true
print(method(B.self)) // 输出 true
print(method(C.self)) // 输出 true

这样不管是传入A本身、继承自A的协议B，还是遵循B的类C，都能正确返回true，符合你的预期。

补充说明 #

• conforms(to:) 方法是 Protocol 协议的核心方法，专门用来判断一个协议（或遵循该协议的类型）是否符合另一个协议的要求，包括协议继承的情况。
• 对于类类型，T.self is A.Type 可以直接生效，因为类遵循协议时，Swift会正确识别这种一致性。

内容的提问来源于stack exchange，提问作者Alberto García