Swift 的 Currying 特性 | SwiftCafe 咖啡时间

Currying 也是 Swift 的众多先进特性之一，用一句话说就是将接受多个参数的函数，转变成每次之接受一个参数的调用序列。

上面一句话说得可能大家感觉不是那么清楚，那么没关系，咱们举一个例子来说明吧。

比如，我们需要一个两个数字相加的函数，我们可以这样定义：

func add(a:Int, b:Int) -> Int {

    return a + b

}

那么，我们直接这样调用，就可以得到结果了：

add(1, b: 2)

这个是最基本的使用角度，那么假如我们把思维提升一个层级，比如:

let numbers = 1...10

let added = numbers.map {

    return add($0, b: 2)

}

这里我们先用 1...10 声明了一个整数数组，然后用它的 map 方法将数组中所有的元素都加 2 然后生成一个新的数组。

注： add($0, b: 2) 中的 $0 表示当前遍历到得数组元素，把它传递给 add 函数并且加 2 后返回，生成新的数组元素。

这样做好像也没什么问题，但实际上我们完全可以不使用闭包来处理 map 中的回调。我们还可以这样：

func add(a:Int) -> (Int -> Int) {

    return { b in
        return a + b
    }

}

let numbers = 1...10
let added = numbers.map (add(2))

怎么样，有没有眼花缭乱呢？ 怎么 map 函数调用变成这个样子的呢：numbers.map (add(2))

相信手机或电脑前的一定很聪明，能看出端倪。

实际上我们是把 add 函数的定义做了修改：

func add(a:Int) -> (Int -> Int)

我们的 add 函数现在只接受一个参数了。并且呢，注意一下它的返回值类型：

(Int -> Int)

在 Swift 中，这种定义格式，表示的是一个函数类型。也就是说，我们的 add 函数它除了接受一个 Int 类型的参数，它还会返回一个函数类型的闭包对象。

接着，再看看这个函数的实现：

func add(a:Int) -> (Int -> Int) {

    return { b in
        return a + b
    }

}

确实，返回的是一个闭包对象，接受了一个参数 b，然后这个闭包对象，继续将 a 和 b 两个参数的值再次返回。

关于闭包的概念，可以参看咱们之前讨论过的文章：了解闭包

这么一来，就明了啦，add 函数其实是一个返回函数的函数。也可以把它看做一个函数生成器。它接受的参数作为它生成的函数的一个标准：

let addTwo = add(2)
let addThree = add(3)
let addFive = add(5)

就像上面的代码那样，给 add 函数传入不同的参数，它就会依照这个参数值，生成一个新的函数，而这个就会对它接受的参数与之前 add 指定的增量进行加法操作了。

嗯，绕了这么一大圈，我们刚才讨论的这种函数行为，就叫做 currying。

对于 add 函数呢，我们还可以有另外一种写法：

func add(a: Int)(b: Int) -> Int {

    return a + b

}

这样写出来的 add 函数以，就显得直观一些了。

然后，我们依然可以像这样，用它来生成各种新的函数：

let addTwo = add(2)
let addThree = add(3)
let addFive = add(5)

还可以向这样，把它作为另一个回调的传入：

let numbers = 1...10
let added = numbers.map (add(2))

总之呢，Swift 诞生之初，就有着它先天的强大与灵活。