Swift中的map 和 flatMap 原理及用法

map 和 flatMap 是 Swift 中两个常用的函数，它们体现了 Swift 中很多的特性。对于简单的使用来说，它们的接口并不复杂，但它们内部的机制还是非常值得研究的，能够帮助我们够好的理解 Swift 语言。

map 简介

首先，咱们说说 map 函数如何使用。

let numbers = [1,2,3,4]
let result = numbers.map { $0 + 2 }
print(result) // [3,4,5,6]


map 方法接受一个闭包作为参数， 然后它会遍历整个 numbers 数组，并对数组中每一个元素执行闭包中定义的操作。 相当于对数组中的所有元素做了一个映射。 比如咱们这个例子里面的闭包是讲所有元素都加 2 。 这样它产生的结果数据就是 [3,4,5,6]。

初步了解之后，我们来看一下 map 的定义：

func map
(@noescape transform: (Self.Generator.Element) throws -> T) rethrows -> [T]

咱们抛开一些和关键逻辑无关的修饰符 @noescape,throws 这些，在整理一下就是这样：

func map
(transform: (Self.Generator.Element) -> T) rethrows -> [T]
 ```

map 函数接受一个闭包， 这个闭包的定义是这样的：

(Self.Generator.Element) -> T
“`

它接受 Self.Generator.Element 类型的参数， 这个类型代表数组中当前元素的类型。 而这个闭包的返回值，是可以和传递进来的值不同的。 比如我们可以这样：

let stringResult = numbers.map { "No. \($0)" }
// ["No. 1", "No. 2", "No. 3", "No. 4"]

这次我们在闭包装把传递进来的数字拼接到一个字符串中， 然后返回一个组数， 这个数组中包含的数据类型，就是我们拼接好的字符串。

这就是关于 map 的初步了解， 我们继续来看 flatMap。

flatMap

map 可以对一个集合类型的所有元素做一个映射操作。 那么 flatMap 呢？

让我们来看一个 flatMap 的例子：

result = numbers.flatM