如何在Factory泛型扩展中实现编译期协议一致性类型检查？

你拥有一个外部泛型结构体Factory<T>，并对其进行了扩展，但当前的扩展允许非法类型转换通过编译（例如将Factory<Loggable>转换为未实现Loggable协议的SomeStruct），最终导致运行时崩溃。你希望在编译阶段就阻止这类非法转换，需要实现类似func callAsFunction<U: T>() -> U的效果。

相关代码 #

外部提供的Factory<T>结构体：

struct Factory<T> {
    public func callAsFunction() -> T {
        registration.resolve(with: ())
    }
}

你的扩展及测试代码：

import Factory

extension Container {
    var logger: Factory<Loggable> { Factory(self) { PrintLogger() } }
}

protocol Loggable { }

struct PrintLogger: Loggable { }

extension Factory {
    func callAsFunction<U>() -> U {
        guard let result = callAsFunction() as? U else {
            preconditionFailure("Found '\(T.self)', but unable to convert it to '\(U.self)'")
        }
        return result
    }
}

struct SomeExample {
    func example() {
        // 合法：PrintLogger实现了Loggable
        let logger: PrintLogger = Container.shared.logger()
        // 可编译但会崩溃：SomeStruct未实现Loggable
        let invalidLogger: SomeStruct = Container.shared.logger()
    }
    
    struct SomeStruct { }
}

解决方案：通过泛型约束在编译期校验类型兼容性 #

Swift中仅当T是协议时，U: T的泛型约束语法才有效，针对你的场景（T为协议Loggable），可以给扩展方法添加where U: T的约束，让编译器在编译阶段就检查U是否遵循T协议：

extension Factory {
    func callAsFunction<U>() -> U where U: T {
        guard let result = callAsFunction() as? U else {
            preconditionFailure("Found '\(T.self)', but unable to convert it to '\(U.self)'")
        }
        return result
    }
}

修改后，let invalidLogger: SomeStruct = Container.shared.logger()会直接编译失败——因为SomeStruct未遵循Loggable协议，不符合U: T的约束。

如果后续T是具体类型（比如Factory<PrintLogger>），则可以根据需求调整约束：

• 若只允许精确匹配，使用where U == T
• 若允许子类转换，使用where U: T（仅当T是类类型时生效）

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