Kotlin 函数0 - 基本函数以及 lambda 表达式
之前已经介绍过主函数了，参看 Kotlin 的基本语法、简单函数、基本数据类型，现在来详细的看一下 Kotlin 中的函数的用法
1. 函数
1. 在 Kotlin 中，函数完整的写法如下：
fun sum(arg1: Int, arg2: Int): Int {
return arg1 + arg2
}
sum(1, 2) // 3
2. 这是一个简单的求和函数，当函数体只有一句话的时候，其实是可以将 {} 用 = 替代的，如果要返回一个值，return 也省略。所以上面的函数可以写成：
有返回值
fun sum(arg1: Int, arg2: Int): Int = arg1 + arg2
sum(1, 2) // 3
无返回值
fun sayHello() = println("Hello Lambda")
sayHello() // Hello Lambda
3. 更灵活就是，还可以用一个变量去接收匿名函数，类似于 c 中的函数指针
val vSum = fun(arg1: Int, arg2: Int): Int = arg1 + arg2
vSum(1, 2) // 3
val vSayHello = fun() { println("Hello Lambda") }
vSayHello() // Hello Lambda
2. Lambda 表达式
1. Kotlin 支持了 lambda 表达式，这使得函数书写变得更加多样化
- 下面分两种情况（有参有返回值以及无参无返回值）来看
- 有参数，有返回值
val vSum = { arg1: Int, arg2: Int -> arg1 + arg2 }
vSum(1, 2) // 3
- 如果函数体内容很多呢
val vSum = { arg1: Int, arg2: Int ->
println("$arg1 + $arg2 = ${arg1 + arg2}")
arg1 + arg2
}
vSum(1, 2) // 3
- 无参数，无返回值
val vSayHello = { -> println("Hello Lambda") }
vSayHello() // Hello Lambda
- 其实在无参数时，还可以将
->省略
val vSayHello = { println("Hello Lambda") }
vSayHello() // Hello Lambda
- 函数体只有一条语句，是不是可以将
{}省略呢？
val vSayHello = println("Hello Lambda") // Hello Lambda
必须注意： 此时只相当于定义一个 vSayHello 变量，它不再是一个指向函数的变量了，而且在定义的时候会执行 = 右侧的内容；
2. 提到 lambda 表达式，不得不看一下数组的遍历的函数
- 一般遍历数组的方式
val array = arrayOf("Tom", 1, "Jerry")
for (it in array) {
print("$it ")
}
// Tom 1 Jerry
- 但是 Kotlin 已经给我们写好了一个
forEach()函数，用于遍历数组
array.forEach { print("$it ") }
- 注意这里必须使用
it来指代元素。
- 这是根据 IDE 的提示书写的代码，咋看起来有些懵，那么先来看一下这个方法的源码
public inline fun <T> Array<out T>.forEach(action: (T) -> Unit): Unit {
for (element in this) action(element)
}
inline内联函数，后面会有讲到；Array<out T>.forEach先不去关心那么多，将它简化一下Array.forEach，可以看出它是Array的一个扩展函数，函数名是forEach；(action: (T) -> Unit)参数列表，action参数名，(T) -> Unit参数类型是一个 lambda 表达式；: Unit无返回值for (element in this) action(element)函数体很简单，就是将数组中的变量通过 for 循环遍历，逐个放入到变量element中，然后交给action去处理；
- 经过分析
forEach()函数的源码，可以将第2点提到的代码还原一下
array.forEach({ element -> print("$element ") })
- 其实在 Kotlin 中，如果函数的最后一个参数是 lambda 表达式（上面分析得到
forEach()函数仅有一个参数，且为 lambda 表达式），可以将小括号提到大括号前面写，于是就变成了这样
array.forEach(){ element -> print("$element ") }
- 由于除了最后一个参数是 lambda 表达式外，前面再没有其他参数（小括号中没有参数），所以小括号也可以省略
array.forEach{ element -> print("$element ") }
- 上面也提到了，lambda 表达式中只有一个参数是可以省略的，所以
array.forEach { print("$it ") }
- lambda 表达式的中断
- 有一个数组，要全部遍历一遍
val array = arrayOf("Tom", 1, "Jerry", 4, "hello")
array.forEach { println(it) }
遍历结果
Tom
1
Jerry
4
hello
- 假设在遍历的过程加上判断，当遇到
4的时候跳过，同时在程序结束时输出The End
val array = arrayOf("Tom", 1, "Jerry", 4, "hello")
array.forEach {
if (it == 4) {
return
}
println(it)
}
println("The End")
上面的这段代码真的可以实现需求吗？先来运行看一下结果
Tom
1
Jerry
结果并没有输出
The End，这是为什么呢？原因在 lambda 表达式上面，因为这是一个表达式，并不是函数，所以println("The End")和上面的内容是顺序的执行，return会将下面的内容也结束了。那如果想要实现需求该怎么办呢？
val array = arrayOf("Tom", 1, "Jerry", 4, "hello")
array.forEach flag@{
if (it == 4) {
return@flag
}
println(it)
}
println("The End")
运行看一下结果
Tom
1
Jerry
hello
The End
到达了目的，其实就是给
{}中的内容起一个名字，在return的时候，指定到这个名字就可以了
