Swift标准库源码之旅 - Zip.swift

zip在Swift里是一个全局函数，参数接收两个Sequence，返回一个可以遍历两个sequence的Element的元组Zip2Sequence.

使用举例

目前项目中有一个视图模型ViewModel有title和image属性，用来给视图赋值用。

struct ViewModel {
  let title: String
  let image: UIImage
}

这里面的title和image都是直接在本地进行配置，总共有5个。

第一种常规方式我们可以这样

let viewModel1 = ViewModel(title: "1", image: image1)
let viewModel2 = ViewModel(title: "2", image: image2)
let viewModel3 = ViewModel(title: "3", image: image3)
let viewModel4 = ViewModel(title: "4", image: image4)
let viewModel5 = ViewModel(title: "5", image: image5)
let viewModels = [viewModel1, xxx]

但是调用五次构造方法不是个好的方式，若是viewModel将来扩展一个参数，我们5个构造函数都需要更改。数据的配置与构建没有分层

第二种方式:

let titles: [String] = ["1", "2", "3", "4", "5"]
let images: [UIImage] = [image1, image2,image3,image4,image5]

var viewModels: [ViewModel] = []
for i in 0 ..< titles.count {
  viewModels.append(ViewModel(title: titles[i], image: images[i])
}

解决了第一种方式提到的问题.配置数据和构建模型是分开的. 构造方法也只调用了一次.但是使用下标从数组里取值还有更优雅的方式.

使用zip

let titles: [String] = ["1", "2", "3", "4", "5"]
let images: [UIImage] = [image1, image2,image3,image4,image5]

let viewModels = zip(titles, images).map(ViewModel.init)

除了元素的配置，构建viewModel数组只需要一行代码就搞定了.
zip方法在两个Sequence中其中一个没有next元素了就进行了停止.

那么如果ViewModel不是有两个元素而是三个呢. 可以选择zip嵌套或者ViewModel属性里用元组的方式.

zip(array1, zip(array2, array3)).map {
      ViewModel.init(a: $0, b: $1.0, c: $1.1)
}

实现

核心就是zip函数返回的Zip2Sequence序列的Iterator方法，也就是next函数.
思路就是将两个sequence全都保存下来,迭代的时候取两个sequence的迭代方法,只要有一个返回了nil就返回nil，取到值就包到元组里进行返回，比较简单.

class Zip2Sequence<S1: MySequence, S2: MySequence>: MySequence {
    typealias Element = (S1.Element, S2.Element)
    
    typealias Iterator = Interator
    let _sequence1: S1
    let _sequence2: S2
    
    init(base1: S1, base2: S2) {
        _sequence1 = base1
        _sequence2 = base2
    }
    
    func makeIterator() -> Interator {
        Interator(it1: _sequence1.makeIterator(), it2: _sequence2.makeIterator())
    }
}

extension Zip2Sequence {
    struct Interator: MyIteratorProtocol {
        
        var it1: S1.Iterator
        var it2: S2.Iterator
        
        init(it1: S1.Iterator, it2: S2.Iterator) {
            (self.it1, self.it2) = (it1, it2)
        }
        
        mutating func next() -> (S1.Element, S2.Element)? {
            guard let e1 = it1.next(), let e2 = it2.next() else {
                return nil
            }
            return (e1, e2)
        }
    }
}

其实根据实现，我们也比较容易实现一个Zip3Sequence,或者可以接收一个Sequence数组的zip.