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 ^ _ ^