高階函數(shù)和 lambda 表達式
高階函數(shù)
高階函數(shù)是將函數(shù)用作參數(shù)或返回值的函數(shù)废菱。
這種函數(shù)的一個很好的例子是 lock()
咖刃,它接受一個鎖對象和一個函數(shù)差油,獲取鎖拗军,運行函數(shù)并釋放鎖:
fun <T> lock(lock: Lock, body: () -> T): T {
lock.lock()
try {
return body()
}
finally {
lock.unlock()
}
}
讓我們來檢查上面的代碼:body
擁有函數(shù)類型:() -> T
,
所以它應(yīng)該是一個不帶參數(shù)并且返回 T
類型值的函數(shù)。
它在 try{: .keyword }-代碼塊內(nèi)部調(diào)用发侵、被 lock
保護侈咕,其結(jié)果由lock()
函數(shù)返回。
如果我們想調(diào)用 lock()
函數(shù)器紧,我們可以把另一個函數(shù)傳給它作為參數(shù)(參見函數(shù)引用):
fun toBeSynchronized() = sharedResource.operation()
val result = lock(lock, ::toBeSynchronized)
通常會更方便的另一種方式是傳一個 lambda 表達式:
val result = lock(lock, { sharedResource.operation() })
Lambda 表達式在下文會有更詳細(xì)的描述,但為了繼續(xù)這一段楼眷,讓我們看一個簡短的概述:
- lambda 表達式總是被大括號括著铲汪,
- 其參數(shù)(如果有的話)在
->
之前聲明(參數(shù)類型可以省略), - 函數(shù)體(如果存在的話)在
->
后面罐柳。
在 Kotlin 中有一個約定掌腰,如果函數(shù)的最后一個參數(shù)是一個函數(shù),并且你傳遞一個 lambda 表達式作為相應(yīng)的參數(shù)张吉,你可以在圓括號之外指定它:
lock (lock) {
sharedResource.operation()
}
//原始
lock(lock, {sharedResource.operation()})
高階函數(shù)的另一個例子是 map()
:
fun <T, R> List<T>.map(transform: (T) -> R): List<R> {
val result = arrayListOf<R>()
for (item in this)
result.add(transform(item))
return result
}
該函數(shù)可以如下調(diào)用:
val doubled = ints.map { value -> value * 2 }
//原始
val doubled = ints.map({ value -> value * 2 })
//最后一個參數(shù)是函數(shù)齿梁,并且傳遞lambda作為參數(shù),可以在圓括號之外指定
val doubled = ints.map(){ value -> value * 2}
//lambda是唯一參數(shù)肮蛹,圓括號可以省略
val doubled = ints.map{ value -> value * 2}
請注意勺择,如果 lambda 是該調(diào)用的唯一參數(shù),則調(diào)用中的圓括號可以完全省略伦忠。
it
:單個參數(shù)的隱式名稱
另一個有用的約定是省核,如果函數(shù)字面值只有一個參數(shù),
那么它的聲明可以省略(連同 ->
)昆码,其名稱是 it
气忠。
ints.map { it * 2 }
這些約定可以寫LINQ-風(fēng)格的代碼:
strings.filter { it.length == 5 }.sortBy { it }.map { it.toUpperCase() }
下劃線用于未使用的變量(自 1.1 起)
如果 lambda 表達式的參數(shù)未使用,那么可以用下劃線取代其名稱:
map.forEach { _, value -> println("$value!") }
在 lambda 表達式中解構(gòu)(自 1.1 起)
在 lambda 表達式中解構(gòu)是作為解構(gòu)聲明的一部分描述的赋咽。
內(nèi)聯(lián)函數(shù)
使用內(nèi)聯(lián)函數(shù)有時能提高高階函數(shù)的性能旧噪。
Lambda 表達式和匿名函數(shù)
一個 lambda 表達式或匿名函數(shù)是一個“函數(shù)字面值”,即一個未聲明的函數(shù)脓匿,
但立即做為表達式傳遞淘钟。考慮下面的例子:
max(strings, { a, b -> a.length < b.length })
函數(shù) max
是一個高階函數(shù)亦镶,換句話說它接受一個函數(shù)作為第二個參數(shù)日月。
其第二個參數(shù)是一個表達式,它本身是一個函數(shù)缤骨,即函數(shù)字面值爱咬。寫成函數(shù)的話,它相當(dāng)于
fun compare(a: String, b: String): Boolean = a.length < b.length
函數(shù)類型
對于接受另一個函數(shù)作為參數(shù)的函數(shù)绊起,我們必須為該參數(shù)指定函數(shù)類型精拟。
例如上述函數(shù) max
定義如下:
fun <T> max(collection: Collection<T>, less: (T, T) -> Boolean): T? {
var max: T? = null
for (it in collection)
if (max == null || less(max, it))
max = it
return max
}
參數(shù) less
的類型是 (T, T) -> Boolean
,即一個接受兩個類型T
的參數(shù)并返回一個布爾值的函數(shù):
如果第一個參數(shù)小于第二個那么該函數(shù)返回 true。
在上面第 4 行代碼中蜂绎,less
作為一個函數(shù)使用:通過傳入兩個 T
類型的參數(shù)來調(diào)用栅表。
如上所寫的是就函數(shù)類型,或者可以有命名參數(shù)师枣,如果你想文檔化每個參數(shù)的含義的話怪瓶。
val compare: (x: T, y: T) -> Int = ……
var sum: ((Int, Int) -> Int)? = null
Lambda 表達式語法
Lambda 表達式的完整語法形式,即函數(shù)類型的字面值如下:
val sum = { x: Int, y: Int -> x + y }
lambda 表達式總是被大括號括著践美,
完整語法形式的參數(shù)聲明放在括號內(nèi)洗贰,并有可選的類型標(biāo)注,
函數(shù)體跟在一個 ->
符號之后陨倡。如果推斷出的該 lambda 的返回類型不是 Unit
敛滋,那么該 lambda 主體中的最后一個(或可能是單個)表達式會視為返回值。
如果我們把所有可選標(biāo)注都留下兴革,看起來如下:
val sum: (Int, Int) -> Int = { x, y -> x + y }
一個 lambda 表達式只有一個參數(shù)是很常見的绎晃。
ints.filter { it > 0 } // 這個字面值是“(it: Int) -> Boolean”類型的
我們可以使用限定的返回語法從 lambda 顯式返回一個值。否則杂曲,將隱式返回最后一個表達式的值庶艾。因此,以下兩個片段是等價的:
ints.filter {
val shouldFilter = it > 0
shouldFilter
}
ints.filter {
val shouldFilter = it > 0
return@filter shouldFilter
}
參見 callSuffix 的語法擎勘。
匿名函數(shù)
-->確實需要顯式指定落竹,可以使用另一種語法: 匿名函數(shù) 。
fun(x: Int, y: Int): Int = x + y
fun(x: Int, y: Int): Int {
return x + y
}
ints.filter(fun(item) = item > 0)
-->指定(或者已假定為 Unit
)货抄。
-->總是在用 fun{: .keyword } 關(guān)鍵字聲明的函數(shù)中返回述召。這意味著 lambda 表達式中的 return{: .keyword }
將從包含它的函數(shù)返回,而匿名函數(shù)中的 return{: .keyword }
將從匿名函數(shù)自身返回蟹地。
閉包
Lambda 表達式或者匿名函數(shù)(以及局部函數(shù)和對象表達式)
可以訪問其 閉包 积暖,即在外部作用域中聲明的變量。 與 Java 不同的是可以修改閉包中捕獲的變量:
var sum = 0
ints.filter { it > 0 }.forEach {
sum += it
}
print(sum)
帶接收者的函數(shù)字面值
Kotlin 提供了使用指定的 接收者對象 調(diào)用函數(shù)字面值的功能怪与。
在函數(shù)字面值的函數(shù)體中夺刑,可以調(diào)用該接收者對象上的方法而無需任何額外的限定符。
這類似于擴展函數(shù)分别,它允許你在函數(shù)體內(nèi)訪問接收者對象的成員遍愿。
其用法的最重要的示例之一是類型安全的 Groovy-風(fēng)格構(gòu)建器。
這樣的函數(shù)字面值的類型是一個帶有接收者的函數(shù)類型:
sum : Int.(other: Int) -> Int
該函數(shù)字面值可以這樣調(diào)用耘斩,就像它是接收者對象上的一個方法一樣:
1.sum(2)
匿名函數(shù)語法允許你直接指定函數(shù)字面值的接收者類型沼填。
如果你需要使用帶接收者的函數(shù)類型聲明一個變量,并在之后使用它括授,這將非常有用坞笙。
val sum = fun Int.(other: Int): Int = this + other
當(dāng)接收者類型可以從上下文推斷時岩饼,lambda 表達式可以用作帶接收者的函數(shù)字面值。
class HTML {
fun body() { …… }
}
fun html(init: HTML.() -> Unit): HTML {
val html = HTML() // 創(chuàng)建接收者對象
html.init() // 將該接收者對象傳給該 lambda
return html
}
html { // 帶接收者的 lambda 由此開始
body() // 調(diào)用該接收者對象的一個方法
}