Kotlin - 函数与Lambda表达式

简介: Kotlin - 函数与Lambda表达式

本系列学习教程笔记属于详细讲解Kotlin语法的教程,需要快速学习Kotlin语法的小伙伴可以查看“简洁” 系列的教程

快速入门请阅读如下简洁教程:
Kotlin学习教程(一)
Kotlin学习教程(二)
Kotlin学习教程(三)
Kotlin学习教程(四)
Kotlin学习教程(五)
Kotlin学习教程(六)
Kotlin学习教程(七)
Kotlin学习教程(八)
Kotlin学习教程(九)
Kotlin学习教程(十)

imgKotlin - 函数与Lambda表达式

#函数

函数是以特定功能组织起来的代码块,函数定义格式如下:

fun sum(arg1: Int, arg2: Int): Int {
    return arg1 + arg2
}

如果函数方法体只有一行代码,那可以将 {} 改为 =,去掉 return,表达式的值将作为函数返回值:

fun sum(arg1: Int, arg2: Int) = arg1 + arg2

#默认返回值类型

任何函数都是有返回值的,一般函数方法体中会使用 return 将结果进行返回,但有的函数就是不需要返回结果,此时可以将 return 省略:

fun main(args: Array<String>) {
    ...
}

即使函数方法体中没有 return 返回结果,但函数还是有返回值类型的,默认的返回值类型就是 Unit ,但一般 Unit 会省略不写:

fun main(args: Array<String>): Unit {
    ...
}

#匿名函数

匿名函数就是没有名字的函数,但是需要赋值给一个变量,否则 IDE 会报错:

// 变量sum,它的值是一个函数
val sum = fun(arg1: Int, arg2: Int): Int {
    return arg1 + arg2
}

fun main(args: Array<String>): Unit {
    val result = sum(1, 2)
    println(result) // 3
}

此时 sum 是一个变量,只是它的值是一个函数。

#Lambda 表达式

Lambda 表达式也是一种"匿名函数",不过格式上会有些不同:

  • {[参数列表] -> [函数体, 最后一行是返回值]}
val sum = { arg1: Int, arg2: Int ->
    println("$arg1 + $arg2 = ${arg1 + arg2}")
    arg1 + arg2
}

fun main(){
    println(sum(1, 2)) // 3
}

Lambda 表达式的返回值是表达式中最后一行的结果。

#Lambda 的类型

Kotlin 中,不管是常量或是变量,都会有各自的类型,如:Human,那如果一个变量的值是 Lambda 表达式,那它的类型是什么呢?没错,就是 Lambda 类型:

  • () -> Unit : 无参,返回值为 null
  • (Int) -> Int : 传入整型,返回一个整型
  • (String, (String)->String) -> Boolean : 传入字符串、Lambda 表达式,返回 Boolean
// lambda的类型:(Int,Int) -> Int
val sum = { arg1: Int, arg2: Int -> arg1 + arg2 }

// lambda的类型:() -> Unit
val printlnHello = {
    println("Hello")
}

#Lambda 表达式的调用

与匿名函数一样,值为 Lambda 表达式的变量可通过以下两种方式执行 Lambda 表达式:

  • () 进行调用
  • invoke() 调用
println(sum(1, 2)) // 3
println(sum.invoke(1, 2)) // 3

实际上,() 就是 invoke()

#Lambda 表达式的简化

  • 函数参数调用时最后一个 Lambda 可以移出去
  • 函数参数只有一个 Lambda,调用时小括号可省略
  • Lambda 只有一个参数可默认为 it
  • 入参、返回值与形参一致的函数可以用函数引用的方式作为实参传入

我们先来看一下 Array 数组的一个扩展函数 forEach

/**
 * Performs the given [action] on each element.
 */
public inline fun <T> Array<out T>.forEach(action: (T) -> Unit): Unit {
    for (element in this) action(element)
}

这个扩展函数为 Array 类扩展了一个 forEach 方法,需要传入一个参数 action,其类型是 (T) -> Unit 的 Lambda 类型,返回值为 Unit(即没有返回值),方法体中使用 for-in 对数组进行遍历,同时使用参数 action 处理遍历的元素。

关于扩展函数这里不展开细讲,后续文章会专门解读,这里只需要知道 forEach 的功能即可。

val args = arrayOf("1", "2", "3", "l", "q", "r")

// 使用 for-in 遍历数组并打印元素内容
for (arg in args) {
    println(arg)
}

这是一个普通的数组遍历打印功能,上面已经使用 for-in 实现了,下面就使用前面学到的扩展函数 forEach 来处理:

// Lambda完整写法
args.forEach({ it -> println(it) })

// Lambda简化写法
args.forEach({ println(it) })

// 重命名参数it,此时ele不可省略
args.forEach({ ele -> println(ele) })

Lambda 只有一个参数时可默认为 it,当然了,如果你不想使用 it 也是可以改名的,但如果改名了,就不能将其省略。除了省略默认参数 it,还有如下方法简化 Lambda 表达式:

// 函数参数调用时最后一个 Lambda 可以移出去
args.forEach(){ println(it) }

// 函数参数只有一个 Lambda,调用时小括号可省略
args.forEach{ println(it) }

以上就是 Kotlin 中 Lambda 表达式的简化流程,以后开发中会经常遇到最后的简化写法。

#Lambda 表达式与函数引用

再回头看看这个扩展函数 forEach 的参数 action 的类型:

// action: (T) -> Unit
public inline fun <T> Array<out T>.forEach(action: (T) -> Unit): Unit {
    for (element in this) action(element)
}

再看看 println() 的定义:

// println: (Any?) -> Unit
public actual inline fun println(message: Any?) {
    System.out.println(message)
}

不难发现 println() 的定义完全符合 action 的 Lambda 类型要求,于是此场景下还可以将 forEach 参数 action 需要的 Lambda 表达式转换成 println 函数的引用,引用函数需要使用 ::

// 入参、返回值与形参一致的函数可以用函数引用的方式作为实参传入
args.forEach(::println)

#Lambda 表达式跳出 forEach

这里严重超纲,看不明白也无所谓,可以先跳过~

需求:遍历数组时,当发现元素是 q 时,中断循环。

val args = arrayOf("1", "2", "3", "l", "q", "r")
for (arg in args) {
    if (arg == "q") break
    println(arg)
}
println("The End") // 1 2 3 l The End

使用 for-in 配置 break 即可实现该功能,下面试试使用 forEach

fun main() {
    val args = arrayOf("1", "2", "3", "l", "q", "r")
    args.forEach {
        if (it == "q") break // IDE报错:'break' and 'continue' are only allowed inside a loop
        print("$it ")
    }
    println("The End")
}

是的,Lambda 表达式仅仅是一个表达式而已,不是循环体,所以无法使用 breakcontinue,那这里为了中断遍历只能使用 return 了:

fun main() {
    val args = arrayOf("1", "2", "3", "l", "q", "r")
    args.forEach { // 1 2 3 l
        if (it == "q") return
        print("$it ")
    }
    println("The End") // 不会被执行
}

乍一看,这里使用了 return 实现了中断遍历,但其实 "The End" 没有输出,说明 return 并非中断遍历,而是中断了当前的 main 函数,原因还是那个:Lambda 表达式仅仅只是表达式而已。

针对 Lambda 表达式,Kotlin 提供了标签式返回,可以在 Lambda 表达式中使用 return@xxx 方法跳出遍历,那使用这个来试试看:

fun main() {
    val args = arrayOf("1", "2", "3", "l", "q", "r")
    args.forEach {
        if (it == "q") return@forEach
        print("$it ")
    }
    println("The End") // 1 2 3 l r The End
}

程序跳过 q 继续往下执行了,这又是什么情况?这还是要回到 Lambda 表达式的本质,forEach里的 Lambda 表达式其实就是一个函数体,因此标签式返回 return@xxx 对这个函数体的退出只能退出当前的调用罢了。那就没有办法使用 forEach 实现这个需求了吗?是的,单靠 forEach 不行,需要配置 StreamApi 对数组进行过滤后再遍历:

val args = arrayOf("1", "2", "3", "l", "q", "r")
args.takeWhile { it != "q" }.forEach { print("$it ") }
println("The End") // 1 2 3 l The End
相关文章
|
26天前
|
弹性计算 人工智能 架构师
阿里云携手Altair共拓云上工业仿真新机遇
2024年9月12日,「2024 Altair 技术大会杭州站」成功召开,阿里云弹性计算产品运营与生态负责人何川,与Altair中国技术总监赵阳在会上联合发布了最新的“云上CAE一体机”。
阿里云携手Altair共拓云上工业仿真新机遇
|
2天前
|
人工智能 Rust Java
10月更文挑战赛火热启动,坚持热爱坚持创作!
开发者社区10月更文挑战,寻找热爱技术内容创作的你,欢迎来创作!
308 14
|
18天前
|
存储 关系型数据库 分布式数据库
GraphRAG:基于PolarDB+通义千问+LangChain的知识图谱+大模型最佳实践
本文介绍了如何使用PolarDB、通义千问和LangChain搭建GraphRAG系统,结合知识图谱和向量检索提升问答质量。通过实例展示了单独使用向量检索和图检索的局限性,并通过图+向量联合搜索增强了问答准确性。PolarDB支持AGE图引擎和pgvector插件,实现图数据和向量数据的统一存储与检索,提升了RAG系统的性能和效果。
|
5天前
|
JSON 自然语言处理 数据管理
阿里云百炼产品月刊【2024年9月】
阿里云百炼产品月刊【2024年9月】,涵盖本月产品和功能发布、活动,应用实践等内容,帮助您快速了解阿里云百炼产品的最新动态。
阿里云百炼产品月刊【2024年9月】
|
20天前
|
人工智能 IDE 程序员
期盼已久!通义灵码 AI 程序员开启邀测,全流程开发仅用几分钟
在云栖大会上,阿里云云原生应用平台负责人丁宇宣布,「通义灵码」完成全面升级,并正式发布 AI 程序员。
|
22天前
|
机器学习/深度学习 算法 大数据
【BetterBench博士】2024 “华为杯”第二十一届中国研究生数学建模竞赛 选题分析
2024“华为杯”数学建模竞赛,对ABCDEF每个题进行详细的分析,涵盖风电场功率优化、WLAN网络吞吐量、磁性元件损耗建模、地理环境问题、高速公路应急车道启用和X射线脉冲星建模等多领域问题,解析了问题类型、专业和技能的需要。
2584 22
【BetterBench博士】2024 “华为杯”第二十一届中国研究生数学建模竞赛 选题分析
|
4天前
|
存储 人工智能 搜索推荐
数据治理,是时候打破刻板印象了
瓴羊智能数据建设与治理产品Datapin全面升级,可演进扩展的数据架构体系为企业数据治理预留发展空间,推出敏捷版用以解决企业数据量不大但需构建数据的场景问题,基于大模型打造的DataAgent更是为企业用好数据资产提供了便利。
177 2
|
2天前
|
编译器 C#
C#多态概述:通过继承实现的不同对象调用相同的方法,表现出不同的行为
C#多态概述:通过继承实现的不同对象调用相同的方法,表现出不同的行为
102 65
|
6天前
|
Linux 虚拟化 开发者
一键将CentOs的yum源更换为国内阿里yum源
一键将CentOs的yum源更换为国内阿里yum源
283 2
|
22天前
|
机器学习/深度学习 算法 数据可视化
【BetterBench博士】2024年中国研究生数学建模竞赛 C题:数据驱动下磁性元件的磁芯损耗建模 问题分析、数学模型、python 代码
2024年中国研究生数学建模竞赛C题聚焦磁性元件磁芯损耗建模。题目背景介绍了电能变换技术的发展与应用,强调磁性元件在功率变换器中的重要性。磁芯损耗受多种因素影响,现有模型难以精确预测。题目要求通过数据分析建立高精度磁芯损耗模型。具体任务包括励磁波形分类、修正斯坦麦茨方程、分析影响因素、构建预测模型及优化设计条件。涉及数据预处理、特征提取、机器学习及优化算法等技术。适合电气、材料、计算机等多个专业学生参与。
1580 16
【BetterBench博士】2024年中国研究生数学建模竞赛 C题:数据驱动下磁性元件的磁芯损耗建模 问题分析、数学模型、python 代码