以 Swift 的方式思考，第二部分：调用数组的 Map 方法

最新推荐文章于 2024-07-08 03:54:10 发布

NSNirvana

最新推荐文章于 2024-07-08 03:54:10 发布

阅读量2.4k

点赞数

分类专栏： Swift 文章标签： swift 数组 map

Swift 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

在本系列文章的第一篇中，我们知道了怎样避免对可选类型强制拆包，尽量少使用 ! 也避免了程序崩溃。在第二部分，我将会精简我们的代码来让它看起来更 “Swift”，同时向你介绍 map() 和 flatMap() 方法。

今天这篇文章我们将要讨论数组的 map() 和 flatMap() 方法。

本系列文章的第一篇里

这是我们上次留下的代码。

class ListItem {
    var icon: UIImage?
    var title: String = ""
    var url: NSURL!

    static func listItemsFromJSONData(jsonData: NSData?) -> [ListItem] {
        guard let nonNilJsonData = jsonData,
            let json = try? NSJSONSerialization.JSONObjectWithData(nonNilJsonData, options: []),
            let jsonItems = json as? Array<NSDictionary>
            else {
                // If we failed to unserialize the JSON or that JSON wasn't an NSArray,
                // then bail early with an empty array
                return []
        }

        var items = [ListItem]()
        for itemDesc in jsonItems {
            let item = ListItem()
            if let icon = itemDesc["icon"] as? String {
                item.icon = UIImage(named: icon)
            }
            if let title = itemDesc["title"] as? String {
                item.title = title
            }
            if let urlString = itemDesc["url"] as? String, let url = NSURL(string: urlString) {
                item.url = url
            }
            items.append(item)
        }
        return items
    }
}

我们的目的是使用更多更 “Swift” 的模式和语法来让我们的代码变得更简洁。

map()函数的介绍

map() 是 Array 提供的方法，它接收一个函数作为参数，旧数组中的每个元素都会被拿去执行这个函数而变成新数组的对应元素，这是一种让数组从 [X] 转化为 [Y] 的方式，你需要提供的就是 X -> Y 的转化方式，而不必新建一个临时数组。

在我们例子里，我们不再像之前一样用 for 来做循环，而是对 jsonItems – 我们 NSDictionary 类型的 JSON 数组，使用 map() 方法，为它提供一个参数函数，将原来的每个 NSDictionary 类型转换成我们所需的 ListItem 实例：

return jsonItems.map { (itemDesc: NSDictionary) -> ListItem in
    let item = ListItem()
    if let icon = itemDesc["icon"] as? String {
        item.icon = UIImage(named: icon)
    }
    if let title = itemDesc["title"] as? String {
        item.title = title
    }
    if let urlString = itemDesc["url"] as? String, let url = NSURL(string: urlString) {
        item.url = url
    }
    return item
}

这看起来只是个很小的改变，但是它让我们可以专注于怎样把 NSDictionary 转化成 ListItem ，毕竟这也是我们要解决的核心问题，更重要的是，我们避免了像我们在Objc里做的那样，新建一个中间数组。我们应该尽可能的避免这种情况发生。

错误数据

我们还有个问题要解决，就是，即便输入的数据是不可用的，我们还是创建了一个 ListItem 实例。这样的话，如果我们的某些 NSDictionary 是不可用的，最后的输出数组里，就会有很多无意义的无内容的 ListItem() 实例，其中也就不包含可用的 NSURL。

更糟的是，代码允许我们创建那些没有可用 NSURL 的 ListItem() 实例，而我们在 NSURL! 中使用了 !，这样一旦执行了 NSURL! 这样的强制拆包，我们的程序就得崩。

为了解决这个问题，如果我们接收到的输入是不可用的，就要返回一个 nil，这比返回一个无内容的或者错误的 ListItem 来导致拆包 NSURL 时崩溃要好得多。

return jsonItems.map { (itemDesc: NSDictionary) -> ListItem? in
    guard …/* condition for valid data */… else { return nil }
    let realValidItem = ListItem()
    … /* fill the ListItem with the values */
    return realValidItem
}

但是我们现在 jsonItems.map 里面传入的参数函数的类型为 NSDictionary -> ListItem?，最后我们得到的是一个 [ListItem?] 数组，那些原来是不可用 NSDictionary 的位置就被我们替换成了 nil。比原来要好一些了，但还不够。

使用flatMap()

这个时候就轮到 flatMap() 来救场了。

flatMap() 与 map() 相似，但 flatMap() 用的是 T->U? 转化而不是 T->U 转化，而且如果转化出的数组元素是 nil 的话，就不会被添加到最后的结果数组里面。

在语法上，你可以这么理解，flatMap 就是你先使用 map 然后把结果数组“压平”（毕竟函数名就是这个意思)，也就是从输出数组里去掉那些 nil。

如果使用了这个方法，代码就会变成这个样子：

return jsonItems.flatMap { (itemDesc: NSDictionary) -> ListItem? in
    guard let title = itemDesc["title"] as? String,
        let urlString = itemDesc["url"] as? String,
        let url = NSURL(string: urlString)
        else { return nil }
    let li = ListItem()
    if let icon = itemDesc["icon"] as? String {
        li.icon = UIImage(named: icon)
    }
    li.title = title
    li.url = url
    return li
}