环境
- Xcode 12.2
前言
自从用上Swift后，就感觉Swift大法真是666，使用越多越觉得爽，下面就用一个场景再来吹一波😀
功能很简单，就是实现类似如下函数：
func queryParameters(for url: URL?) -> [String: String] { }
相信这个功能大家都能很快搞定，但这里我想对这个函数再加上3个要求：
- 函数功能明确简单，所以就不要写return了
- 转换步骤要一个一个来，尽量清晰，秒懂
- 尽量避免条件判断
满足这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大法呱呱叫🙃
