Swift-闭包

599 阅读14分钟

闭包是自包含的函数代码块,可以在代码中杯传递和使用。Swift中的闭包与C和OC中的代码块(blocks)比较相似。 闭包可以捕获和储存所在上下文中任意常量和变量的应用。被称为包裹常量和变量。 闭包采取下面三种形式之一:

  • 全局函数是一个有名字但不会捕获任何值的闭包
  • 嵌套函数是一个有名字并可以捕获其封装函数域内值的闭包
  • 闭包表达式是一个利用轻量级语法所写的可以捕获其上下文中变量或常量值的匿名闭包

Swift的闭包表达式拥有简洁的风格,并鼓励在常见场景中进行语法优化,主要优化如下:

  • 利用上下文推断参数和返回值类型
  • 隐式返回单表达式闭包,即单表达式闭包可以省略return关键字
  • 参数名称缩写
  • 尾随闭包语法

本文涉及的内容有闭包表达式尾随闭包值捕获闭包是引用类型逃逸闭包自动闭包


闭包表达式

闭包表达式是一种简洁语法构建内联闭包的方式。闭包表达式提供了一些语法优化,使得撰写闭包变得简单明了。下面闭包表达式的例子通过几次迭代展示了sorted(by:)方法定义和语法优化的方式。每一次迭代都用更简洁的方式描述了相同的功能。

排序方法

Swift标准库提供了名为sorted(by:)的方法,它会根据你所提供的用于排序的闭包函数将已知类型数组中的值进行排序。一旦排序完成,sorted(by:)方法会返回一个与原数组大小相同,包含同类型元素且元素已正确排序的新数组。原数组不会被sorted:(by:)方法修改。

下面的闭包表达式示例使用sorted:(by:)方法对一个String类型的数组进行字母逆序排列。如下:

let names = ["Chris", "Alex", "Ewa", "Barry", "Daniella"]
func backward(_ s1: String, _ s2: String) -> Bool {
    return s1 > s2
}
var reversedNames = names.sorted(by: backward)
// reversedNames 为 ["Ewa", "Daniella", "Chris", "Barry", "Alex"]

sorted(by:)方法接受一个闭包,该闭包函数需要传入数组元素类型相同的两个值,并返回一个布尔类型值来表明当排序结束后传入的第一个参数排在第二个参数后面还是前面,如果第一个参数值出现在第二个参数值前面,排序闭包函数需要返回trun,反之返回falsebackward(_:_:)是一个(String, String) -> Bool,如果第一个字符串(s1)大于第二个字符串(s2),backward(_:_:) 函数会返回 true,表示在新的数组中 s1 应该出现在 s2 前。

闭包表达式语法

闭包表达式语法有如下一般形式:

{ (parameters) -> return type in
    statements
}

闭包表达式参数可以是in-out参数,但不能设定默认值。如果你命名了可变参数,也可以使用此可变参数。元组也可以作为参数和返回值。 下面的例子展示了上面backward(_:_:)函数对应的闭包表达式版本的代码:

reversedNames = names.sorted(by: { (s1: String, s2: String) -> Bool in
    return s1 > s2
})

需要注意的是内联闭包参数和返回值类型声明与backwar(_:_:)函数类型声明相同。在这两种方式中,都写成了(s1: String, s2: String)-> Bool。然而在内联闭包表达式中,函数和返回值类型都写在大括号内,而不是大括号外。 闭包的函数体部分由关键字in引入。该关键字表示闭包的参数和返回值类型定义已经完成,闭包函数体即将开始。 由于这个闭包的函数体部分如此短,以至于可以将其改写成一行代码:

reversedNames = names.sorted(by: { (s1: String, s2: String) -> Bool in return s1 > s2 } )

该例中sorted:(by:)方法的整体调用保持不变,一对圆括号任然包裹住了方法的整个参数。然而,参数现在变成了内联闭包。

根据上下文推断类型

因为排序闭包函数是作为sorted:(by:)方法的参数传入的,swift可以推断其参数和返回值的类型。sor:(by:)方法被一个字符串数组调用,因此其参数必须是(String, String)Bool类型并不需要作为闭包表达式定义的一部分。因为所有的类型都可能被正确推断,返回箭头->和围绕在参数周围的括号也可以被省略:

reversedNames = names.sorted(by: { s1, s2 in return s1 > s2 } )
// 等价于
reversedNames = names.sorted(by: { (s1, s2) -> Bool in
    return s1 > s2 
})

实际上,通过内联闭包表达式构造的闭包作为参数传递给函数或方法时,总是能够推断出闭包的参数和返回值类型。这意味着闭包作为函数或方法的参数时,几乎不需要利用完整格式构造内联闭包。但是为了代码的可读性,还是需要完整格式的闭包。

单表达式闭包隐式返回

单行表达式闭包可以通过省略return关键字来隐式返回单行表达式的结果,上面的例子还可以写成:

reversedNames = names.sorted(by: { s1, s2 in s1 > s2 } )

在这个例子中,sorter:(by:)方法的参数类型明确了闭包必须返回一个Bool类型值。因为闭包函数体只包含了一个单一表达式s1 > s2,该表达式返回Bool类型值,因此这里没有歧义,return关键字可以省略

参数名称缩写

Swift自动为内联闭包提供了参数名称缩写功能,可以直接通过$0$1$2来顺序调用闭包的参数,以此类推。如果在闭包表达式中使用参数名称缩写,可以在闭包定义中省略参数列表,并且对应参数名称缩写的类型会通过函数类型推断。in关键字也同样可以被省略,因为此时闭包表达式完全由闭包函数体构成,如下:

reversedNames = names.sorted(by: { $0 > $1 } )

这个例子中,$0$1 表示闭包中第一个和第二个String类型的参数。

运算符方法

实际上还有一种更简短的方式来编写上面例子中的闭包表达式。Swift的String类型定义了关于大于号(>)的字符串实现,其作为一个函数接受两个String类型的参数并返回Bool类型的值。而这正好与sorted(by:)方法的参数需要的函数类型相符合。因此,你可以简单地传递一个大于号,Swift可以自动推断出你想使用大于号的字符串函数实现:

reversedNames = names.sorted(by: >)

尾随闭包

如果需要将一个很长的闭包表达式作为最后一个参数传递给函数,可以使用尾随闭包来增强函数的可读性。尾随闭包是一个书写在函数括号之后的闭包表达式,函数支持将其作为最后一个参数调用。在使用尾随闭包时,你不用写出它的参数标签:

func someFunctionThatTakesAClosure(closure: () -> Void) {
    // 函数体部分
}

// 以下是不使用尾随闭包进行函数调用
someFunctionThatTakesAClosure(closure: {
    // 闭包主体部分
})

// 以下是使用尾随闭包进行函数调用
someFunctionThatTakesAClosure() {
    // 闭包主体部分
}

方法参数的字符串排序闭包可以改写为:

reversedNames = names.sorted() { $0 > $1 }

如果闭包表达式是函数或方法的唯一参数,则当你使用尾随闭包时,你甚至可以把 () 省略掉:

reversedNames = names.sorted { $0 > $1 }

当闭包非常长以至于不能在一行中进行书写时,尾随闭包变得非常有用。举例来说,Swift的Array类型有一个map(_:)方法。这个方法获取一个闭包表达式作为其唯一参数。该闭包函数会为数组中的每一个元素调用一次,并返回该元素所映射的值。具体的映射方式和返回值类型由闭包来指定。 当提供给数字的闭包应用于每个数组元素后,map(_:)方法将返回一个新的数组,数组中包含了与原数组中的元素一一对应的映射后的值。 下例介绍了如何在map(:_)方法中使用尾随闭包将Int类型数组[16, 58, 510]转换为包含对应String类型的值数组["OneSix", "FiveEight", "FiveOneZero"]

let digitNames = [
    0: "Zero", 1: "One", 2: "Two",   3: "Three", 4: "Four",
    5: "Five", 6: "Six", 7: "Seven", 8: "Eight", 9: "Nine"
]
let numbers = [16, 58, 510]

let strings = numbers.map {
    (number) -> String in
    var number = number
    var output = ""
    repeat {
        output = digitNames[number % 10]! + output
        number /= 10
    } while number > 0
    return output
}
// strings 常量被推断为字符串类型数组,即 [String]
// 其值为 ["OneSix", "FiveEight", "FiveOneZero"]

map(_:)为数组中每一个元素调用了一次闭包表达式。不需要指定闭包的输入参数number的类型,因为可以通过要映射的数组类型进行推断。 在该例中,局部变量number的值由闭包中的number参数获得,因此可以在闭包函数体内对其进行修改(闭包或者函数的参数总是常量),闭包表达式指定了返回类型为String,以表明存储映射值的新数组类型为String。 闭包表达式在每次被调用的时候创建了一个叫做output的字符串并返回。其使用求余数运算符(number % 10)计算最后一位数字并利用digitName字典获取所隐射的字符串。这个闭包能够用于创建任意正整数的字符表示。 注: 字典digitName下标后跟着一个叹号(!),因为字典下标返回一个可选值,表面该键不存在时会查找失败。在上例中,由于可以确定number % 10总是digitName字典的有效下标,因此叹号可以用于强制解包储存在下标的可选类型的返回值中的String类型的值。

值捕获

闭包可以在其被定义的上下文中捕获常量或变量。即使定义这些常量和变量的原作用域已经不存在,闭包仍然可以在闭包函数体内引用和修改这些值。 Swift中,可以捕获值的闭包的最简单形式是嵌套函数,也就是定义在其他函数的函数体内的函数。嵌套函数可以捕获其外部函数所有的参数以及定义的常量个变量。 举个例子,这有个叫做makeIncrementer的函数。其包含了一个叫做increment的嵌套函数。嵌套函数increment()从上下文中捕获了两个值,runningTotalamount。捕获这些值之后,makeIncrementincrement作为闭包返回。每次调用increment时,其会以amount作为增量增加runningTotal的值如下:

func makeIncrementer(forIncrement amount: Int) -> () -> Int {
    var runningTotal = 0
    func incrementer() -> Int {
        runningTotal += amount
        return runningTotal
    }
    return incrementer
}

makeIncrementer返回类型为() -> Int。这意味着其返回值是一个函数,而非是一个简单类型的值。该函数在每次调用时不接受参数,只返回约一个Int类型的值。

makeIncrementer(forIncrement:) 函数定义了一个初始值为0的整形变量runningTotal,用来存储当前总计数值。该值为incrementer的返回值。

makeIncrementer(forIncrement:) 有一个Int类型的参数,其外部参数名为forIncrement,内部参数名为amount,该参数表示每次incrementer被调用时runningTotal将要增加的量。makeIncrementer函数还定义了一个嵌套函数incrementer,用来执行实际的增加操作。该函数简单的增加amount,并将其返回。

下面是一个使用 makeIncrementer 的例子:

let incrementByTen = makeIncrementer(forIncrement: 10)

该例子定义了一个叫做 incrementByTen 的常量,该常量指向一个每次调用会将其 runningTotal 变量增加 10incrementer 函数。调用这个函数多次可以得到以下结果:

incrementByTen()
// 返回的值为10
incrementByTen()
// 返回的值为20
incrementByTen()
// 返回的值为30

如果你创建了另一个 incrementer,它会有属于自己的引用,指向一个全新、独立的 runningTotal 变量:

let incrementBySeven = makeIncrementer(forIncrement: 7)
incrementBySeven()
// 返回的值为7

再次调用原来的 incrementByTen 会继续增加它自己的 runningTotal 变量,该变量和 incrementBySeven 中捕获的变量没有任何联系:

incrementByTen()
// 返回的值为40

注: 如果将闭包赋值给一个类实例的属性,并且该闭包通过访问该实例或其成员而捕获了该实例,将在闭包和该实例间创建一个循环强引用。Swift使用捕获列表来打破这种循环强引用。

闭包是引用类型

上面例子中,incrementBySevenincrementByTen 都是常量,但是这些常量只想的闭包仍然可以增加其捕获的变量的值。这是因为函数和闭包都是引用类型

无论你将函数或闭包赋值给一个常量还是变量,你实际上都是将常量或变量的值设置为对应函数或闭包的引用。上面的例子中,指向闭包的引用incrementByTen是一个常量,而并非闭包内容本身。

这也意味着如果你将闭包赋值给两个不同的常量或变量,两个值都会指向一个闭包:

let alsoIncrementByTen = incrementByTen
alsoIncrementByTen()
// 返回的值为50

逃逸闭包

当一个闭包作为参数传到一个函数中,但是这个闭包在函数返回之后才被执行,称之为该闭包从函数中逃逸。当你定义接受闭包作为参数的函数时,可以在参数名之前标注@escaping,用来指明这个闭包是允许逃逸出这个函数的。

一种能使闭包逃逸出函数的方法时,将这个闭包保存在一个函数外部定义的变量中。举个例子,很多启动异步操作的函数接受一个闭包参数作为completion handler。这类函数会在异步操作开始之后立刻返回,但是闭包知道异步操作结束后才会被调用。这种情况下,闭包需要逃逸出函数,因为闭包需要在函数返回之后被调用。例如:

var completionHandlers: [() -> Void] = []
func someFunctionWithEscapingClosure(completionHandler: @escaping () -> Void) {
    completionHandlers.append(completionHandler)
}

someFunctionWithEscapingClosure(_:)函数接受一个闭包作为参数,该闭包被添加到一个函数外定义的数组中。如果不将这个参数标记为@wscaping,就会得到一个编译错误。

将一个闭包标记为@escaping意味着必须在闭包中显示地引用self。比如说,在下面的代码中,传递到someFunctionWithEscapingClosure(_:) 中的闭包是一个逃逸闭包,这意味着它需要显示地引用self。相对的,传递到someFunctionWithEscapingClosure(_:) 中的闭包是一个非逃逸闭包,这意味着它可以隐式引用self

func someFunctionWithNonescapingClosure(closure: () -> Void) {
    closure()
}

class SomeClass {
    var x = 10
    func doSomething() {
        someFunctionWithEscapingClosure { self.x = 100 }
        someFunctionWithNonescapingClosure { x = 200 }
    }
}

let instance = SomeClass()
instance.doSomething()
print(instance.x)
// 打印出 "200"

completionHandlers.first?()
print(instance.x)
// 打印出 "100"

自动闭包

自动闭包是一种自动创建的闭包,用于包装传递给函数作为参数的表达式。这种闭包不接受任何参数,当它被调用的时候,会返回被包装在其中的表达式。这种遍历语法能扣省略闭包的花括号,用一个普通的表达式来代替显示的闭包。

我们经常会调用采用自动闭包的函数,但是很少去实现这样的函数。举个例子说,**assert(condition:message:file:line:)**函数接受自动闭包作为它的condition参数和message参数。它的condition参数仅会在debug模式下被求值,它的message参数仅当condition参数为false时被计算求值。

自耦东闭包让你能延迟求职,因为知道你调用这个闭包,代码段才会被执行。延迟求值对于那些有副作用和高计算成本的代码来说是很有益处的,因为它使得你能控制代码的执行时机。举例如下:

var customersInLine = ["Chris", "Alex", "Ewa", "Barry", "Daniella"]
print(customersInLine.count)
// 打印出 "5"

let customerProvider = { customersInLine.remove(at: 0) }
print(customersInLine.count)
// 打印出 "5"

print("Now serving \(customerProvider())!")
// Prints "Now serving Chris!"
print(customersInLine.count)
// 打印出 "4"

尽管在闭包的代码中,customersInLine的第一个元素被移除了,不过在闭包被调用之前,这个元素是不会被移除的。如果这个闭包永远不被调用,那么在闭包里面的表达式永远不被执行,那意味着列表中的元素永远不会被移除。请注意,customerProvider的类型不是String,而是() -> String,一个没有参数且返回值为String的函数。

将闭包作为参数传递给函数是,通能能够获得延时求值的行为,举例如下:

customersInLine is ["Ewa", "Barry", "Daniella"]
func serve(customer customerProvider: @autoclosure () -> String) {
    print("Now serving \(customerProvider())!")
}
serve(customer: customersInLine.remove(at: 0))
// 打印 "Now serving Ewa!"

注: 过度使用autoclosures会让代码变得很难理解。上下文和函数名应该能够清晰的表明求值是被延迟执行的。如果你想让一个自动闭包可以逃逸,则应该同时使用@autoclosureescaping属性

// customersInLine i= ["Barry", "Daniella"]
var customerProviders: [() -> String] = []
func collectCustomerProviders(_ customerProvider: @autoclosure @escaping () -> String) {
    customerProviders.append(customerProvider)
}
collectCustomerProviders(customersInLine.remove(at: 0))
collectCustomerProviders(customersInLine.remove(at: 0))

print("Collected \(customerProviders.count) closures.")
// 打印 "Collected 2 closures."
for customerProvider in customerProviders {
    print("Now serving \(customerProvider())!")
}
// 打印 "Now serving Barry!"
// 打印 "Now serving Daniella!"

在上面的代码中,collectCustomerProviders(_:)函数并没有调用传入的customerProvider闭包,而是将闭包追加到了customerProviders数组中。这个数组定义在函数作用域范围外,这意味着数组内的闭包能够在函数返回之后被调用。因此,customerProvider参数必须允许逃逸出函数作用域。