Kotlin 运算符重载

在 Kotlin 中，我们可以重载运算符，使得我们能够使用适当的语法来操作自定义数据类型。在本文中，我们将介绍如何在 Kotlin 中重载常见的运算符。

什么是运算符重载？

运算符重载是一种特殊的技术，它允许我们对已有的运算符进行自定义操作。简而言之，就是用自定义代码替换默认的运算符行为。

Kotlin 中的运算符重载

在 Kotlin 中，我们可以通过定义函数来重载运算符。以下是一些常见的运算符及其对应的函数：

一元运算符

| 运算符 | 函数 | | --- | --- | | +a | a.unaryPlus() | | -a | a.unaryMinus() | | !a | a.not() |

例如，我们可以按以下方式重载 unaryMinus 函数：

data class Vector(val x: Int, val y: Int) {
    operator fun unaryMinus() = Vector(-x, -y)
}

二元运算符

| 运算符 | 函数 | | --- | --- | | a + b | a.plus(b) | | a - b | a.minus(b) | | a * b | a.times(b) | | a / b | a.div(b) | | a % b | a.rem(b) 或 a.mod(b) | | a..b | a.rangeTo(b) | | a in b | b.contains(a) | | a !in b | !b.contains(a) | | a += b | a.plusAssign(b) | | a -= b | a.minusAssign(b) | | a *= b | a.timesAssign(b) | | a /= b | a.divAssign(b) |

例如，我们可以按以下方式重载 plus 函数：

data class Vector(val x: Int, val y: Int) {
    operator fun plus(other: Vector) = Vector(x + other.x, y + other.y)
}

比较运算符

| 运算符 | 函数 | | --- | --- | | a == b | a?.equals(b) ?: (b === null) | | a != b | !(a?.equals(b) ?: (b === null)) | | a > b | b < a | | a < b | a?.compareTo(b) ?: -1 > 0 | | a >= b | b <= a | | a <= b | a?.compareTo(b) ?: -1 >= 0 |

例如，我们可以按以下方式重载 compareTo 函数：

data class Person(val name: String, val age: Int) : Comparable<Person> {
    operator fun compareTo(other: Person) = compareValuesBy(this, other, Person::age, Person::name)
}

总结

在本文中，我们学习了 Kotlin 中的运算符重载技术，并且学习了如何使用定义的函数来替换默认的运算符行为。这个技术允许我们编写更加自然和直观的代码，使得我们可以更好地处理自定义数据类型。