Kotlin之扩展函数和运算符重载

1、扩展函数

扩展函数就是在不修改源码的情况下，向某个类中新增加函数。下面举个例子：我们想统计一个字符串中字母的数量，我们一般的写法如下：

 fun getLettersCount(str:String):Int{
        var count=0
        for(chr in str){
            if(chr.isLetter())
                count++
        }
        return count
    }

这种写法绝对可以正常工作，但是有了扩展函数之后就不一样了，我们可以使用一种更加面向对象的方式来实现这个功能，先来看下扩展函数的语法结构：

fun ClassName.methodName(params1:Int,params:Int):Int{}

可以看到和普通的方法类似，只不过在方法名之前加上了ClassName.，学习了语法结构后我们来实现一下上面的功能。

• 1、新建文件String.kt
  文件名不是固定的，只不过我们给String类新增扩展函数，所以命名为String更方便查找和维护。
• 2、添加顶层方法

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

可以看到给String类新增扩展函数，那么函数中就自动拥有了String实例的上下文，因此函数不再接收已String参数了，直接遍历本身即可，this就代表着字符串本身。
注意：扩展函数不一定非要定义成顶层函数，可以定义在任何一个类中。

• 3、调用

 "Start eating fruit".getLettersCount()

除了String类我们可以向任意类添加扩展函数。

2、运算符重载

Java中有许多内置的运算符关键字：+ - * / % ++ --，在Kotlin中允许对所有的运算符甚至其他的关键字进行重载，在进行运算符 重载后我们可以实现任意对象进行相加或者其他更多的操作。
运算符重载使用的是operator关键字，只要在指定的函数前加上operator关键字就可以实现运算符的重载，指定的函数指的是什么？下面我们看张表

语法糖表达式	指定函数
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++	a.inc()
a--	a.dec()
+a	a.unaryPlus()
-a	a.unaryMinus()
!a	a.not()
a==b	a.equals(b)
a>b	a.compareTo(b)
a..b	a.rangeTo(b)
a[b]	a.get(b)
a[b]=c	a.set(b,c)
a in b	b contains(a)

我们这里以加号运算符为例，来实现两个对象相加的功能。

class Money(val value: Int) {
     //运算符重载
    operator fun plus(money: Money): Money =Money(value+money.value)
}
//调用
 val money:Money=Money(12)+Money(23)

Kotlin允许我们对同一个运算符进行多重重载，下面我们再重载plus函数

class Money(val value: Int) {
     //运算符重载
    operator fun plus(money: Money): Money =Money(value+money.value)
    operator fun plus(newValue:Int):Money=Money(value+newValue)
}
//调用 他们输出结果一致
val money:Money=Money(12)+ Money(23)
val newMoney:Money=Money(12)+23

下面再看一个实例：

fun getRandomLengthString(str:String):String{
        val stringBuilder=StringBuilder()
        val n=Random().nextInt(20)+1
        repeat(n){
            stringBuilder.append(str)
        }
        return stringBuilder.toString()
    }

上面代码的功能就是让字符串重复n遍，那么我们能不能直接使用str*n来实现呢？
由于要实现str*n它对应的指定函数是str.times(n)，所以就要把函数定义在String类中，此时我们就可以借助我们刚刚学习的扩展函数，然后结合运算符重载，定义如下：

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

由于String内部实现了repeat(n)用于将字符串重复n遍的函数，所以上面代码可简化为：

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

调用如下：

str*(Random().nextInt(20)+1)