美文网首页收藏ios
[iOS-Objective-C] 格式化字符串变参函数的实现

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

作者: 水止云起 | 来源:发表于2016-11-22 18:33 被阅读554次

    有些时候,可能需要通过 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_star、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 不再指向栈。

    相关文章

      网友评论

        本文标题:[iOS-Objective-C] 格式化字符串变参函数的实现

        本文链接:https://www.haomeiwen.com/subject/thzppttx.html