有些時(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語(yǔ)言中關(guān)于變參的一組宏:va_start、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ì)象,從而無(wú)需開發(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語(yǔ)言的函數(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 不再指向棧。