Kotlin内联函数 - Cherry's Blog

503 字

3 分钟

Kotlin内联函数

2023-07-20

Coding

Kotlin

前言#

TIP
以下的文章均基于以下示例代码

1
class Calculator {
2
    var a: Int = 0
3
    var b: Int = 0
4
    fun add() = a + b
5
    fun multiply() = a * b
6
}
7

8
fun main() {
9
    val calculator = Calculator()
10
    doCalculation(calculator)
11
}

run 函数#

run 的标准函数形式#

1
fun doCalculation(calculator: Calculator) {
2
    val result = run {
3
        calculator.a = 1
4
        calculator.b = 2
5
        calculator.add()
6
    }
7
    println(result)
8
}

run 的标准函数形式能够在 Lambda 中执行操作，并返回一个结果，此结果即为 Lambda 中最后一行表达式

run 的扩展函数形式#

1
fun doCalculation(calculator: Calculator) {
2
    val result = calculator.run {
3
        a = 1
4
        b = 2
5
        add()
6
    }
7
    println(result)
8
}

run 此时被定义为接收者的扩展函数，它能够接收一个带有接收者(reciver)的 Lambda。在reciver中，能够访问调用者的上下文

with 函数#

在 with 内处理结果#

1
fun doCalculation(calculator: Calculator) {
2
    val resultOne = with(calculator) {
3
        a = 1
4
        b = 2
5
        add()
6
    }
7
    println(resultOne)
8
}

with 中也拥有reciver(接收者)。此时 with 的最后一个表达式即为 with 的返回值。若要让代码返回到 with,而不是直接 return 整个函数,可以使用return@with。

在 with 外处理结果#

1
fun doCalculation(calculator: Calculator) {
2
    with(calculator) {
3
        a = 3
4
        b = 4
5
    }
6
    val resultTwo = calculator.add()
7
    println(resultTwo)
8
}

在这一种处理方式中，with 仅用于对对象统一的操作，而不直接返回值。

apply 函数#

1
fun doCalculation(calculator: Calculator) {
2
    val result = calculator.apply {
3
        a = 1
4
        b = 2
5
    }.add()
6
    println(result)
7
}

在 apply 的 Lambda 中，同样存在着接收者(reciver)。在 apply 函数中，Lambda 在调用过后将会返回函数自身，因此不会打断函数后面的调用链。

let 函数#

1
fun main(){
2
    val values = listOf("aaaa", "bbbbb", null, "ccccccc", "ddddd", null)
3
    for (str in values) {
4
        str?.let { println(it) }
5
    }
6
}

let 对应着 run,为 run 不含有receiver的版本。通过使用?配合 let，能够确保空安全。

also#

1
fun doCalculation(calculator: Calculator) {
2
    val result = calculator.apply {
3
        a = 1
4
        b = 2
5
    }.add().also { println(it) }.times(5) // .times()实际上就是 * 的重载函数
6
    println(result)
7
}