Kotlin标准库中的五种高阶函数

简述

在Kotlin中，高阶函数即指：将函数用作一个函数的参数或者返回值的函数。

Kotlin中的5个通用扩展函数，这些函数都存在Standard.kt文件中，run,with,let,also,apply区别和使用场景如下：

let

根据标准库的解释：调用指定的函数 [block]，将接收者作为参数传入代码块，并返回其结果。即只能使用 it 调用自身。

@kotlin.internal.InlineOnly
public inline fun <T, R> T.let(block: (T) -> R): R {
    contract {
        callsInPlace(block, InvocationKind.EXACTLY_ONCE)
    }
    return block(this)
}

run

根据标准库的解释：调用指定的函数 [block] 并返回其结果。其源代码中，有两种使用方式，源代码如下：

@kotlin.internal.InlineOnly
public inline fun <R> run(block: () -> R): R {
    contract {
        callsInPlace(block, InvocationKind.EXACTLY_ONCE)
    }
    return block()
}

@kotlin.internal.InlineOnly
public inline fun <T, R> T.run(block: T.() -> R): R {
    contract {
        callsInPlace(block, InvocationKind.EXACTLY_ONCE)
    }
    return block()
}

我们在使用第一种时，直接将代码块写在大括号中，其所指向的是所在的类,可以直接访问其中的成员

在了解第一种的使用后，第二种就很好理解了，第二种的可访问的范围就是调用 run 的对象本身

with

根据标准库解释：with函数的返回值指定了receiver为接收者。与T.run()在作用上是相同的，如下源码，它们的区别是，with是正常的高阶函数，T.run()是扩展的高阶函数。

@kotlin.internal.InlineOnly
public inline fun <T, R> with(receiver: T, block: T.() -> R): R {
    contract {
        callsInPlace(block, InvocationKind.EXACTLY_ONCE)
    }
    return receiver.block()
}

下面的代码是 with 与 T.run() 的对比：

val newStr : String? = "kotlin"

with(newStr) {    
    println( "length = ${this?.length}" )    
}
    
newStr?.run {
    println( "length = $length" ) 
}

apply

根据标准库的解释：执行传入的代码块，并且获取函数接收者本身（也就是this，即这里的T），最后返回this。

T.apply() 与 T.run() 的区别就是返回的对象不一样，前者返回函数调用者本身的引用，后者是返回需要执行的代码块。

@kotlin.internal.InlineOnly
public inline fun <T> T.apply(block: T.() -> Unit): T {
    contract {
        callsInPlace(block, InvocationKind.EXACTLY_ONCE)
    }
    block()
    return this
}

also

根据标准库的解释：执行传入的代码块，并且获取函数接收者本身（也就是this，即这里的T），将this作为参数传入到代码块中，最后返回this。

@kotlin.internal.InlineOnly
@SinceKotlin("1.1")
public inline fun <T> T.also(block: (T) -> Unit): T {
    contract {
        callsInPlace(block, InvocationKind.EXACTLY_ONCE)
    }
    block(this)
    return this
}

与 apply 的区别是，前者只能使用 it 调用自身，后者只能用 this 调用自身。如下：

var str : String = "12345"

str.apply [
    print("$length")
]

str.also {
    print("it.length)
}