10.3 使用闭包表达式

Swift中的闭包表达式很灵活,其标准语法格式如下:

  1. { (参数列表) ->返回值类型 in
  2. 语句组
  3. }

其中,参数列表与函数中的参数列表形式一样,返回值类型类似于函数中的返回值类型,但不同的是后面有in关键字。

Swift提供了多种闭包简化写法,这一节我们将介绍几种不同的形式。

10.3.1 类型推断简化

类型推断是Swift的强项,Swift可以根据上下文环境推断出参数类型和返回值类型。以下代码是标准形式的闭包:

  1. {(a:Int, b:Int) -> Int in
  2. return a + b
  3. }

Swift能推断出参数abInt类型,返回值也是Int类型。简化形式如下:

  1. {a, b in return a + b }

使用这种简化方式修改后的示例代码如下:

  1. func calculate(opr :String)-> (Int,Int)-> Int {
  2. var result : (Int,Int)-> Int
  3. switch (opr) {
  4. case "+" :
  5. result = {a, b in return a + b }
  6. default:
  7. result = {a, b in return a - b }
  8. }
  9. return result
  10. }
  11. let f1:(Int,Int)-> Int = calculate("+")
  12. println("10 + 5 = \(f1(10,5))")
  13. let f2:(Int,Int)-> Int = calculate("-")
  14. println("10 + 5 = \(f2(10,5))")

上述代码第①行和第②行的闭包是上一节示例的简化写法,其中ab是参数,return后面是返回值。怎么样?很简单吧?

10.3.2 隐藏return关键字

如果在闭包内部语句组只有一条语句,如return a + b等,那么这种语句都是返回语句。前面的关键字return可以省略,省略形式如下:

  1. {a, b in a + b }

使用这种简化方式修改后的示例代码如下:

  1. func calculate(opr :String)-> (Int,Int)-> Int {
  2. var result : (Int,Int)-> Int
  3. switch (opr) {
  4. case "+" :
  5. result = {a, b in a + b }
  6. default:
  7. result = {a, b in a - b }
  8. }
  9. return result
  10. }

上述代码第①行和第②行的闭包return关键字省略了,需要注意的是,省略的前提是闭包中只有一条return语句。下面这样有多条语句是不允许的。

  1. {a, b in var c; a + b }

10.3.3 缩写参数名称

上一节介绍的闭包表达式已经很简洁了,不过,Swift的闭包还可以再进行简化。Swift提供了参数名称缩写功能,我们可以用$0$1$2来表示调用闭包中参数,$0指代第一个参数,$1指代第二个参数,$2指代第三个参数,以此类推$n+1指代第n个参数。

使用参数名称缩写,还可以在闭包中省略参数列表的定义,Swift能够推断出这些缩写参数的类型。此外,in关键字也可以省略。参数名称缩写之后如下所示:

  1. {$0 + $1}

使用参数名称缩写修改后的示例代码如下:

  1. func calculate(opr :String)-> (Int,Int)-> Int {
  2. var result : (Int,Int)-> Int
  3. switch (opr) {
  4. case "+" :
  5. result = {$0 + $1}
  6. default:
  7. result = {$0 - $1}
  8. }
  9. return result
  10. }
  11. let f1:(Int,Int)-> Int = calculate("+")
  12. println("10 + 5 = \(f1(10,5))")
  13. let f2:(Int,Int)-> Int = calculate("-")
  14. println("10 + 5 = \(f2(10,5))")

上述代码第①行和第②行的闭包采用了参数名称缩写。

10.3.4 使用闭包返回值

闭包表达本质上是函数类型,是有返回值的,我们可以直接在表达式中使用闭包的返回值。重新修改addsub闭包,示例代码如下:

  1. let c1:Int = {(a:Int, b:Int) -> Int in
  2. return a + b
  3. }(10,5)
  4. println("10 + 5 = \(c1)")
  5. let c2:Int = {(a:Int, b:Int) -> Int in
  6. return a - b
  7. }(10,5)
  8. println("10 - 5 = \(c2)")

上述代码有两个表达式,第①行代码是给c1赋值,后面是一个闭包表达式。但是闭包表达式不能直接赋值给c1,因为c1Int类型,需要闭包的返回值。这就需要在闭包结尾的大括号后面接一对小括号(10,5),通过小括号(10,5)为闭包传递参数。第②行代码也是如此。通过这种方法可以为变量和常量直接赋值,在有些场景下使用非常方便。