Kotlin中的扩展函数和尾调用优化(TCO)

问题描述 投票:0回答:1

我有以下使用TCO的功能:

tailrec fun superDigit(n: String): Int {
    val sum = n.fold(0) { sum, char -> sum + char.toString().toInt() }
    return if (sum < 10) sum else superDigit(sum.toString())
}

如果我实现与这样的扩展函数相同的功能:

fun String.superDigit(): Int {
    val sum = fold(0) { sum, char -> sum + char.toString().toInt() }
    return if (sum < 10) sum else sum.toString().superDigit()
}

扩展函数尾调用也是优化的吗?

IMO调用扩展函数仍然是一个常规函数调用,this作为参数,所以它仍然是一个递归调用,因为tailrec不能用于扩展函数,我的假设是它没有被编译器优化。这个假设是否正确?

kotlin tail-recursion
1个回答
2
投票

因为tailrec不能用于扩展功能

你确定吗?

我刚刚通过查看IntelliJ IDEA中的Kotlin字节码来测试它。首先,扩展函数上的tailrec代码编译成功。更进一步:比较下面的两个Kotlin代码和字节码,一个用tailrec,另一个用没有。

科特林:

fun Double.tailrecTestExtension(): Double
        = (this - 1.0).tailrecTestExtension()

字节码:

  // access flags 0x19
  public final static tailrecTestExtension(D)D
    // annotable parameter count: 1 (visible)
    // annotable parameter count: 1 (invisible)
   L0
    LINENUMBER 13 L0
    DLOAD 0
    DCONST_1
    DSUB
    INVOKESTATIC com/example/TestKt.tailrecTestExtension (D)D
    DRETURN
   L1
    LOCALVARIABLE $receiver D L0 L1 0
    MAXSTACK = 4
    MAXLOCALS = 2

科特林:

tailrec fun Double.tailrecTestExtension(): Double
        = (this - 1.0).tailrecTestExtension()

字节码:

  // access flags 0x19
  public final static tailrecTestExtension(D)D
    // annotable parameter count: 1 (visible)
    // annotable parameter count: 1 (invisible)
   L0
    LINENUMBER 13 L0
    DLOAD 0
    DCONST_1
    DSUB
    DSTORE 0
    GOTO L0
   L1
    LOCALVARIABLE $receiver D L0 L1 0
    MAXSTACK = 4
    MAXLOCALS = 2

请注意,在第一个示例中,有INVOKESTATIC调用(对应于常规递归),在第二个版本中用常规跳转(GOTO)替换(对应于tailrec引入的循环 - 预期行为)。

注意:我不是Kotlin字节码的专家,我的理解是基于汇编语言的一些基本知识。在这里,我假设这些知识可以转移到Kotlin字节码。

© www.soinside.com 2019 - 2024. All rights reserved.