(备注:答案已更新以反映 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?
。number
is 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