Kotlin 高阶函数与 Lambda 表达式

在 Kotlin 中函数也是一等公民,这意味着我们定义的变量、函数参数、返回值都可以是函数类型的,可以像操作其它非函数值一样操作函数,确实也方便了不少。对 Android 开发者而言这无疑是一个较大的变化(虽然从 Java8 开始也有了类似的操作),同时也是 Kotlin 中相对重要的知识点,值得我们深入学习。

一、高阶函数

高阶函数是将函数作为参数或返回值的函数。这里有三个问题需要我们先思考:

  1. 如何定义一个函数的参数为函数类型?
  2. 如何使用这个函数类型的参数?
  3. 函数作为参数如何传递?

先看一个高阶函数实现的例子:

class Calculator {
    fun sum(a: Int, b: Int): Int = a + b

    fun calculate(a: Int, b: Int, cal: (Int, Int) -> Int) {
        print("a + b = ${cal(a, b)}")
    }
}

fun main(args: Array<String>) {
    val calculator = Calculator()
    calculator.calculate(1, 1, calculator::sum)
}
// 输出
a + b = 2

现在回答上边的三个问题

  1. cal: (Int, Int) -> Int就定义了calculate()函数的第三个参数为函数类型,同时该类型的函数有两个Int类型参数,返回值为Int类型。所以->左边括号部分为函数参数类型声明,右边为函数的返回值。注意,如果是无参函数括号()不能省略。
  2. cal(a, b)就是使用这个函数类型的参数,和普通函数的调用没啥区别。
  3. 注意main()方法的最后一行,我们使用了calculator::sum这种写法,注意sum()方法的声明和cal参数的类型是一致的。

同样的原理,函数也可以作为返回值,简单修改上边的代码:

class Calculator {
    fun sum(a: Int, b: Int): Int = a + b
    fun getSum(): (Int, Int) -> Int = this::sum
}

fun main(args: Array<String>) {
    val f = Calculator().getSum()
    print(f(1, 1))
}
// 输出
2

二、Lambda 表达式

1、初识 Lambda 表达式

Lambda 表达式本质上是可以传递给其它函数来作为参数的代码块。咦?这样说 Lambda 表达式也可作为高阶函数的函数类型参数的值?当然,Lambda 表达式为函数式编程提供了更好的实现。先修改上边的代码:

class Calculator {
    fun calculate(a: Int, b: Int, cal: (Int, Int) -> Int) {
        print("a + b = ${cal(a, b)}")
    }
}

fun main(args: Array<String>) {
    val calculator = Calculator()
    calculator.calculate(1, 1, { a, b -> a + b })
}
// 输出
a + b = 2

和第一部分中主要的不同之处是,调用calculate()方法时使用了{ a, b -> a + b }作为第三个参数的值,其实{ a, b -> a + b }就是一个 Lambda 表达式,这里省略了它的参数类型,完整的如下:

{ a: Int, b: Int -> a + b }

通过例子我们可以看出 Lambda 表达式的书写规则如下:

  1. Lambda 表达式总是被花括号{}包裹着
  2. ->左边为参数定义部分,多个参数用逗号,间隔,类似函数的参数声明,但 Lambda 表达式的参数类型可以省略(编译器可以推断类型)。
  3. ->右边为 Lambda 表达式要执行的业务逻辑,类似于函数体

2、Lambda 表达式的一些特性

  1. 如果函数的最后一个参数接受函数,那么传入的 Lambda 表达式可以放在圆括号之外:
fun calculate(a: Int, b: Int, cal: (Int, Int) -> Int) {
    print("a + b = ${cal(a, b)}")
}

fun main(args: Array<String>) {
    calculate(1, 1) { a, b -> a + b }
}
  1. 如果 Lambda 表达式只有一个参数,并且编译器自己可以识别出签名,也可以不用声明唯一的参数并忽略->, 该参数会隐式声明为it
fun calculate(a: Int, cal: (Int) -> Int) {
    print(cal(a))
}

fun main(args: Array<String>) {
    calculate(2) { it * it }
}
// 输入
4
  1. 如果 Lambda 表达式没有参数,也可以忽略->, 并且不会隐式声明参数it,可参考下一点的例子。

  2. 如果 Lambda 表达式是调用时唯一的参数,那么圆括号也可以省略:

fun myPrint(p: () -> Unit) {
    p()
}

fun main(args: Array<String>) {
    myPrint {
        print("timestamp is ${System.currentTimeMillis()}")
    }
}
  1. Lambda 表达式将隐式返回最后一个表达式的值,但可以使用限定的返回语法,即通过标签显式返回一个值:
fun calculate(a: Int, cal: (Int) -> Int) {
    print(cal(a))
}

fun main(args: Array<String>) {
    calculate(2) {
        it * it
    }

    calculate(2) {
        return@calculate it * it
    }
}
  1. 从 Kotlin1.1 起,如果 Lambda 表达式的参数未使用,则可以用下划线代替其名称:
fun calculate(a: Int, b: Int, cal: (Int, Int) -> Int) {
    print(cal(a, b))
}

fun main(args: Array<String>) {
    calculate(1, 2) { _, b ->
        println("a 参数未使用")
        b * b
    }
}
  1. 从 Kotlin1.1 起,Lambda 表达式参数支持解构声明语法,我们通过 Kotlin 内置的forEach()方法遍历map来测试:
fun main(args: Array<String>) {
    val map = mapOf(1 to 1, 2 to 2, 3 to 3)

    map.forEach {
        println("${it.key} to ${it.value}")
    }

    // 显示声明it参数的类型
    map.forEach { entry: Map.Entry<Int, Int> ->
        println("${entry.key} to ${entry.value}")
    }
    // 将Map.Entry类型的it参数解构
    map.forEach { (k, v) ->
        println("$k to $v")
    }
}

三、匿名函数

其实 Lambda 表达式有一个问题,就是无法显示的指定其返回值的类型,虽然可以自动推断出返回值类型,如果确实需要显式指定返回值类型,可以匿名函数,和普通的函数类似,只是省略了函数名:

fun(a: Int, b: Int): Int {
    return a + b
}
// 或者
fun(a: Int, b: Int): Int = a + b

用法和 Lambda 表达式也有差别的:

fun calculate(a: Int, b: Int, cal: (Int, Int) -> Int) {
    println("a + b = ${cal(a, b)}")
}

val sum = fun(a: Int, b: Int): Int = a + b

fun main(args: Array<String>) {
    calculate(1, 1, sum)
    calculate(2, 2, fun(a: Int, b: Int): Int = a + b)
}

四、访问闭包

Lambda 表达式或者匿名函数(以及局部函数对象表达式) 可以访问其 闭包 ,即在外部作用域中声明的变量。 与 Java 不同的是可以修改闭包中捕获的变量:

fun calculate(a: Int, b: Int, cal: (Int, Int) -> String) {
    print(cal(a, b))
}

fun main(args: Array<String>) {
    var str: String
    calculate(1, 1) { a, b ->
        str = "a + b = "
        "$str${a + b}"
    }
}
// 输出
a + b = 2
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,384评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,845评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,148评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,640评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,731评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,712评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,703评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,473评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,915评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,227评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,384评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,063评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,706评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,302评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,531评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,321评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,248评论 2 352

推荐阅读更多精彩内容