一聚教程网:一个值得你收藏的教程网站

热门教程

Kotlin基础学习之lambda中return语句详解

时间:2022-06-25 22:55:14 编辑:袖梨 来源:一聚教程网

前言

当我们爱上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  Array.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) {
 doSomething {
  return
 }
}

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

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

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

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

热门栏目