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

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对CuoXin错新网的支持。