Swift 源码解读 - Sequence.swift

Map.swift 源码解读 中,多处涉及到 Sequence 的内容,今天抽时间来阅读下 Sequence.swift 的源码。

Sequence 协议

首先我们大概过一次 Sequence 协议的内容:

public protocol Sequence {
  /// 元素
  associatedtype Element

  /// 迭代器
  associatedtype Iterator : IteratorProtocol where Iterator.Element == Element

  /// 子序列
  associatedtype SubSequence : Sequence = AnySequence<Element>
    where Element == SubSequence.Element,
          SubSequence.SubSequence == SubSequence

  /// 返回当前迭代器
  func makeIterator() -> Iterator
 
  var underestimatedCount: Int { get }

  // 下面是 Sequence extension方法
  func map<T>(
    _ transform: (Element) throws -> T
  ) rethrows -> [T]
  func filter(
    _ isIncluded: (Element) throws -> Bool
  ) rethrows -> [Element]
  func forEach(_ body: (Element) throws -> Void) rethrows

  // 下面还有一大串的方法,但都是扩展方法,这里不一一列举了
}

Sequence 协议中,主要有两个参数, 一个是 Element,也即是 sequence 里的元素,别一个则是 Iterator(迭代器),Iterator 是实现 IteratorProtocol 协议并且迭代器的元素和 sequence 的元素是相同类型(Iterator.Element == Element)。

首先我们先看下 IteratorProtocol 源码:

public protocol IteratorProtocol {
  
  associatedtype Element

  mutating func next() -> Element?
}

IteratorProtocol 的核心是 next() 方法,这个方法在每次被调用时返回序列中的下一个值。当序列下一个值为空时,next() 应该返回 nil。这里我们自定义一个 iterator:

struct CustomIterator: IteratorProtocol {
    mutating func next() -> Int? {
        return 1
    }
}

var iter = CustomIterator()
print(iter.next())   // Optional(1)

首先这里先了解下关键字 mutating,在 swift 中,struct 和 enum 可以定义自己的方法,但是默认情况下,实例方法中是不可以修改值类型的属性,如果需要修改,则需要添加 mutating 关键字。上面的例子并没有修改类型的属性,所以说 mutating 可写也可以不写, xcode 默认提示是带上的。这里再来一个标准点的写法:

struct CustomIterator: IteratorProtocol {
    var num = 0
    mutating func next() -> Int? {
        num += 1
        if num == 10 {
            return nil
        }
        return num
    }
}

var iter = CustomIterator()
while let num = iter.next() {
    print(num)  // 1,2,3,4,5,6,7,8,9
}

大概了解 Iterator 之后,我们再回到 Sequence,我们如果需要自定义一个 sequence 时,其实只需要实现 makeIterator 方法就可以了,如:

struct CustomIterator: IteratorProtocol {
    var num = 0
    mutating func next() -> Int? {
        num += 1
        if num == 10 {
            return nil
        }
        return num
    }
}

struct CustomSequence: Sequence {
    func makeIterator() -> CustomIterator {
        return CustomIterator()
    }
}

let sequence = CustomSequence()
for item in sequence {
    print(item)   // 1,2,3,4,5,6,7,8,9
}

关于 Sequence 最基本的内容大概就是这一些了,而 Sequence 里面定义了一大串的方法,如果 map、dropFirst、dropLast 等等,Swift 中通过协议扩展的形式来实现,如果在自定义 sequence 时需要修改相应的方法,就需要我自己重写,如果无需修改,直接使用则可。这里不一一细说,建议有时间可以完整读下,欢迎交流学习。

推荐阅读
Swift 解读 - 前言
Swift 解读 - Collection 大家族(上篇)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 196,583评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,669评论 2 374
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,684评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,682评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,533评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,396评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,814评论 3 387
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,458评论 0 255
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,745评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,789评论 2 314
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,565评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,410评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,828评论 3 300
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,050评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,342评论 1 253
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,793评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,010评论 2 337

推荐阅读更多精彩内容