Kotlin 内联函数

一、内联函数原理

使用高阶函数为开发带来了便利,但同时也产生了一些性能上的损失,官方是这样描述这个问题:

使用高阶函数会带来一些运行时的效率损失:每一个函数都是一个对象,并且会捕获一个闭包。 即那些在函数体内会访问到的变量。 内存分配(对于函数对象和类)和虚拟调用会引入运行时间开销,但是通过内联化 Lambda 表达式可以消除这类的开销。

为了解决这个问题,可以使用内联函数,用inline修饰的函数就是内联函数,inline修饰符影响函数本身和传给它的 Lambda 表达式,所有这些都将内联到调用处,即编译器会把调用这个函数的地方用这个函数的方法体进行替换,而不是创建一个函数对象并生成一个调用。

接下来用代码验证这个说法,先定义一个普通的高阶函数,然后调用两次:

fun calculate(a: Int, b: Int, cal: (Int, Int) -> String) {
    println(cal(a, b))
}
fun main(args: Array<String>) {
    calculate(3, 7) { a, b ->
        "$a + $b = ${a + b}"
    }

    calculate(3, 7) { a, b ->
        "$a * $b = ${a * b}"
    }
}
// 输出
3 + 7 = 10
3 * 7 = 21

这样其实是看不出什么问题的,Kotlin 文件编译后会生成对应的 class 文件,所以我们将 class 文件反编译成 Java 文件后再看。如果使用Android Studio或者IntelliJ IDEA,可以按照如下方式查看 Kotlin 文件对应反编译后的 Java 文件:

  1. 打开目标 Kotlin 文件
  2. 查看 Kotlin 文件字节码:Tools –> Kotlin –> Show Kotlin ByteCode
  3. 在 kotlin 文件字节码页面中点击左上角的 decompile 按钮,就会生成对应的 Java 文件

我们来看上边代码对应的 Java 代码:

1

虽然不是正常的 Java 代码,但不妨碍我们分析流程,可以看出,编译器创建了两个 Lambda 的实例,并进行了两次calculate函数调用。

那如果将calculate声明为内联函数呢:

inline fun calculate(a: Int, b: Int, cal: (Int, Int) -> String) {
    println(cal(a, b))
}

我们再看最终的 Java 文件:


2

即编译器会把调用这个函数的地方用这个函数的方法体进行替换,这样验证了之前的说法。

需要注意的是, 内联函数提高代码性能的同时也会导致代码量的增加,所以应避免内联函数过大。

二、禁用内联(noinline)

如果一个内联函数可以接收多个 Lambda 表达式作为参数,默认这些 Lambda 表达式都会被内内联到调用处,如果需要某个 Lambda 表达式不被内联,可以使用noinline修饰对应的函数参数:

inline fun calculate(a: Int, b: Int, noinline title: () -> Unit, cal: (Int, Int) -> String) {
    title()
    println(cal(a, b))
}
fun main(args: Array<String>) {
    calculate(3, 7, { println("开始计算") }) { a, b ->
        "$a * $b = ${a * b}"
    }
}
// 输出
开始计算
3 * 7 = 21

title对应的 Lambda 确实没有被内联,看图:

3

一个内联函数没有可内联的函数参数并且没有具体化的类型参数,编译器会有警告,因为这样并不能带来什么好处,如果你不愿去掉内联修饰,可以使用@Suppress("NOTHING_TO_INLINE") 注解关闭这个警告。

三、非局部返回

我们知道默认情况下,在高阶函数中,要显式的退出(返回)一个 Lambda 表达式,需要使用 return@标签的语法,不能使用裸return,但这样也不能使高阶函数和包含高阶函数的函数退出。例如:

fun message(block: () -> Unit) {
    block()
    println("-----")
}

fun test() {
    message {
        println("Hello")
        return@message
    }
    println("World")
}
fun main(args: Array<String>) {
    test()
}
// 输出
Hello
-----
World

但如果把 Lambda 表达式作为参数传递给一个内联函数,就可以在 Lambda 表达式中正常的使用return语句了,并且会使该内联函数和包含该内联函数的函数退出(返回),这种操作就是非局部返回。例如:

inline fun message(block: () -> Unit) {
    block()
    println("-----")
}

fun test() {
    message {
        println("Hello")
        return
    }
    println("World")
}
fun main(args: Array<String>) {
    test()
}
// 输出
Hello

注意,由于非局部返回的原因,这里只输出了Hello

在使用了非局部返回后,Lambda 表达式中return的返回值受调用该内联函数的函数的返回值类型影响。例如:

fun test(): Boolean {
    message {
        println("Hello")
        return false
    }
    println("World")
    return true
}

四、禁用非局部返回(crossinline)

从前边已经知道,通过内联函数可以使 Lambda表达式实现非局部返回,但是,如果一个内联函数的函数类型参数被crossinline修饰,则对应传入的 Lambda表达式将不能非局部返回了,只能局部返回了。还是用之前的例子修改:

inline fun message(crossinline block: () -> Unit) {
    block()
    println("-----")
}

fun test() {
    message {
        println("Hello")
        return@message
    }
    println("World")
}
fun main(args: Array<String>) {
    test()
}
// 输出
Hello
-----
World

通过crossinline可以禁用掉非局部返回,但有什么意义呢?这其实是有实际的场景需求的,看个例子:

interface Calculator {
    fun calculate(a: Int, b: Int): Int
}

inline fun test(block: (Int, Int) -> Int) {
    val c = object : Calculator {
        override fun calculate(a: Int, b: Int): Int = block(a, b)
    }
    c.calculate(3, 7)
}

首先定义一个Calculator计算接口,然后在内联函数test中创建Calculator的一个对象表达式,重写calculate方法时,我们让calculate的函数体是test函数的block参数,当block是 Lambda表达式时,由于非局部返回的原因,导致calculate函数的返回值不是预期的,进而发生异常,为了避免这种情况的发生,所以就有必要使用crossinline来禁用非局部返回,来保证calculate的返回值类型是安全的。

上边的代码会有这样一个错误提示:

Can't inline 'block' here: it may contain non-local returns. Add 'crossinline' modifier to parameter declaration 'block'

使用crossinline后就正常了:

inline fun test(crossinline block: (Int, Int) -> Int) {
    val c = object : Calculator {
        override fun calculate(a: Int, b: Int): Int = block(a, b)
    }
    c.calculate(3, 7)
}

五、具体化的类型参数(reified)

对于一个泛型函数,如果需要访问泛型参数的类型,但由于泛型类型被擦除的原因,可能无法直接访问,但通过反射还是可以做到的,例如:

fun <T> test(param: Any, clazz: Class<T>) {
    if (clazz.isInstance(param)) {
        println("参数类型匹配")
    } else {
        println("参数类型不匹配")
    }
}
fun main(args: Array<String>) {
    test("Hello World", String::class.java)
    test(666, String::class.java)
}
// 输出
参数类型匹配
参数类型不匹配

功能虽然实现了,但是不够优雅,Kotlin 中有更好的办法来实现这样的功能。

在内联函数中支持具体化的参数类型,即用reified来修饰需要具体化的参数类型,这样我们用reified来修饰泛型的参数类型,以达到我们的目的:

inline fun <reified T> test(param: Any) {
    if (param is T) {
        println("参数类型匹配")
    } else {
        println("参数类型不匹配")
    }
}

调用的过程也变得简单了:

fun main(args: Array<String>) {
    test<String>("Hello World")
    test<String>(666)
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,117评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,328评论 1 293
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,839评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,007评论 0 206
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,384评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,629评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,880评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,593评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,313评论 1 243
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,575评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,066评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,392评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,052评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,082评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,844评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,662评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,575评论 2 270

推荐阅读更多精彩内容