(备注:答案已更新以反映 Swift 3 及更高版本中的语法变化,例如取消ImplicitlyUnwrappedOptional.)
Optional.map()并Optional.flatMap()声明如下(我省略了与此处无关的 throws/rethrows 修饰符):
func map<U>(_ transform: (Wrapped) -> U) -> U?
func flatMap<U>(_ transform: (Wrapped) -> U?) -> U?
让我们考虑使用“map”的第一个示例的简化版本:
let number: Int? = 1
let res1 = number.map { $0 + 1 }
print(res1) // Optional(2)
number具有类型Int?并且闭包类型被推断为(Int) -> Int. U是Int,返回值的类型是Int?。numberis not nil,所以它被解包并传递1给闭包。闭包返回2并map返回Optional(2)。如果number是nil那么结果将是nil。
现在我们考虑使用“flatMap”的第二个示例的简化版本:
let number: Int? = 1
let res2 = number.flatMap { $0 + 1 }
print(res2) // Optional(2)
flatMap期望闭包 type (Wrapped) -> U?,但{ $0 + 1 }不返回可选项。为了使其编译,编译器将其转换为
let res2 = number.flatMap { return Optional($0 + 1) }
现在闭包有 type (Int) -> Int?,并且又U是Int。再次,number被打开并传递给闭包。闭包返回Optional(2),它也是来自的返回值flatMap。如果number是nil 或者如果闭包会返回nil,那么结果将是nil.
所以这些调用之间确实没有区别:
let res1 = number.map { $0 + 1 }
let res2 = number.flatMap { $0 + 1 }
然而,这不是flatMap目的。一个更现实的例子是
func foo(_ s : String?) -> Int? {
return s.flatMap { Int($0) }
}
print(foo("1")) // Optional(1)
print(foo("x")) // nil (because `Int($0)` returns nil)
print(foo(nil)) // nil (because the argument is nil)
通常,map采用闭包类型(Wrapped) -> U和转换
Optional<Wrapped>.none --> Optional<U>.none
Optional<Wrapped>.some(wrapped) --> Optional<U>.some(transform(wrapped))
flatMap接受类型的闭包(Wrapped) -> U?并转换
Optional<Wrapped>.none --> Optional<U>.none
Optional<Wrapped>.some(wrapped) --> transform(wrapped)
这里transform(wrapped)也可以Optional<U>.none。
如果(如您的示例中)使用不flatMap返回可选项的闭包调用,则编译器会自动将其转换为可选项,并且不再有区别。map