logo头像
Snippet 博客主题

Kotlin基础之内联函数

内联函数

使用高阶函数会给运行时带来一些坏处:每个函数都是一个对象,捕获闭包(如:访问函数体内的变量),内存分配(函数对象或Class),虚拟调用引入的运行过载。 使用内联Lambda表达式在多数情况下可以消除这种过载。比如下面的函数就是这种情况下的很好的例子,lock()函数可以很容易地在调用点进行内联扩展。

1
lock(l){ foo() }

编译能够产生下面的代码,而不是创建一个函数对象参数,生成调用。

1
2
3
4
5
6
7
l.lock()
try {
foo()
}
finally {
l.unlock()
}

也是我们一开始想要的。 为了让编译器能够这样执行,需要用inline修饰符来标记lock函数。

1
2
3
inline fun lock<T>(lock: Lock , body: () -> T): T{
...
}

inline修饰符既影响函数对象本身,也影响传入的Lambda参数:两者都会被内联到调用点。

编译预处理器对内联函数进行扩展,省去了参数压栈、生成汇编语言的CALL调用、返回参数、执行return等过程,从而提高了运行速度。 使用内联函数的优点,在函数被内联后编译器就可以通过上下文相关的优化技术对结果代码执行更深入的优化。 内联不是万能药,它以代码膨胀为代价,仅仅省去了函数调用的开销,从而提高程序的执行效率。

说明:函数调用开销并不包括执行函数体所需要的开销,而是仅指参数压栈、跳转、退栈和返回等操作。如果执行函数体内代码的时间比函数调用的开销大得多,那么内联函数的效率收益会笑很多。另一方面每一处内联函数的调用都要拷贝代码,将使程序的总代码增大、消耗更多的内存空间。

noinline

如果只需要在内联函数中内联部分Lambda表达式,可以使用noinline来标记不需要内联的参数。

1
2
3
inline fun foo(inlined: () -> Unit, noinline notInlined: () -> Unit) {
// ...
}

内联Lambda只能在内联函数中调用或作为内联参数,但noinline的Lambda可随意使用。
说明:没有内联函数参数和reified type parameters的内联函数,编译器会发出警告,因为内联这样的函数不见得有好处。

非局部返回

在Kotlin中可以使用正常、无条件的return退出有名和匿名函数,也意味需要使用一个标签来退出Lambda,在Lambda中禁止使用赤裸return语句,因为Lambda不能够使闭合函数返回。

1
2
3
4
5
6
fun foo(){
ordinaryFunction{
return // ERROR: can not make `foo` return here
}
}

如果Lambda传入内联函数,则返回也是被内联,所以被允许。

1
2
3
4
5
fun foo(){
inlineFunction {
return // OK: the lambda is inlined
}
}

这样的return(位于在Lambda中,但能够退出闭合函数)被称为非局部返回。Kotlin使用这种构造在有循环条件的闭合内联函数中。

1
2
3
4
5
6
7
fun hasZeros(ints: List<Int>): Boolean{
ints.forEach{
if(it == 0) return true // returns from hasZeros
}
return false
}

一些内联函数可能不是从函数体中直接调用传入的Lambda参数,而是从其他的执行上下文,如本地对象或嵌套函数。在这些情况下,non-local 控制流则不允许出现在Lambda中。使用crossinline修饰符来标记。

1
2
3
4
5
6
inline fun f(crossinline body: () -> Unit) {
val f = object: Runnable {
override fun run() = body()
}
// ...
}

具体化类型参数

有时需要访问传入函数中参数的类型。例如:

1
2
3
4
5
6
7
8
fun <T> TreeNode.findParentOfType(clazz: Class<T>): T? {
var p = parent
while (p != null && !clazz.isInstance(p)) {
p = p.parent
}
@Suppress("UNCHECKED_CAST")
return p as T?
}

在上述代码中,沿着树结构,使用反射来检查节点是否有指定类型。

1
treeNode.findParentOfType(MyTreeNode::class.java)

实际上想要只是简单给函数传入一个类型,如:

1
treeNode.findParentOfType<MyTreeNode>()

内联函数支持具体化参数类型,因此可以这样写:

1
2
3
4
5
6
7
inline fun <reified T> TreeNode.findParentOfType(): T? {
var p = parent
while (p != null && p !is T) {
p = p.parent
}
return p as T?
}

使用reified修饰符限制参数类型,可以在内联函数中访问,就像是普通的Class。因为函数是内联的,不在需要反射,像!is和as的普通操作符执行。也可以像上述说的那样调用。

1
myTree.findParentOfType<MyTreeNodeType>()

尽管反射在很多情况不需要,仍需要使用它来具体话参数类型。

1
2
3
4
5
inline fun <reified T> membersOf() = T::class.members
fun main(s: Array<String>) {
println(membersOf<StringBuilder>().joinToString("\n"))
}

内联属性

inline修饰符可以用在没有Backing Filed属性的访问函数。可以注解单独属性的访问函数。

1
2
3
4
5
6
val foo: Foo
inline get() = Foo()
var bar: Bar
get() = ...
inline set(v) { ... }

甚至可以注解整个属性,让属性访问函数都变为内联函数。

1
2
3
inline var bar: Bar
get() = ...
set(v) { ... }

在调用时,内联访问函数与常规内联函数调用方式一样。

支付宝打赏 微信打赏

如果文章对你有帮助,欢迎点击上方按钮打赏作者

上一篇