Kotlin语言一大特性就是支持函数扩展,你可以通过一个简单的String对象发现,它比Java中的String对象强大太多,而这是怎么做到的呢?来看今天要讲的扩展函数。
扩展函数
先来个栗子,看看Kotlin中的String类有多强大。
val str = "adafjaidjadgadkjgaf"
str.filter { it != ' ' }
.replace('b', 'd')
.groupBy {
if (it > 'h') 1 else 2
}
.forEach { key, value ->
println("key:$key value:$value")
}
复制代码
函数式调用结合了String类的过滤(filter)、替换(replace)和分组(group),然而这些工具方法并不在String类中,实际上是在StringsKt类中,以filter为例:
# StringsKt.kt
public inline fun String.filter(predicate: (Char) -> Boolean): String {
return filterTo(StringBuilder(), predicate).toString()
}
复制代码
通过上面的代码,我们开始学习扩展函数的写法。 与函数的声明几乎一样,唯一不同的是需要在方法名前加类名表示是为哪个类做的扩展。 下面的例子展示了为String的一个扩展--所有字符串都会sayHello。
fun String.sayHello(): String {
return "hello $this"
}
//test
println("lili".sayHello())
//输出
hello lili
复制代码
在扩展函数体内通过this关键字获取函数的调用者,如果调用方与扩展函数不在一个包中,需要import包名+扩展函数名。
反编译代码可以看到扩展方法的java实现是,生成一个类,类名为文件名+Kt,在这个类中生成一个静态方法。当调用者使用扩展方法时编译器会将此调用转为静态调用。
//反编译结果
public final class TestExtensionKt {
@NotNull
public static final String sayHello(@NotNull String $receiver) {
Intrinsics.checkParameterIsNotNull($receiver, "receiver$0");
return "hello " + $receiver;
}
}
//调用
TestExtensionKt.sayHello("lili");
复制代码
扩展属性
Kotlin中属性也支持扩展,来看下面的例子,为List对象扩展一个lastIndex属性。