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 ^ _ ^
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对我们的支持。