一个map家族常用场景吹一波Swift

环境

• Xcode 12.2

前言

自从用上Swift后，就感觉Swift大法真是666，使用越多越觉得爽，下面就用一个场景再来吹一波😀

功能很简单，就是实现类似如下函数：

func queryParameters(for url: URL?) -> [String: String] { }

相信这个功能大家都能很快搞定，但这里我想对这个函数再加上3个要求：

1. 函数功能明确简单，所以就不要写return了
2. 转换步骤要一个一个来，尽量清晰，秒懂
3. 尽量避免条件判断

满足这3个要求后，最后写出来的代码应该就能比较Swifty了吧😝

先放个最后自己写好的，感兴趣的话可以再看看后面的过程：

func queryParameters(for url: URL?) -> [String: String] {
    (url?.absoluteString)
        .flatMap { URLComponents(string: $0) }?
        .queryItems?
        .compactMap { item in
            item.value?.removingPercentEncoding.flatMap { (item.name, $0) }
        }
        .toDictionary() ?? [:]
}

思路

首先，想省去return，但中间又会有各种转换，那就用链式调用方式呗。

然后我这里想使用URLComponents的queryItems来省去自己拆分参数的烦恼。。

URL -> URLComponents

使用URLComponents(string:)，而不是URLComponents(url:, resolvingAgainstBaseURL:)可以多遇到一个问题。。

要对可选类型做转换，flatMap出场，不过刚开始就遇到个问题🤣，第一次代码：

url?.absoluteString.flatMap { URLComponents(string: $0) }

发现报错:

Cannot convert value of type 'String.Element' (aka 'Character') to expected argument type 'String'

flatMap里面的$0是Character，而不是String，因为String实现了Collection协议(这个在Swift大版本中改过去又改过来)。

那试试加个?号：

url?.absoluteString?.flatMap { URLComponents(string: $0) }

结果还是报错，想了下，最后加括号搞定：

(url?.absoluteString)
    .flatMap { URLComponents(string: $0) }?
    .queryItems?

上面说的多遇到一个问题就是如此，需要引入括号来避免歧义，但其实这里最优的写法应该如下：

url.flatMap { URLComponents(url: $0, resolvingAgainstBaseURL: true) }?
        .queryItems?

ok，拿到queryItems了，不过是[URLQueryItem]类型，下面继续。

[URLQueryItem] -> [(String，String)]

[URLQueryItem]不方便直接转换[String: String]，需要先转换成 [(String，String)] ，再使用Dictionary(uniqueKeysWithValues:)来完成目标。

Map

那还不简单么，集合转换，一个map不就搞定了：

url.flatMap { URLComponents(url: $0, resolvingAgainstBaseURL: true) }?
        .queryItems?
        .map { ($0.name, $0.value) }

这样语法是没错，但有个问题，URLQueryItem的value是String?，而不是String，

所以这个转换结果是 [(String，String?)] ，而不是 [(String，String)] ，不符合要求。

那就只有过滤掉value为nil的情况了，有请map大哥compactMap出场。

compactMap这货也是中间杀出来多，最开始只有map和flatMap。

compactMap

要过滤掉nil，那是不是得先判断下？没有值就返回nil？

其实不用，flatMap可以优雅的实现这个功能🤩，顺便把value decode了：

url.flatMap { URLComponents(url: $0, resolvingAgainstBaseURL: true) }?
      .queryItems?
      .compactMap { item in
          item.value?.removingPercentEncoding.flatMap { (item.name, $0) }
      }

OK，这里我们顺利转换成[(String，String)] 了，还有最后一步。

[(String，String)] -> [String: String]

最后发现好像没法直接链式调用Dictionary(uniqueKeysWithValues: )，杂办呢？

extension出场，加上where可以限制元素类型，真是666😎

extension Array where Element == (String, String) {
    public func toDictionary() -> [String: String] {
        Dictionary(uniqueKeysWithValues: self)
    }
}

可选链到最后也是可选的，所以加个默认空值，搞定！！！

func queryParameters(for url: URL?) -> [String: String] {
    url.flatMap { URLComponents(url: $0, resolvingAgainstBaseURL: true) }?
        .queryItems?
        .compactMap { item in
            item.value?.removingPercentEncoding.flatMap { (item.name, $0) }
        }
        .toDictionary() ?? [:]
}

总结

Swift大法呱呱叫🙃