Swift-技巧（十一）重写运算符

摘要

基础数据的运算可以直接使用四则运算符。在 Swift 中也可以通过重写四则运算符的方式，让 struct 或者 class 创建的结构体或者对象也能像基础数据那样直接使用四则运算符。

Swift 中有经常用到加、减、乘、除的操作，在代码中编写这些操作，实现功能中需要的基础计算。比如下面代码中实现的加法操作。

let a = 1

let b = 2

let c = a + b

// c = 3

当遇到 struct 或者 class 中的变量做计算操作时，一般都是把属性依次拿出来计算，比如两个 CGPoint 的对象相加：

let point1 = CGPoint(x: 1, y: 2)

let point2 = CGPoint(x: 3, y: 4)

// point1 加 point2

let x = point1.x + point2.x

let y = point1.y + point2.y

let result = CGPoint(x: x, y: y)

看代码里面，两个 point 相加要先得到 x 的和，y 的和，然后再创建新的坐标，生成新的坐标。

逻辑上是没有问题的，如果多个地方出现这样的相加操作，就想到把相加操作给封装成一个函数：

func addFunc(with point1: CGPoint, point2: CGPoint) -> CGPoint {

    let x = point1.x + point2.x

    let y = point1.y + point2.y

    return CGPoint(x: x, y: y)

}

之后两个坐标的相加操作就可以直接调用 addFunc 函数：

let result = addFunc(with point1, point2)

这是常规的封装处理，没有毛病，就是有没有更加好的封装方式？比如直接用 + 这个操作呢？

Swift 中恰好有重写运算符的方式，直接在 struct 或者 class 上使用加、减、乘、除这些操作。那么如何重写运算符呢？

这里依旧以相加两个坐标为例，重写加运算符：

static func + (_ point1: CGPoint, _ point2: CGPoint) -> CGPoint {

    return CGPoint(x: point1.x + point2.x, y: point1.y + point2.y)

}

代码中的重写格式一定要是 static func +(属性...)。在函数体中的处理就按照正常的四则运算规则处理。

之后处理两个坐标相加时，就可以像最开始代码中的那样直接用 + 运算。

let result = point1 + point2

其他的减、乘、除等运算也可以依照这样的重写格式去处理。

进阶

坐标是 CGPoint 结构体的，所以和坐标相关的重写的运算符可以写在 CGPoint 的 extension 中，避免重写方法影响到系统级别的运算符

extension CGPoint {

  static func + (_ point1: CGPoint, _ point2: CGPoint) -> CGPoint {

      return CGPoint(x: point1.x + point2.x, y: point1.y + point2.y)

  }

}

这里再实现 += 运算符，说明结构体自身运算后仍赋值到自身的场景，用到的是 inout 修饰。

static func += (point1: inout CGPoint, _ point2: CGPoint) {

    point1.x += point2.x

    point1.y += point2.y

}

使用上和基础数据的 += 方式也是一样的。

题外话

时间仓促，说的东西可能不全面，在你查看的过程中遇到什么问题，评论区给我留言，我会尽快回复

巴特西

Swift-技巧（十一）重写运算符

进阶

题外话

最新文章

热门文章