Swift 运算符重载简介

作者:COSMIN PUPĂZĂ,原文链接,原文日期:2016-03-29
译者:zltunes;校对:shanks;定稿:小锅

在任何一门计算机编程语言中,运算符重载都是非常强大的特性之一,因此苹果决定为 Swift 也提供这一机制。然而,"能力越强责任越大"。利用运算符重载你很容易实现一些奇怪的场景,例如用减法运算符实现两数相加,或者用乘法运算符实现两数相除,但这显然都不是你希望出现的。

好了,闲话少叙 —— 让我们看看运算符重载究竟是怎么一回事。

挑战

这一小节的任务很简单:扩展乘法运算符的标准功能,使其适用于字符串。你将会用到字符串拼接运算符,想象一下这种用法:

"abc" * 5 = "abc" + "abc" + "abc" + "abc" + "abc" = "abcabcabcabcabc"

正式编码之前,思考一下应该怎么做,分几步来实现。我的做法是这样的:

  • 定义变量 result 并进行初始化 —— 默认字符串。
  • 从 2 开始循环,一直到待拼接的字符串数目终止,每次迭代只做一件事 —— 把字符串拼接到 result 末尾。
  • 打印 result

算法大致就是这样,接下来让我们付诸实践。

基本运算符重载


启动 Xcode 并新建一个 playground 文件。删除原有代码,添加乘法运算符的函数原型:

func *(lhs: String, rhs: Int) -> String {
 
}

函数有两个参数 —— 左操作数是 String 类型,右操作数是 Int 类型,函数返回类型为 String
函数体内应该完成三件事。首先,定义 result 变量并初始化为函数的 String 参数 —— 这是一个变量,稍后会修改它的值。

var result = lhs

接下来使用 for in 控制流语句及闭区间运行符从 2 开始循环,直到函数的 Int 参数时为止:

for _ in 2...rhs {
 
}

注意:这里使用了 _ 作为通配符,因为我们希望忽略序列的具体值 —— 关于循环的更多说明可以看这里

循环体内只有一个任务 —— 更新 result

result += lhs

注意:你也可以按如下方式来写 —— 上边这种写法更短,是因为用了加法复合运算符。

result = result + lhs

最后返回 result:

return result

现在我们直接使用运算符:

let u = "abc"
let v = u * 5

搞定了!只是还有一个问题 —— 你只能将其用于字符串,那其它类型的数据怎么办?我们使用范型运算符来完善。

泛型运算符


泛型默认是不支持运算符的,所以需要协议来支持。向 playground 中添加协议原型:

protocol Type {
 
}

现在向协议中添加加法复合运算符函数的原型:

func +=(inout lhs: Self, rhs: Self)

函数拥有左右操作数,并且都设置为 Self 类型 —— 这是一种巧妙的方式,说明二者的类型都是实现了该协议的类。左操作数标记为inout,因为它的值是要被修改并且最后被函数返回的。

或者,你也可以定义加法运算符的函数原型:

func +(lhs: Self, rhs: Self) -> Self

函数拥有 Self 类型的左右操作数,并且加法运算的返回结果也是 Self 。这种情况下就不需要使用 inout 参数了。

接下来,为 String , Int , Double , Float 等实现了 Type 协议的类型创建扩展。

extension String: Type {}
extension Int: Type {}
extension Double: Type {}
extension Float: Type {}

注意:这些扩展的实现是空的,因为你并不打算为默认类型添加任何东西,仅仅是要让他们遵循 Type 协议。

现在向 playground 中添加乘法操作符函数原型:

func *<T: Type>(lhs: T, rhs: Int) -> T {
 
}

函数有两个参数,左操作数是 T 类型,右操作数是 Int 类型,函数返回类型为 T 。利用类型约束使 T 类型遵循 Type 协议,这样它就可以使用加法复合运算符了。

注意:你可以使用 where 关键字定义类型约束——尽管上边的方法更简短:

func *<T where T: Type>(lhs: T, rhs: Int) -> T

函数的实现跟之前一样:

var result = lhs
 
for _ in 2...rhs {
 
    result += lhs
    
}
 
return result

注意:可以使用加法操作符替代,但要确保它的函数原型添加到了协议中。
测试一下:

let x = "abc"
let y = x * 5
 
let a = 2
let b = a * 5
 
let c = 3.14
let d = c * 5
 
let e: Float = 4.56
let f = e * 5

搞定了!不过有一个问题:你使用的是标准乘法运算符,这个可能造成歧义。如果换成其它运算符会更好。接下来我们试着用自定义运算符解决这个问题。

自定义运算符

首先添加下面一行到 playground:

infix operator ** {associativity left precedence 150}

一步一步解释:

  • 自定义乘法运算符的名称是 **。
  • 类型是 中缀运算符(infix) 因为它有两个操作数。
  • 运算顺序从左至右,因此是左结合。
  • 优先级设置为 150 —— 与标准乘法运算符相同,因为它是高优先级运算符。

注意:关于运算符优先级和结合性的更多说明可以看这里

自定义运算符的函数原型与标准运算符类似 —— 只有函数名不同:

func **<T: Type>(lhs: T, rhs: Int) -> T {
 
}

函数实现跟之前完全一样:

var result = lhs
 
for _ in 2...rhs {
 
    result += lhs
    
}
 
return result

测试一下:

let g = "abc"
let h = g ** 5
 
let i = 2
let j = i ** 5
 
let k = 3.14
let l = k ** 5
 
let m: Float = 4.56
let n = m ** 5

搞定了!还有一个问题——运算符的复合类型还没有定义,接下来我们解决这个问题:

复合运算符


复合运算符的类型、优先级和结合性和之前一样 —— 只有名称不同:

infix operator **= {associativity left precedence 150}

接着向 playground 添加复合运算符的函数原型:

func **=<T: Type>(inout lhs: T, rhs: Int) {
 
}

函数没有返回类型,因为左操作数被标记为 inout
函数体只做一件事 —— 运用之前的自定义运算符返回乘法结果:

lhs = lhs ** rhs

测试一下:

var o = "abc"
o **= 5
 
var q = 2
q **= 5
 
var s = 3.14
s **= 5
 
var w: Float = 4.56
w **= 5

搞定了!这已经是最简版本了!

总结

如果谨慎地使用运算符重载,它便能够发挥强大的功能 —— 我希望你能在自己的项目中找到合适使用的方法。
作为参考,这里有一个完整版的 Playground,我已经在 Xcode 7.3 用 Swift 2.2 测试过了。

如果你对本教程或者运算符重载有什么看法的话可以给我留言。

致谢:连环画是在 MakeBeliefsComix.com 制作的。
本文由 SwiftGG 翻译组翻译,已经获得作者翻译授权,最新文章请访问 http://swift.gg

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

推荐阅读更多精彩内容