iOS开发中使用可变参数

前言

许多编程语言得益于编译器和自身的语法都有可变参数语法形式. 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指针根据相同的偏移量依次获取参数,就像指向一个有相同类型元素的数组首地址的指针,通过指针正确的偏移量取出数组的元素.