Kotlin扩展知识（二）：扩展函数和运算符重载

Kotlin扩展知识（一）：延迟初始化和密封类

扩展函数：

扩展函数表示即使在不修改某个类的源码的情况下，仍然可以打开这个类，向该类添加新的函数。

当我们想要实现一个 统计某个字符串中的字母数量 的方法时，之前通常需要创建一个单例类，来实现一个函数，函数参数为字符串，进行统计，如下：

object StringUtil {
    fun lettersCount(str: String): Int {
        var count = 0
        for (char in str) {
            if (char.isLetter()) {
                count++
            }
        }
        return count
    }
}

调用方法：

val str = "ABC123xyz!@#"
val count = StringUtil.lettersCount(str)

但是有了扩展函数之后，我们直接可以在String类中添加一个扩展函数，从而实现功能。

扩展函数的语法结构：

fun ClassName.methodName(param1: Int, param2: Int): Int {
    return 0
}

相比于定义一个普通的函数，定义扩展函数只需要在函数名的前面加上一个ClassName.的语法结构，就表示将该函数添加到指定类当中了。

我们希望向String类中添加一个扩展函数，因此需要先创建一个String.kt文件。文件名虽然并没有固定的要求，但是我建议向哪个类中添加扩展函数，就定义一个同名的Kotlin文件，这样便于你以后查找。当然，扩展函数也是可以定义在任何一个现有类当中的，并不一定非要创建新文件。不过通常来说，最好将它定义成顶层方法，这样可以让扩展函数拥有全局的访问域。

fun String.lettersCount(): Int {
    var count = 0
    for (char in this) {
        if (char.isLetter()) {
            count++
        }
    }
    return count
}

现在我们将lettersCount()方法定义成了String类的扩展函数，那么函数中就自动拥有了String实例的上下文。因此lettersCount()函数就不再需要接收一个字符串参数了，而是直接遍历this即可，因为现在this就代表着字符串本身即可。

调用方法：

val count = "ABC123xyz!@#".lettersCount()

运算符重载：

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

但问题在于这个指定函数是什么？这是运算符重载里面比较复杂的一个问题，因为不同的运算符对应的重载函数也是不同的。比如说加号运算符对应的是plus()函数，减号运算符对应的是minus()函数。

以一个Money类进行加法运算符重载为例：

class Money(val value: Int)

进行运算符重载：

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

我们将当前Money对象的value和参数传入的Money对象的value相加，然后将得到的和传给一个新的Money对象并将该对象返回。这样两个Money对象就可以相加了。

Kotlin允许我们对同一个运算符进行多重重载，可以重载money和一个整数相加，如下：

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)
    }
}

调用方法：

val money1 = Money(5)
val money2 = Money(10)
val money3 = money1 + money2
val money4 = money3 + 20
println(money4.value)

这里的money1 + money2时Kotlin提供的一种语法糖，他在编译时会被转换成：money1.plus(money2)

结合实际情况使用，可以玩出很多花样。

其它的运算符，以及对应的函数为：