iOS实现传递不定长的多个参数

西门桃桃 2022-03-28 PM 821℃ 0条

背景

在使用系统UIAlertView的时候会发现otherButtonTitles:后边可以传不定数的多个参数例如:

// [UIAlertView alloc] initWithTitle:<#(nullable NSString *)#> message:<#(nullable NSString *)#> delegate:<#(nullable id)#> cancelButtonTitle:<#(nullable NSString *)#> otherButtonTitles:<#(nullable NSString *), ...#>, nil
UIAlertView *alert = [[UIAlertView alloc]initWithTitle:@"title" message:@"message" delegate:self cancelButtonTitle:@"cancel" otherButtonTitles:@"other1",@"other2",@"other3", nil ];
[alert show];

那么如何实现这种效果?

方案一:va_list传递不定长的多个参数

/*iOS实现传递不定长的多个参数的方法是使用va_list。va_list是C语言提供的处理变长参数的一种方法。在调用的时候要在参 数结尾的时候加nil
  va_list的使用需要注意:        
  1.首先在函数里定义va_list型的变量,这个变量是指向参数的指针;        
  2.然后用va_start初始化刚定义的va_list变量;        
  3.然后用va_arg返回可变的参数,va_arg的第二个参数是你要返回的参数的类型.如果函数有多个可变参数的,依次调用va_arg获取各个参数;        
  4.最后用va_end宏结束可变参数的获取                
  NS_REQUIRES_NIL_TERMINATION,是一个宏,用于编译时非nil结尾的检查。 调用时要以nil结尾,否则会崩溃。  
*/
- (void)testParams:(NSString *)title addMoreParams:(NSString *)string, ...NS_REQUIRES_NIL_TERMINATION {
    NSLog(@"传多个参数的第一个参数 %@",string);//是other1
    //1.定义一个指向个数可变的参数列表指针;
    va_list args;
    //2.va_start(args, str);string为第一个参数,也就是最右边的已知参数,这里就是获取第一个可选参数的地址.使参数列表指针指向函数参数列表中的第一个可选参数,函数参数列表中参数在内存中的顺序与函数声明时的顺序是一致的。
    va_start(args, string);
    if (string) {
        //依次取得除第一个参数以外的参数
        //4.va_arg(args,NSString):返回参数列表中指针所指的参数,返回类型为NSString,并使参数指针指向参数列表中下一个参数。
        while (va_arg(args, NSString *)) {
            NSString *otherString = va_arg(args, NSString *);
            NSLog(@"otherString %@",otherString);
        }
    }
    //5.清空参数列表,并置参数指针args无效。
    va_end(args);
}

方案二:performSelector多参数传递解决方案

1、使用NSInvocation进行消息转发从而实现对performSelector的多参数传递:

/*以下是对NSObject类的扩展方法:
  NS_REQUIRES_NIL_TERMINATION:是对多参数传递值得一个宏
  va_list args:定义一个指向个数可变的参数列表指针;
  va_start(args,object):object为第一个参数,也就是最右边的已知参数,这里就是获取第一个可选参数的地址.使参数列表指针指向函数参数列表中的第一个可选参数,函数参数列表中参数在内存中的顺序与函数声明时的顺序是一致的。
  va_arg(args,id):返回参数列表中指针所指的参数,返回类型为id,并使参数指针指向参数列表中下一个参数。
  va_end(args):清空参数列表,并置参数指针args无效。
*/

- (void)viewDidLoad {
    [super viewDidLoad];
    id object = [self glt_performSelector:@selector(testMoreArg:arg2:arg3:) withObject:@"username",@"sex",@"height", nil];
    NSLog(@"%@",object);
}

- (NSString *)testMoreArg:(NSString *)arg1 arg2:(NSString *)arg2 arg3:(NSString *)arg3 {
    NSLog(@"arg1 = %@ arg2 = %@ arg3 = %@", arg1, arg2, arg3);
    return [NSString stringWithFormat:@"%@%@%@", arg1, arg2, arg3];
}

- (id)glt_performSelector:(SEL)selector withObject:(id)object, ... NS_REQUIRES_NIL_TERMINATION {
    //根据类名以及SEL 获取方法签名的实例
    NSMethodSignature *signature = [[self class] instanceMethodSignatureForSelector:selector];
    if (signature == nil) {
        NSLog(@"--- 使用实例方法调用 为nil ---");
        signature = [self methodSignatureForSelector:selector];
        if (signature == nil) {
            NSLog(@"使用类方法调用 也为nil, 此时return");
            return nil;
        }
    }
    //NSInvocation是一个消息调用类,它包含了所有OC消息的成分:target、selector、参数以及返回值。
    NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:signature];
    invocation.target = self;
    invocation.selector = selector;
    NSUInteger argCount = signature.numberOfArguments;
    // 参数必须从第2个索引开始,因为前两个已经被target和selector使用
    argCount = argCount > 2 ? argCount - 2 : 0;
    NSMutableArray *objs = [NSMutableArray arrayWithCapacity:0];
    if (object) {
        [objs addObject:object];
        va_list args;
        va_start(args, object);
        while ((object = va_arg(args, id))) {
            [objs addObject:object];
        }
        va_end(args);
    }
    if (objs.count != argCount) {
        NSLog(@"--- objs.count != argCount! please check it! ---");
        return nil;
    }
    //设置参数列表
    for (NSInteger i = 0; i < objs.count; i++) {
        id obj = objs[i];
        if ([obj isKindOfClass:[NSNull class]]) {
            continue;
        }
        [invocation setArgument:&obj atIndex:i + 2];
    }
    [invocation invoke];
    //获取返回值
    if (signature.methodReturnLength != 0 && signature.methodReturnLength) {
        void *returnValue;
        [invocation getReturnValue:&returnValue];
        return (__bridge id)returnValue;
    }
    return nil;
}

2、使用runtime中的objc_msgSend进行消息的发送:

objc_msgSend(void /* id self, SEL op, ... */ )

在一个类中有以下两个方法:

@implementation CustomClass
 
-(void)fun {
    NSLog(@"fun");
}

-(void)eat:(NSString *)food say:(NSString *)some {
    NSLog(@"%@ %@",food, some);
}
 
@end

当我们使用的时候,使用objc_msgSend进行调用,如下:这样便调用了类中的fun方法

TestClass *cls = [[TestClass alloc] init];

objc_msgSend(cls, @selector(fun)); //错误写法(arm64崩溃偶尔发生)

((void (*)(id, SEL))objc_msgSend)(cls, @selector(fun)); //正确写法

解决 objc_msgSend arm64 崩溃问题:

1、解决 objc_msgSend crash

1.png

2、解决 objc_msgSend arm64 崩溃

按照文档 64-Bit Transition Guide for Cocoa Touch 给出了以下代码片段:

- (int) doSomething:(int) x { ... }
- (void) doSomethingElse {
    int (*action)(id, SEL, int) = (int (*)(id, SEL, int)) objc_msgSend;
    action(self, @selector(doSomething:), 0);
}

所以必须先定义原型才可以使用,这样才不会发生崩溃,调用的时候则如下:

void (*glt_msgsend)(id, SEL, NSString *, NSString *) = (void (*)(id, SEL, NSString *, NSString *))objc_msgSend;
 
glt_msgsend(cls, @selector(eat:say:), @"123", @"456");

参考:

iOS performSelector多参数传递解决方案以及objc_msgSend的使用注意事项_高刘通的博客-CSDN博客

runtime - 消息发送(objc_msgSend) - 简书

标签: runtime

非特殊说明,本博所有文章均为博主原创。

评论啦~