Kotlin 手记 -- 运算符重载

运算符重载使用的是operator关键字,只要在指定函数的前面加上operator关键字,就可以实现运算符重载的功能了。

不同的运算符对应的重载函数也是不同的。
比如说加号运算符对应的是plus()函数,
减号运算符对应的是minus()函数。

如果想要实现让两个对象相加的功能,那么它的语法结构如下:

class Obj {
    operator fun plus(obj: Obj): Obj {
        // 处理相加的逻辑
    }
}

对应的调用方式如下:

val obj1 = Obj()
val obj2 = Obj()
val obj3 = obj1 + obj2

这种obj1 + obj2的语法看上去好像很神奇,
但其实这就是Kotlin给我们提供的一种语法糖,
它会在编译的时候被转换成obj1.plus(obj2)的调用方式。

对同—个运算符进行多重重载

class Money(val value: Int) {
    operator fun plus(money: Money): Money {
        val sum = value + money.value
        return Money(sum)
    }

    operator fun plus(newValue: Int): Money {
        val sum = value + newValue
        return Money(sum)
    }
}

Kotlin允许我们重载的运算符和关键字多达十几个

如果我们能够使用str * n这种写法来表示让str字符串重复n次,
在Kotlin中这是可以实现的。

operator fun String.times(n: Int): String {
    val builder = StringBuilder()
    repeat(n) {
        builder.append(this)
    }
    return builder.toString()
}

必须说明的是,其实Kotlin的String类中已经提供了—个用于将字符串重复n遍的repeat()函数,
因此times()函数还可以进—步精简成如下形式:

operator fun String.times(n: Int) = repeat(n)