swift - 协议(protocol)中的泛型

Question

我有一个 UnitDimension 类:

class UnitDimension: {
    var symbol: String
    init(symbol: String) {
        self.symbol = symbol
    }
}

和这个类的一个 UnitVolume 子类:

class UnitVolume: UnitDimension {
    static let liter = UnitVolume(symbol: "L")
}

我想要一个协议(protocol)UnitDimensionHandler，它允许我执行一些简单的功能。首先它必须有一个 allUnits 变量:

protocol UnitDimensionHandler: class {
    static var allUnits: [UnitDimension] { get }
}

是否可以让 allUnits 成为一个通用类型的数组，它必须是 UnitDimension 的子类？ 然后我可以在 UnitVolume 如下:

extension UnitVolume: UnitDimensionHandler {
    static var allUnits: [UnitVolume] {
        return [liter]
    }
}

然后我想包含一个函数，该函数还允许我使用启动 UnitDimension 实例的通用子类类型:

extension UnitDimensionHandler {
    static func unit(for symbol: String) -> UnitDimension? {
        return allUnits.first() { $0.symbol == symbol }
    }
}

这样 UnitVolume.unit(for: "L") 返回可选的 UnitVolume 而不是可选的 UnitDimension。

感谢您的帮助。

Answer 1

是的，使用 associatedtype 是可能的:

protocol UnitDimensionHandler: class {
    // here we define `generic` type variable `Dimension`, and specify it do be descendant of `UnitDimension`
    associatedtype Dimension: UnitDimension

    // and use it here, instead of `UnitDimension`
    static var allUnits: [Dimension] { get }
}

和

extension UnitDimensionHandler {
    // same `UnitDimension` -> `Dimension` replacement
    static func unit(for symbol: String) -> Dimension? {
        return allUnits.first() { $0.symbol == symbol }
    }
}