前言
许多编程语言得益于编译器和自身的语法都有可变参数语法形式. Swift和Objecitve-C作为iOS开发的主要语言, 自然也有方法使用各自可变参数的语法格式.并且iOS开发中接触到可变参数的方法并不少.
Begin
首先来看看iOS开发中平常使用的可变参数函数的样子.
Swift 版本
// NSString构造方法
init(format: NSString, _ args: CVarArgType...) //CVarArgType是Swift对 C的相关API进行的封装
// 日志输出函数
print(items: Any..., separator: String = default, terminator: String = default)
Objective-C版本
// 日志输出
NSLog(NSString *format, ...);
// NSString实例的创建
+(instancetype)stringWithFormat:(NSString *)format, ...;
// NSArray实例的创建
+(instancetype)arrayWithObjects:(ObjectType)firstObj, ... NS_REQUIRES_NIL_TERMINATION;
Next
现在看看怎么来实现可变参数方法了.
Swift
由于语法的灵活加上编译器的优化支持,实现可变参数方法特别简单.模仿官方的写法,也能写出简单的可变参数方法, 并且不需要将可变参数放置参数列表末尾
func sum(input: Int..., log: Bool) -> Int{
var sum = 0
for num in input {
sum += num
}
//input实为[Int], reduce为数组的方法
if log {
print("the sum is \(sum)")
}
return sum
}
let total = sum(1,3,4,5,6, log: true)
// total = 19
// the sum is 19
class Person {
let name:String
var age: Int
init(name:String, age: Int) {
self.name = name
self.age = age
}
}
func totalAge(peoples: Person...) -> Int {
let age = peoples.reduce(0) { (a: Int, person: Person) -> Int in
return person.age + a
}
return age
}
let haha = Person(name: "haha", age: 11)
let wrcj = Person(name: "wrcj", age: 22)
let tim = Person(name: "tim", age: 33)
let totalAges = totalAge(haha, wrcj, tim)
// totalAges = 66
接下来就是Objective-C版本,先直接上代码.
@interface Person : NSObject
@property (nonatomic, readwrite, assign) NSInteger age;
@end
@implementation Person
- (instancetype)initAge:(NSInteger)age {
self = [super init];
if (self) {
_age = age;
}
return self;
}
+ (NSInteger)totalAge: (nonnull Person* )person, ...NS_REQUIRES_NIL_TERMINATION {
NSInteger totalAge = 0;
va_list people; // C语言的字符指针, 指针根据offset来指向需要的参数,从而读取参数
va_start(people, person); // 设置指针的起始地址为方法的...参数的第一个参数
if (person) { // 第一个参数 person totalAge = person.age;
for(;;) {
Person *person = va_arg(people, Person *); // 获取当前va_list指针指向的参数, 并以Person对象内存大小的偏移量移向下一个参数 if (!person) {
break; // 当参数取完,跳出循环
}
totalAge += person.age;
}
va_end(people); // 针对va_start进行的安全处理,将va_list指向Null.
return totalAge;
}
@end
// Test
int main(int argc, const char * argv[]) {
@autoreleasepool {
Person *haha = [[Person alloc]initAge:11];
Person *wrcj = [[Person alloc]initAge:22];
Person *tim = [[Person alloc]initAge:33];
NSInteger totalAge = [Person totalAge:haha, tim, wrcj, nil];
// totalAge = 66;
}
return 0;
}
上面明显可以看出,OC中的可变参数就是利用C语言API的va_list, va_arg,va_end, va_start,利用一个特别的指针通过偏移地址,指向不同参数.而为了保证通过偏移后指向正确的参数, 必须要求传入的可变参数必须是同一种类型的, 使得指针根据固定的地址偏移量来指向下一个参数.
End
在我理解中, 向函数传入可变参数就好比把这些参数放进一个数组,利用�数组元素的内存地址连续存储的特点,内部遍历数组进行逐一访问. 而OC中利用va_list指针根据相同的偏移量依次获取参数,就像指向一个有相同类型元素的数组首地址的指针,通过指针正确的偏移量取出数组的元素.