作者:Ole Begemann,原文链接,原文日期:2017-03-06
译者:Cwift;校对:numbbbbb;定稿:CMB

假设你有一个 Swift 的枚举:

enum Expression {
case number(Double)
case string(String)
}

你希望它遵守 Equatable 协议。由于该枚举具有关联值,必须手动添加,所以需要实现 == 函数

extension Expression: Equatable {
static func ==(lhs: Expression, rhs: Expression)
-> Bool {
switch (lhs, rhs) {
case let (.number(l), .number(r)): return l == r
case let (.string(l), .string(r)): return l == r
default: return false
}
}
}

这里处理了参数类型相同的两种情况,比较类型不同时会执行 default case 并返回 false。这种做法简单直接,也没错:

Expression.number(1) == .number(1) // → true
Expression.number(1) == .string("a") // → false

Default case 会使得枚举对穷尽的检查无效

然而这段代码有一个严重的缺陷:如果你在枚举中添加另一个 case,编译器不会警告你当前枚举的实现是不完整的。现在给枚举添加第三种 case:

enum Expression {
case number(Double)
case string(String)
case bool(Bool)
}

只要能通过编译器的检查,那么这种行为就是合理的。但执行以下操作时代码会返回错误的结果:

Expression.bool(true) == .bool(true) // → false!

switch 语句中的 default case 会使编译器对枚举的检查无效。所以,通常来说尽可能避免在 switch 语句中使用 default。

如果可能,尽量不要在 switch 语句中使用 default

模式匹配大爆炸

没有 default case 的缺点也很明显:你需要写更多的模式匹配代码。下面是完全覆盖三种 case 的写法:

extension Expression: Equatable {
static func ==(lhs: Expression, rhs: Expression)
-> Bool {
switch (lhs, rhs) {
case let (.number(l), .number(r)): return l == r
case let (.string(l), .string(r)): return l == r
case let (.bool(l), .bool(r)): return l == r
case (.number, .string),
(.number, .bool),
(.string, .number),
(.string, .bool),
(.bool, .number),
(.bool, .string): return false
}
}
}

o(>﹏<)o!写起来一点都不愉快,而且当枚举的 case 增加时会变得更糟。switch 语句必须区分的状态数会随着枚举中 case 的数量呈平方增长。

从平方增长到线性增长

_ 占位符可以简化你的 switch 语句。虽然不能使用 default 语句,但是我们可以把上一段代码最后的六行简化成三行:

extension Expression: Equatable {
static func ==(lhs: Expression, rhs: Expression)
-> Bool {
switch (lhs, rhs) {
case let (.number(l), .number(r)): return l == r
case let (.string(l), .string(r)): return l == r
case let (.bool(l), .bool(r)): return l == r
case (.number, _),
(.string, _),
(.bool, _): return false
}
}
}

这个方案更好,枚举中每增加一个 case,switch 语句只会增加两行,不再是平方级的增加。而且你保留了编译器穷尽检查的优势:添加一个新 case,编译器会报 == 的错误。

Sourcery

如果你觉得重复代码还是太多,可以看看 Krzysztof Zabłocki 开发的代码生成工具 Sourcery。在类似的应用场景中,它可以自动为枚举以及其他类型生成 Equatable 协议所需的代码(并且不断更新生成代码)。

本文由 SwiftGG 翻译组翻译,已经获得作者翻译授权,最新文章请访问 http://swift.gg

文章目录
  1. 1. Default case 会使得枚举对穷尽的检查无效
  2. 2. 模式匹配大爆炸
  3. 3. 从平方增长到线性增长
  4. 4. Sourcery