定义
运算符重载就是对已有的运算符赋予他们新的含义。重载的修饰符是operator。
比如我们的+号,它的含义是两个数值相加:1+1=2。+号对应的函数名是plus
比如我们定义一个类,如何才能使用 + 这个符号呢?其实也很简单,使用 operator 重载 plus 函数就能搞定,我们试试吧
重载 plus
首先我们定义一个类 Point ,然后重载 plus 函数
class Point(val x: Int,val y: Int) {
operator fun plus(other: Point): Point {
return Point(x + other.x, y + other.y)
}
}
请注意,一定要用 operator 关键字修饰,否则我们是不能用 + 号的。
测试代码:
fun mian() {
val point = Point(1, 1)
val point2 = Point(2, 2)
val point3 = point.plus(point2)
//等价于
val point4 = point + point2
}
我们知道原理后,我们甚至可以吧 Int 的 plus 进行改写,让 Int 类型可以和 Point 对象相加
//自定义plus函数
operator fun Int.plus(other: Point): Int {
return this + other.x + other.y
}
使用
val point = Point(2, 3)
val a = 1
val b = a + point
Log.d("zhaoyanjun", "$b")
重载 get
class Point(val x: Int, val y: Int) {
operator fun get(index: Int): Int {
return index + 1
}
}
使用
val point = Point(2, 3)
val s1 = point.get(9)
//等价于
val s2 = point[9]
重载 compareTo
class User(val name: String, val age: Int) {
//根据年龄做比较,年龄越大,返回正值
operator fun compareTo(other: User): Int {
return age.compareTo(other.age)
}
}
测试代码:
val user1 = User("a", 10)
val user2 = User("b", 20)
val user3 = User("c", 30)
val result = user1 > user2
val result2 = user3 > user2
Log.d("yy", "$result $result2")
结果如下:
D/yy: false true