Swift中的可变参数函数介绍

时间:2021-07-28 15:15:21

可变参数函数指的是可以接受任意多个参数的函数,我们最熟悉的可能就是 NSString 的 -stringWithFormat:方法了。在 Objective-C 中,我们使用这个方法生成字符串的写法是这样的:

复制代码 代码如下:

NSString *name = @"Tom";
NSDate *date = [NSDate date];
NSString *string = [NSString stringWithFormat:
                @"Hello %@. Date: %@", name, date];


这个方法中的参数是可以任意变化的,参数的第一项是需要格式化的字符串,后面的参数都是向第一个参数中填空。在这里我们不再详细描述 Objective-C 中可变参数函数的写法 (毕竟这是一本 Swift 的书),但是我相信绝大多数即使有着几年 Objective-C 经验的读者,也很难在不查阅资料的前提下正确写出一个接受可变参数的函数。

 

但是这一切在Swift中得到了前所未有的简化。现在,写一个可变参数的函数只需要在声明参数时在类型后面加上 ... 就可以了。比如下面就声明了一个接受可变参数的 Int 累加函数:

复制代码 代码如下:

func sum(input: Int...) -> Int {
    //...
}


输入的 input 在函数体内部将被作为数组 [Int] 来使用,让我们来完成上面的方法吧。当然你可以用传统的 for...in 做累加,但是这里我们选择了一种看起来更 Swift的方式:

复制代码 代码如下:


func sum(input: Int...) -> Int {
    return input.reduce(0, combine: +)
}

 

println(sum(1,2,3,4,5))
// 输出:15


在使用可变参数时需要注意的是可变参数只能作为方法中的最后一个参数来使用,而不能先声明一个可变参数,然后再声明其他参数。这是很容易理解的,因为编译器将不知道输入的参数应该从哪里截断。另外,在一个方法中,最多只能有一组可变参数。

 

一个比较恼人的限制是可变参数都必须是同一种类型的,当我们想要同时传入多个类型的参数时就需要做一些变通。比如最开始提到的-stringWithFormat:方法。可变参数列表的第一个元素是等待格式化的字符串,在 Swift 中这会对应一个 String 类型,而剩下的参数应该可以是对应格式化标准的任意类型。一种解决方法是使用 Any 作为参数类型,然后对接收到的数组的首个元素进行特殊处理。不过因为 Swift 提供了使用下划线 _ 来作为参数的外部标签,来使调用时不再需要加上参数名字。我们可以利用这个特性,在声明方法是就指定第一个参数为一个字符串,然后跟一个匿名的参数列表,这样在写起来的时候就 "好像" 是所有参数都是在同一个参数列表中进行的处理,会好看很多。比如Swift 的 NSString 格式化的声明就是这样处理的:

复制代码 代码如下:

extension NSString {
    convenience init(format: NSString, _ args: CVarArgType...)
    //...
}


调用的时候就和在 Objective-C 时几乎一样了,非常方便:

复制代码 代码如下:

let name = "Tom"
let date = NSDate()
let string = NSString(format: "Hello %@. Date: %@", name, date)