1 一元运算符函数

1.1 符号和函数

1.2 案例

fun main() {
    var stu = Student("Tom", 13)
    println(-stu) // 打印: [moT, 31]
}

class Student(var name: String, var age: Int) {
    operator fun unaryMinus(): Student {
        return Student(name.reversed(), age.toString().reversed().toInt())
    }

    override fun toString(): String {
        return "[$name, $age]"
    }
}

2 二元运算符函数

2.1 基础运算符

2.1.1 符号和函数

2.1.2 案例

fun main() {
    var stu1 = Student("Tom", 13)
    var stu2 = Student("Mary", 18)
    println(stu1 + stu2) // 打印: [TomMary, 31]
}

class Student(var name: String, var age: Int) {
    operator fun plus(other: Student): Student {
        return Student(this.name + other.name, this.age + other.age)
    }

    override fun toString(): String {
        return "[$name, $age]"
    }
}

2.2 自增简化运算符

2.2.1 符号和函数

说明：如果类中同时定义了 plus 和 plusAssign 运算，a += b 就会产生歧义，因为 a += b 等价于 a = a + b，编译器不知道是执行 plus 函数还是 plusAssign 函数，就会编译报错，其他运算符同理。

2.2.2 案例

fun main() {
    var stu1 = Student("Tom", 13)
    var stu2 = Student("Mary", 18)
    stu1 += stu2
    println(stu1) // 打印: [TomMary, 31]
}

class Student(var name: String, var age: Int) {
    operator fun plusAssign(other: Student): Unit {
        this.name += other.name
        this.age += other.age
    }

    override fun toString(): String {
        return "[$name, $age]"
    }
}

2.3 比较运算符函数

2.3.1 符号和函数

2.3.2 案例

fun main() {
    var stu1 = Student("Tom", 13)
    var stu2 = Student("Mary", 18)
    var res = stu1 >= stu2
    println(res) // 打印: false
}

class Student(var name: String, var age: Int) {
    operator fun compareTo(other: Student): Int {
        return this.age - other.age
    }
}

3 括号运算符函数

3.1 小括号运算符函数

3.1.1 符号和函数

3.1.2 案例

fun main() {
    var stu = Student("Mary", 18)
    stu() // 打印: Mary
    var age = stu(1) // 打印: a
    println(age) // 打印: 18
}

class Student(var name: String, var age: Int) {
    operator fun invoke(): Unit {
        println(name)
    }

    operator fun invoke(i: Int): Int {
        println(name[i])
        return age
    }
}

3.2 中括号运算符函数

3.2.1 符号和函数

3.2.2 案例

fun main() {
    var stu = Student("Mary")
    println(stu[1]) // 打印: a
    stu[1] = 'W'
    println(stu.name) // 打印: MWry
}

class Student(var name: String) {
    operator fun get(i: Int): Char {
        return name[i]
    }

    operator fun set(i: Int, c: Char): Unit {
        name = name.substring(0, i) + c + name.substring(i + 1)
    }
}

4 迭代器运算符函数

迭代器运算符即：for(item in items) 运算符，有以下两种实现方式。

• 继承 Iterator 接口，实现 hasNext 和 next 函数。
• 重载 iterator 函数，返回一个 Iterator 对象。

4.1 继承 Iterator 接口

fun main() {
    var group = Group()
    for (stu: Student in group) {
        println(stu) // 打印: [Tom, 20]、[Mary, 18]
    }
}

class Group: Iterator<Student> {
    private var students = arrayOf(Student("Tom", 20), Student("Mary", 18))
    private var index = 0

    override operator fun hasNext() = index < students.size
    override operator fun next() = students[index++]
}

class Student(var name: String, var age: Int) {
    override fun toString(): String = "[$name, $age]"
}

4.2 重载 iterator 函数

fun main() {
    var group = Group()
    for (stu: Student in group) {
        println(stu) // 打印: [Tom, 20]、[Mary, 18]
    }
}

class Group {
    private var students = arrayOf(Student("Tom", 20), Student("Mary", 18))

    operator fun iterator(): GroupIterator = GroupIterator()

    inner class GroupIterator: Iterator<Student> {
        private var index = 0

        override operator fun hasNext() = index < students.size
        override operator fun next() = students[index++]
    }
}

class Student(var name: String, var age: Int) {
    override fun toString(): String = "[$name, $age]"
}

5 解构函数

5.1 解构属性

fun main() {
    var stu = Student("Tom", 13)
    var (name, age) = stu
    println("$name, $age") // 打印: Tom, 13
    var (_, age2) = stu // 只需要部分参数, 其他参数可以使用_忽略掉
}

class Student(var name: String, var age: Int) {
    operator fun component1() = name
    operator fun component2() = age
}

5.2 解构在 Lambda 表达式中的使用

Lambda 表达式详细介绍见 → Lambda表达式。

fun main() {
    var stu = Student("Tom", 13)
    var myFun: (Student) -> Unit = {(name, age) ->
        println("$name, $age")
    }
    myFun(stu) // 打印: Tom, 13
}

class Student(var name: String, var age: Int) {
    operator fun component1() = name
    operator fun component2() = age
}

6 中缀函数

中缀函数是使用 infix 关键字标记的函数，在使用时，可以省略点和括号，如：位运算 shl 就是一个中缀函数。函数必须满足以下条件。

• 必须是成员函数（不是顶层函数）；
• 只能有一个参数；
• 参数不能有默认值。

中缀函数调用的优先级低于算术运算符、类型转换和 rangeTo 运算符，高于布尔运算符（&&、||、is）。

fun main() {
    var stu = Student("Tom")
    stu play "basketball"
}

class Student(var name: String) {
    infix fun play(str: String): Unit {
        println("$name play $str")
    }

    fun test() {
        // play "badminton" // 编译报错
        this play "badminton"
    }
}

说明： 如果在类中使用中缀函数，必须明确函数的调用方（接收器）。

声明：本文转自【Kotlin】运算符函数、解构函数、中缀函数。