谈谈Swift的extension

字数 854阅读 11939

3.times { puts 'hello world' }
这是一条Ruby语句,它会打印“hello world”三次,意图清晰,语法简洁。

如果你是C或者Java等语言的使用者,而从未接触过Ruby,此时是不是有种打开了新世界的大门的感觉_。反正当初我刚接触Ruby那会儿时常会感叹,啊原来代码还可以这么写。。。Ruby的灵活与优雅当然远不仅于此,这里就不展开了。今天的重点是,Swift也可以这么写。

首先我们要明确,Ruby之所以可以这么写是因为它是一个纯面向对象的语言,在Ruby的世界中,一切皆对象,所有的事物都有一个共同的祖先——Object3的类型是Fixnum,当然也是一个对象,所以它有自己的方法。times就是一个方法,可以看出应该是接收一个代码块(本例中为{puts 'hello world'}),然后调用它self(本例中self即为3)次。

而在Swift中,Int是一个struct(结构体),并不是个class。但是Swift中的struct与别的语言中的struct不同的是,struct内部也可以定义方法。所以条件满足了,现在就差个times方法了。

嗯,很遗憾Swift的Int并没有times方法。不过没关系,这个时候extension就派上用场了。没有我们就自己写一个嘛,私人化定制,满足一切突如其来的需求。先上代码好了:
定义:

extension Int {
    func times(closure: () -> ()) {
       for _ in 1...self {
            closure()
        }
    }
}

调用: 3.times { println("hello world") }

是不是跟开头Ruby语句有异曲同工之妙?extension不仅可以扩展方法,还可以扩展计算属性。比如《The Swift Programming Language》中的一个例子:

extension​ ​Double​ {
​    ​var​ ​km​: ​Double​ { ​return​ ​self​ * ​1_000.0​ }
​    ​var​ ​m​: ​Double​ { ​return​ ​self​ }
​    ​var​ ​cm​: ​Double​ { ​return​ ​self​ / ​100.0​ }
​    ​var​ ​mm​: ​Double​ { ​return​ ​self​ / ​1_000.0​ }
​    ​var​ ​ft​: ​Double​ { ​return​ ​self​ / ​3.28084​ }
​}
​let​ ​oneInch​ = ​25.4​.​mm
​println​(​"One inch is ​\(​oneInch​)​ meters"​)
​// prints "One inch is 0.0254 meters"
​let​ ​threeFeet​ = ​3​.​ft
​println​(​"Three feet is ​\(​threeFeet​)​ meters"​)
​// prints "Three feet is 0.914399970739201 meters”

extension在实际的开发中可以有很多应用,使用得当能显著提高代码的可读性,平常我们自己写的大多数辅助类其实都可以转化为extension。譬如我最近在自己的项目中给被点击的UIView(比如button)写了一个小动画,点击了之后组件会上浮一下,并且同时会有一下阴影效果。
定义:

extension UIView {
    func animateWhenClicked() {
        self.backgroundColor = UIColor(white: 0.9, alpha: 0.5)
        self.layer.transform = CATransform3DMakeScale(1, 1, 0)
        UIView.animateWithDuration(0.9, animations: {
            self.backgroundColor = UIColor.whiteColor()
            self.layer.transform = CATransform3DMakeTranslation(1, 1, 1)
        })
    }
}

使用(仅点击动画部分代码):

button.addTarget(viewCtrl, action: "clicked:", forControlEvents: .TouchUpInside)
func clicked(sender: UIButton) {
    sender.superview!.animateWhenClicked()
}
iOS Simulator Screen Shot 2015年8月26日 下午11.39.45.png

大概就是右上角的效果,呃,静态图看不出来,各位要是有兴趣可以拿我的扩展方法然后自己初始化个Button去跑跑看(直接sender.animateWhenClicked()就好,不用superview)。因为我在项目中的Button是一个自定义的贪婪Button,只要点击Button的父视图就会响应点击事件,所以是Button的superview调用了animateWhenClicked,这也是我把这个动画扩展到UIView而不是UIButton的原因。

我个人很喜欢extension,平常也经常使用。extension也可用于后期项目维护,可以在不修改旧代码的前提下扩展旧代码的功能。而且对于NSObject的子类或者有dynamic标记的方法,利用扩展还可以在运行时对某些方法的实现进行替换,类似于OC的Swizzle,今天就不展开了,下次有空继续。

推荐阅读更多精彩内容