Kotlin:lambda中return语句

当我们爱上lambda并且大范围使用它的时候,我想大家都会被lambda中的return语句狠狠地调戏过,所以今天我们需要一起来揭开lambda中return的神秘面纱。

首先来看一个例子:

fun demo() {
    val indexes = arrayOf(1, 2, 3, 4, 5, 6, 7)
    indexes.forEach {
        if (it > 5) {
           return
        }
        println(it)
    }
    println("End")
}

按照我们的预期,调用demo后,它应该输出:

1
2
3
4
5
End 

事实真是如此吗?大家可以运行一下上面的代码,它的实际输出是:

1
2
3
4
5

什么鬼?那个End被运行时给偷吃了吗?不是,为了说明其中的缘由,让我们看一下forEach的定义:

public inline fun <T> Array<out T>.forEach(action: (T) -> Unit): Unit {
    for (element in this) action(element)
}

从定义中,我们可以看到forEach函数被定义为了内联函数,而根据带有lambda的内联函数处理机制(相关讲解可参考:Kotlin:关于内联函数的一些理解),我们的demo因为被内联最终会被编译为:

fun demo() {
    val indexes = arrayOf(1, 2, 3, 4, 5, 6, 7)
    for (element in indexes) {
        if (element > 5) {
           return
        }
        println(element)
    }
    println("End")
}

从最终编译后的代码中我们可以很清楚的发现,最后的End不是被偷吃了,而是因为demo函数在判断条件满足的情况下提前返回了,这样的return用高端一点的语句就叫做:非局部返回。

如果我们想要return从forEach(即lambda)中返回怎么办?让我们先改造一下上面对forEach的调用:

indexes.forEach {
    if (it > 5) {
        return@forEach
    }
    println(it)
}

再次运行一下 demo

1
2
3
4
5
End  

这个时候的输出就跟我们的预期一样了,很神奇对不对?我们只要在return后面加一个@forEach就可以了,这就是所谓标签返回(或局部返回),其完整的语法如下:

indexes.forEach label@ {
    if (it > 5) {
        return@label
    }
    println(it)
}

如果我们省略forEach后面label的定义,那么默认的label就是把lambda作为参数的函数名(这里是forEach)。

这就是lambda中return语句的所有内容,很简单是不是?接下来就是尽情玩耍的时间了?别着急,这里还有一些注意事项,比如:

fun doSomething(action: () -> Unit) {
   action()
}

inline fun doOtherThing(action: () -> Unit) {
   action()
}

fun main(args: Array<String>) {
    doSomething {
       return
    }
}

fun main(args: Array<String>) {
    var action = {
       return
    }
    doOtherThing(action)
}

运行上面的例子会得到怎样的输出?不好意思,这个调用会因为编译失败不给你运行的机会,

通过上面的讨论我们知道,非局部返回是从lambda调用点所在的函数中返回,所以这就要求我们lambda中的return语句只能出现在内联函数且该lambda表达式以参数形式直接传递给该函数的情况,其他情况不允许则是因为lambda可以绑定给一个变量以便在函数返回后继续使用(比如闭包),而这个时候的return已经为时已晚。

好了,关于lambda控制流的内容就讨论到这里了,最后祝愿大家快乐coding ^ _ ^

    原文作者:泪已无痕
    原文地址: https://www.jianshu.com/p/92cd94cba709
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞