Swift中的类型抹消

首先献上本文的Demo


标准库中的类型抹消

不知道大家有没有听说过Swift中的类型抹消,所谓类型抹消就是不将某实例的真实类型暴露出去,对外只暴露一个必要的类型。举个:当我们编写一个class或者struct并实现了一个协议,当我们对外提供该实例时,只想让外界知道这个东西实现了该协议,可是又不想让外界知道实现了这个协议的class或者struct是哪一个类型的,这时我们就需要用到类型抹消,其实在Swift的标准库就有这样的东西。

struct FibsIterator: IteratorProtocol {
    var status = (0, 1)
    mutating func next() -> Int? {
        guard status.0 < 100 else { return nil }
        let num = status.0
        self.status = (status.1, status.0 + status.1)
        return num
    }
}

struct SquareIterator: IteratorProtocol {
    var status = 1
    mutating func next() -> Int? {
        guard status < 100 else { return nil }
        defer {
            status = status * 2
        }
        return status
    }
}
// fibsIterator的type: AnyIterator<Int>
let fibsIterator = AnyIterator(FibsIterator())
// squareIterator的type: AnyIterator<Int>
let squareIterator = AnyIterator(SquareIterator())
// anyIteratorArray的type: [AnyIterator<Int>]
let anyIteratorArray = [fibsIterator, squareIterator]
// fibsSequence的type: AnySequence<Int>
let fibsSequence = AnySequence { return FibsIterator() }
// 这里因为实现一个Collection协议比较麻烦就直接用数组好了
// c的type: AnyCollection<Int>
let c = AnyCollection(Array<Int>())

关于IteratorProtocolSequenceCollection这里不做介绍......

在我们实现了一个斐波拉契的迭代器后,我们想让外界使用该迭代器来产生斐波拉契序列,可是我们又不想让外界知道我这个东西是是用FibsIterator这个struct来实现的。于是我们就用AnyIteratorAnySequenceAnyCollection来包装它,外界使用者只知道迭代这个东西时是产生一个Int类型的迭代器、序列或集合。这样我们就将FibsIterator的类型给隐藏了。


尝试造轮子

既然我们知道类型抹消这个小技巧,那我们就来学造轮子。首先我们就模仿标准库来实现一个抹消类型的迭代器,首先我们先按照上面的套路定义一个协议和实现两个迭代器:

protocol MyIteratorProtocol {
    associatedtype Element
    mutating func next() -> Self.Element?
}

struct FibsIterator: MyIteratorProtocol {
    var status = (0, 1)
    mutating func next() -> Int? {
        guard status.0 < 100 else { return nil }
        let num = status.0
        self.status = (status.1, status.0 + status.1)
        return num
    }
}

struct SquareIterator: MyIteratorProtocol {
    var status = 1
    mutating func next() -> Int? {
        guard status < 100 else { return nil }
        defer {
            status = status * 2
        }
        return status
    }
}

然后我们来实现迭代器包装的盒子,对外只暴露迭代器next()方法放回的类型(这里以Int类型为例),而不暴露实现的迭代器类型,一开始我们可能会写出这样的代码:

class IteratorBox<I: MyIteratorProtocol> {
    var iterator: I
    init(_ iterator: I) {
        self.iterator = iterator
    }
    func next() -> I.Element? {
        return iterator.next()
    }
}

// fibsIteratorBox的type: IteratorBox<FibsIterator>
let fibsIteratorBox = IteratorBox(FibsIterator())
// squareIteratorBox的type: IteratorBox<SquareIterator>
let squareIteratorBox = IteratorBox(SquareIterator())
// 报错!类型不同,所以Boxs需要转成[Any]
let Boxs = [fibsIteratorBox, squareIteratorBox]❌

在这里出现了两个问题:
第一个问题是我们定义的泛型I直接就暴露出去了,这样就违背了我们的初衷,我们只想要外界看到的是I.Element的类型,而不是暴露I的类型。
第二个问题是,用数组存放这些IteratorBox时,由于泛型不相同而报错。


方法一:通过属性保存迭代器方法的实现

首先我们明确知道为了防止实现MyIteratorProtocol的类型被暴露,IteratorBox的泛型应为MyIteratorProtocolElement类型,而不是MyIteratorProtocol的类型。其次我们也必须要IteratorBox实现MyIteratorProtocol协议,确保IteratorBox包含完整的功能。

struct IteratorBox<A>: MyIteratorProtocol {
    private var nextIMP: () -> A?
    init<I: MyIteratorProtocol>(_ iterator: I) where I.Element == A {
        var iteratorCopy = iterator // Swift中参数被隐式声明为let
        self.nextIMP = { iteratorCopy.next() }
    }
    func next() -> A? {
        return nextIMP()
    }
}
// fibsIteratorBox的type: IteratorBox<Int>
let fibsIteratorBox = IteratorBox(FibsIterator())
// squareIteratorBox的type: IteratorBox<Int>
let squareIteratorBox = IteratorBox(SquareIterator())
// Boxs的type: [IteratorBox<Int>]
let Boxs = [fibsIteratorBox, squareIteratorBox]

这样的方法实现起来是非常简单方便的,也实现了标准库中同样的效果。但这样只适合方法较少的类型,因为MyIteratorProtocol每多一个方法就要多一个属性来保存,而且在init方法中也需要编写额外的代码。
MyIteratorProtocol方法数量多时我们只能像上面那样保存实现MyIteratorProtocol的迭代器,通过迭代器来调用。但是上面造轮子处不是尝试过了吗?而且还问题多多呢...所以我们来看看接下来的方法。


方法二:利用面向对象中的继承和多态特性

首先我们还是要明确刚才提及的需求。我们实现的IteratorBox的泛型应为MyIteratorProtocolElement类型,而为了保存实现MyIteratorProtocol的迭代器我们又要将泛型约束为实现MyIteratorProtocol的类型,这样就会造成冲突。既然一个类不能同时满足这两个类型,那我们就用两个类来实现好了。

class IteratorBox<A>: MyIteratorProtocol {
    func next() -> A? {
        fatalError("This method is abstract, you need to implement it!")
    }
}

首先我们让IteratorBox实现MyIteratorProtocol中的所有方法,而在方法的实现里我们仅仅让它崩溃。接下来我们实现一个IteratorBoxHelper来保存迭代器,并实现协议里的方法:

class IteratorBoxHelper<I: MyIteratorProtocol> {
    var iterator: I
    init(_ iterator: I) {
        self.iterator = iterator
    }
    func next() -> I.Element? {
        return iterator.next()
    }
}

最后我们让IteratorBoxHelper继承自IteratorBox

class IteratorBoxHelper<I: MyIteratorProtocol>: IteratorBox<I.Element> {
    var iterator: I
    init(_ iterator: I) {
        self.iterator = iterator
    }
    override func next() -> I.Element? {
        return iterator.next()
    }
}

继承IteratorBox时我们将泛型A的类型约束为IteratorBoxHelperIElement的类型。
我们通过IteratorBoxHelper来包装迭代器后,将该类型声明为IteratorBox,也就所谓的向上转型(将子类类型声明为父类类型),然后我们来调用这些迭代器,调用方法时会根据多态性的特点调用真实类型的方法。

// fibsIteratorBox的type: IteratorBox<Int>
let fibsIteratorBox: IteratorBox = IteratorBoxHelper(FibsIterator())
// squareIteratorBox的type: IteratorBox<Int>
let squareIteratorBox: IteratorBox = IteratorBoxHelper(SquareIterator())
// Boxs的type: [IteratorBox<Int>]
let Boxs = [fibsIteratorBox, squareIteratorBox]

print("调用fibsIteratorBox的next方法")
while let num = fibsIteratorBox.next() {
    print(num)
}
print("调用squareIteratorBox的next方法")
while let num = squareIteratorBox.next() {
    print(num)
}

包装后不再具有值语义

在这里提一点是,我们的FibsIteratorSquareIterator都是struct,具有值语义,而当我们将它赋值给另一个变量后再迭代就会进行写时复制:

var fib1 = FibsIterator()
print(fib1.next()!) // 0
print(fib1.next()!) // 1
print(fib1.next()!) // 1
var fib2 = fib1
print(fib1.next()!) // 2
print(fib1.next()!) // 3
print(fib2.next()!) // 2
print(fib2.next()!) // 3

对于方法一,虽然IteratorBox的类型和标准库中的AnyIterator同为struct,当它们两个都并不具备写时复制的特性,IteratorBox的闭包属性捕获了iteratorCopy,即使赋值给另一个变量,闭包属性也还是引用用一个iteratorCopy
对于方法二,由于IteratorBoxIteratorBoxHelper都是class类型,那么就会变成了引用语义,给另一变量赋值时只是指针拷贝:

let fibsIteratorBox: IteratorBox = IteratorBoxHelper(FibsIterator())
print(fibsIteratorBox.next()!) // 0
print(fibsIteratorBox.next()!) // 1
print(fibsIteratorBox.next()!) // 1
let fibsBox2 = fibsIteratorBox
print(fibsIteratorBox.next()!) // 2
print(fibsIteratorBox.next()!) // 3
print(fibsBox2.next()!)        // 5
print(fibsBox2.next()!)        // 8

至此我们就成功模仿了Swift标准库中AnyIterator的实现。如有错漏,请多指教。

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

推荐阅读更多精彩内容

  • 136.泛型 泛型代码让你可以写出灵活,可重用的函数和类型,它们可以使用任何类型,受你定义的需求的约束。你可以写出...
    无沣阅读 1,463评论 0 4
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,957评论 25 707
  • 本章将会介绍 泛型所解决的问题泛型函数类型参数命名类型参数泛型类型扩展一个泛型类型类型约束关联类型泛型 Where...
    寒桥阅读 633评论 0 2
  • ⒈A同学问:老师,B同学哭啦!你为什么不去安慰她? 答:我不喜欢爱哭的学生! 问:为什么呀?爱哭的女生一般不都比较...
    李豫一阅读 922评论 0 0
  • 2017年11月19日,星期天,今天是杨峰瑞同学跆拳道晋级考试的日子,今天早晨从家出发的时候,给他鼓劲,...
    航航2阅读 161评论 0 0