swift - 如何在 Swift 中打开类类型？

Question

我有以下代码:

class Base {}
class A: Base {}
class B: Base {}
class C: Base {}

func next(obj: Base) -> Base {
    if obj is A { return B() }
    else if obj is B { return C() }
    else if obj is C { return A() }
    else { return A() }
}

我如何表达链式 if s 作为一个人 switch声明代替？

Answer 1

extension Base {

    func next() -> Base {
        switch self {
        case is A: return B()
        case is B: return C()
        case is C: return A()
        default: return A()
        }
    }

}