Kotlin —— 内联函数

一、前言

Kotlin标准库中所有集合操作的函数都是内联的( inline ),例如:

public inline fun <T> Iterable<T>.forEach(action: (T) -> Unit): Unit {
    for (element in this) action(element)
}
public inline fun <T> Iterable<T>.forEachIndexed(action: (index: Int, T) -> Unit): Unit {
    var index = 0
    for (item in this) action(checkIndexOverflow(index++), item)
}

这个 inline 修饰符有多重要呢?
假设我们有 5000 件商品,我们需要对已购买的商品计算总价,我们可以通过以下方式完成:

products.filter{ it.bought }.sumByDouble { it.price }

在我的机器上,运行上述代码平均需要38毫秒。如果这个函数不是内联的话会是多长时间呢? 不是内联在我的机器上大概平均42毫秒。你们可以自己检查尝试下,这里是完整源码. 这似乎看起来差距不是很大,但每调用一次这个函数对集合进行处理时,你都会注意到这个时间差距大约为10%左右。

当我们修改lambda表达式中的局部变量时,可以发现差距将会更大。对比下面两个函数:

inline fun repeat(times: Int, action: (Int) -> Unit) {
    for (index in 0 until times) {
        action(index)
    }
}

fun noinlineRepeat(times: Int, action: (Int) -> Unit) {
    for (index in 0 until times) {
        action(index)
    }
}

除了函数名不一样之外,唯一的区别就是第一个函数使用inline修饰符,而第二个函数没有。用法也是完全一样的:

var a = 0
repeat(100_000_000) {
    a += 1
}

var b = 0
noinlineRepeat(100_000_000) {
    b += 1
}

上述代码在执行时间上对比有很大的差异。内联的repeat函数平均运行时间是0.335ns, 而noinlineRepeat函数平均运行时间是153980484.884ns。大概是内联repeat函数运行时间的466000倍! 你们可以自己检查尝试下,这里是完整源码.

为什么这个如此重要呢? 这种性能的提升是否有其他的成本呢? 我们应该什么时候使用内联(inline)修饰符呢?这些都是重点问题,我们将尽力回答这些问题。然而这一切都需要从最基本的问题开始: 内联修饰符到底有什么作用?

二、内联修饰符有什么作用?

我们都知道函数通常是如何被调用的。先执行跳转到函数体,然后执行函数体内所有的语句,最后跳回到最初调用函数的位置。尽管强行对函数使用inline修饰符标记,但是编译器将会以不同的方式来对它进行处理。在代码编译期间,它用它的主体替换这样的函数调用。 print函数是inline函数:

public inline fun print(message: Int) {
    System.out.print(message)
}

当我们在main函数中调用它时:

fun main(args: Array<String>) {
    print(2)
    print(2)
}

编译后,它将变成下面这样:

 public static final void main(@NotNull String[] args) {
    System.out.print(2)
    System.out.print(2)
}

这里有一点不一样的是我们不需要跳回到另一个函数中。虽然这种影响可以忽略不计。

这就是为什么你定义这样的内联函数时会在IDEA IntelliJ中发出以下警告:

Expected performance impact from inline is insignificant. Inlining works best for functions with parameters of functional types
意思就是:内联不会有显著的影响,它比函数要好。

为什么IntelliJ建议我们在含有lambda表达式作为形参的函数中使用内联呢?因为当我们内联函数体时,我们不需要从参数中创建lambda表达式实例,而是可以将它们内联到函数调用中来。这个是上述repeat函数的调用:

repeat(100) { println("A") }

将会编译成这样:

for (index in 0 until 1000) {
    println("A")
}

正如你所看见的那样,lambda表达式的主体println("A")替换了内联函数repeat中action(index)的调用。
让我们看另一外个例子。filter函数的用法:

val products2 = products.filter { it.bought }

替换为:

val destination = ArrayList<T>()
for (element in this) 
    if (predicate(element))
        destination.add(element)
val products2 = destination

这是一项非常重要的改进。这是因为JVM天然地不支持lambda表达式。
说清楚lambda表达式是如何被编译的是件很复杂的事。但总的来说,有两种结果:

  • 匿名类
  • 单独的类

我们来看个例子。我们有以下lambda表达式:

val lambda: ()->Unit = {
    // body
}

它变成了JVM中的匿名类:

// Java
Function0 lambda = new Function0() {
   public Object invoke() {
      // code
   }
};

或者它变成了单独的文件中定义的普通类:

// Java
// Additional class in separate file
public class TestInlineKt$lambda implements Function0 {
   public Object invoke() {
      // code
   }
}
// Usage
Function0 lambda = new TestInlineKt$lambda()

第二种效率更高,我们尽可能使用这种。仅仅当我们需要使用局部变量时,第一种才是必要的。

这就是为什么当我们修改局部变量时,repeat和noinlineRepeat之间存在如此之大的运行速度差异的原因:

非内联函数中的Lambda需要编译为匿名类!

这是一个巨大的性能开销,从而导致它们的创建和使用都较慢。
当我们使用内联函数时,我们根本不需要创建任何其他类。

自己检查一下。编译这段代码并把它反编译为Java代码:

fun main(args: Array<String>) {
    var a = 0
    repeat(100_000_000) {
        a += 1
    }
    var b = 0
    noinlineRepeat(100_000_000) {
        b += 1
    }
}

你会发现一些相似的东西

// Show Kotlin Bytecode
public static final void main(@NotNull String[] args) {
   Intrinsics.checkParameterIsNotNull(args, "args");
   int a = 0;
   int times$iv = 100000000;
   int $i$f$repeat = false;
   int var4 = 0;

   for(int var5 = times$iv; var4 < var5; ++var4) {
      int var7 = false;
      ++a;
   }

   final IntRef b = new IntRef();
   b.element = 0;
   noinlineRepeat(100000000, (Function1)(new Function1() {
      public Object invoke(Object var1) {
         this.invoke(((Number)var1).intValue());
         return Unit.INSTANCE;
      }

      public final void invoke(int it) {
         ++b.element;
      }
   }));
}

在filter函数例子中,使用内联函数改进效果不是那么明显,这是因为lambda表达式在非内联函数中是编译成普通的类而非匿名类。所以它的创建和使用效率还算比较高,但仍有性能开销,所以也就证明了最开始那个filter例子为什么只有10%的运行速度差异。

三、集合流处理方式与经典处理方式

内联修饰符是一个非常关键的元素,它能使集合流处理的方式与基于循环的经典处理方式一样高效。它经过一次又一次的测试,在代码可读性和性能方面已经优化到极点了,并且相比之下经典处理方式总是有很大的成本。例如,下面的代码:

return data.filter { filterLoad(it) }.map { mapLoad(it) }

工作原理与下面代码相同并具有相同的执行时间:

val list = ArrayList<String>()
for (it in data) {
    if (filterLoad(it)) {
        val value = mapLoad(it)
        list.add(value)
    }
}
return list

基准测量的具体结果(源码在这里):

Benchmark           (size) Mode  Cnt        Score    Error  Units
filterAndMap           10  avgt  200      561.249 ±      1  ns/op
filterAndMap         1000  avgt  200    29803.183 ±    127  ns/op
filterAndMap       100000  avgt  200  3859008.234 ±  50022  ns/op

filterAndMapManual     10  avgt  200      526.825 ±      1  ns/op
filterAndMapManual   1000  avgt  200    28420.161 ±     94  ns/op
filterAndMapManual 100000  avgt  200  3831213.798 ±  34858  ns/op

四、内联修饰符的成本

通过上面已经得出,内联函数,实际就是替换掉你原来的代码,改为内联函数中的代码。
因此,内联的优点是代码简洁,可读性强;缺点是编译后的代码体积会变大(变大多少取决于用了多少 inline )。

五、内联修饰符在不同方面的用法

内联修饰符因为它特殊的语法特性而发生的变化远远超过我们在本篇文章中看到的内容。
它可以实化泛型类型。但是它也有一些局限性。

\color{red}{一般来说,我们应该什么时候使用内联修饰符呢?}

我们使用内联修饰符时最常见的场景就是:

  • 把函数作为另一个函数的参数时(高阶函数);
  • 集合或字符串处理(如filter,map或者joinToString);
  • 一些独立的函数(如repeat)

这就是为什么inline修饰符经常被库开发人员用来做一些重要优化的原因了。他们应该知道它是如何工作的,哪里还需要被改进以及使用成本是什么。当我们使用函数类型作为参数来定义自己的工具类函数时,我们也需要在项目中使用inline修饰符。当我们没有函数类型作为参数,没有reified实化类型参数并且也不需要非本地返回时,那么我们很可能不应该使用inline修饰符了。这就是为什么我们在非上述情况下使用inline修饰符会在Android Studio或IDEA IntelliJ得到一个警告原因。

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

推荐阅读更多精彩内容

  • 在Kotlin中,lambda表达式会被正常的编译成匿名类。这表示每调用一次lambda表达式,一个额外的类就会被...
    leilifengxingmw阅读 973评论 0 3
  • 一、内联函数原理 使用高阶函数为开发带来了便利,但同时也产生了一些性能上的损失,官方是这样描述这个问题: 使用高阶...
    SheHuan阅读 5,106评论 1 21
  • Kotlin 中新增了「内联函数」,内联函数起初是在 C++ 里面的。 那在 Kotlin 中加入内联函数,是有什...
    chendroid阅读 23,293评论 12 61
  • kotlin的内联函数属于kotlin的高级特性了,也是不同于java的区别之一;至于为什么kotlin要使用内联...
    Villa__Mou阅读 9,204评论 0 25
  • Kotlin里使用关键 inline 来表示内联函数,那么到底什么是内联函数呢,内联函数有什么好处呢? 1. 什么...
    云飞扬1阅读 2,867评论 0 58