[iOS-Objective-C] 格式化字符串变参函数的实现

有些时候,可能需要通过 Objective-C 实现一个接收格式化字符串可变参数的函数,如 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);

实现变参函数需要用到C语言中关于变参的一组宏:va_start、va_arg、va_end。va 是可变参数 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];
}
  1. 声明一个va_list类型的变量,如 ap,这个变量是指向参数的指针。
  2. va_start宏初始化变量 ap,这个宏的第二个参数是可变参数的前一个固定参数。这就使得我们实现的函数在可变参数前,必须至少包含一个固定参数。
  3. va_arg返回可变的参数,这个宏的第二个参数是你要返回的参数的类型。
  4. va_end宏结束可变参数的获取。

从上例可以看出,这种方式的使用场景十分有限。首先这组宏没有提供对参数个数的检测,只能通过在参数末尾传入 nil,或者像NSLog函数一样,根据第一个固定参数 format 来判断参数个数。另外,需要在函数体内知道可变参数中的每个参数的类型,同样需要通过固定参数 format 来获取相关信息。

所以可以通过这种方式实现一个接收格式化字符串可变参数的函数。比如可以在自定义 log 函数的实现中应用。NSString提供了方法- initWithFormat:arguments:直接接收 format 和 ap 参数转换成 string 对象,从而无需开发者自己根据 format 判断要获取的参数类型和数量。

- (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语言的函数参数是以栈这种数据结构来存取的,在函数参数列表中,从右至左依次入栈存入参数的内存地址,我们运行va_start(ap, v)后,ap就指向第一个可变参数在栈的地址,然后我们用va_arg(ap, t)取得类型t的可变参数值。之后 ap 就会指向这个参数后的地址。最后通过va_end使 ap 不再指向栈。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 我们在C语言编程中会遇到一些参数个数可变的函数,例如printf() 这个函数,它的定义是这样的: int pri...
    Mr_Me阅读 4,090评论 0 2
  • 代码: -(NSString*)makeDrink:(NSString*)drink Fruit:(NSStrin...
    bada阅读 5,770评论 0 0
  • 1.在C中,变长参数的函数即参数个数可变、参数类型不定 的函数。当我们无法列出传递函数的所有实参的类型和数目时,以...
    孤独杂货铺阅读 5,781评论 1 1
  • 不管我们天生是什么样子,是什么状况,父母给我们带来了什么。我们所遭遇的一切痛苦与快乐,其最终我们都是在自觉与...
    烽火煤阅读 1,163评论 0 0
  • 曾经意气风发特立独行如我, 自由浪漫固执感性如我 一场似曾相识的缘分在最不经意时相遇 于是,美好如意料之外的命中注...
    鱼与瑜阅读 2,948评论 2 2