有些時(shí)候，可能需要通過 Objective-C 實(shí)現(xiàn)一個(gè)接收格式化字符串可變參數(shù)的函數(shù)驯镊，如 Foundation 中的某些方法一樣：
FOUNDATION_EXPORT void NSLog(NSString *format, ...) NS_FORMAT_FUNCTION(1,2) NS_NO_TAIL_CALL;
- (instancetype)initWithFormat:(NSString *)format, ... NS_FORMAT_FUNCTION(1,2);
實(shí)現(xiàn)變參函數(shù)需要用到C語言中關(guān)于變參的一組宏：va_star、va_arg塞蹭、va_end趾代。va 是可變參數(shù) variable argument 的意思障般。使用方式為：
- (void)vaTestMethod:(NSInteger)unavaliable, ... {
va_list ap;
va_start(ap, unavaliable);
while (YES) {
NSString *string = va_arg(ap, NSString*);
if (!string) {
break;
}
NSLog(@"%@",string);
}
va_end(ap);
}
- (void)test {
[self vaTestMethod:0, @"1", @"2", @"3", nil];
}
- 聲明一個(gè)
va_list類型的變量缚窿，如 ap棘幸，這個(gè)變量是指向參數(shù)的指針。 - 用
va_start宏初始化變量 ap滨攻，這個(gè)宏的第二個(gè)參數(shù)是可變參數(shù)的前一個(gè)固定參數(shù)够话。這就使得我們實(shí)現(xiàn)的函數(shù)在可變參數(shù)前蓝翰，必須至少包含一個(gè)固定參數(shù)光绕。 - 用
va_arg返回可變的參數(shù)女嘲，這個(gè)宏的第二個(gè)參數(shù)是你要返回的參數(shù)的類型。 - 用
va_end宏結(jié)束可變參數(shù)的獲取诞帐。
從上例可以看出欣尼，這種方式的使用場(chǎng)景十分有限。首先這組宏沒有提供對(duì)參數(shù)個(gè)數(shù)的檢測(cè)停蕉，只能通過在參數(shù)末尾傳入 nil愕鼓，或者像NSLog函數(shù)一樣，根據(jù)第一個(gè)固定參數(shù) format 來判斷參數(shù)個(gè)數(shù)慧起。另外菇晃，需要在函數(shù)體內(nèi)知道可變參數(shù)中的每個(gè)參數(shù)的類型，同樣需要通過固定參數(shù) format 來獲取相關(guān)信息蚓挤。
所以可以通過這種方式實(shí)現(xiàn)一個(gè)接收格式化字符串可變參數(shù)的函數(shù)磺送。比如可以在自定義 log 函數(shù)的實(shí)現(xiàn)中應(yīng)用。NSString提供了方法- initWithFormat:arguments:直接接收 format 和 ap 參數(shù)轉(zhuǎn)換成 string 對(duì)象灿意，從而無需開發(fā)者自己根據(jù) format 判斷要獲取的參數(shù)類型和數(shù)量估灿。
- (void)log:(NSString *)format, ... {
va_list ap;
va_start(ap, format);
NSString *information = [[NSString alloc] initWithFormat:format arguments:ap];
va_end(ap);
fprintf(stderr,"%s\n", [information UTF8String]);
}
va 宏原理
C語言的函數(shù)參數(shù)是以棧這種數(shù)據(jù)結(jié)構(gòu)來存取的，在函數(shù)參數(shù)列表中缤剧，從右至左依次入棧存入?yún)?shù)的內(nèi)存地址馅袁，我們運(yùn)行va_start(ap, v)后，ap就指向第一個(gè)可變參數(shù)在棧的地址荒辕，然后我們用va_arg(ap, t)取得類型t的可變參數(shù)值汗销。之后 ap 就會(huì)指向這個(gè)參數(shù)后的地址。最后通過va_end使 ap 不再指向棧抵窒。
