iOS 通过反射的方式调用目标函数

1. 目标

工作中，需要解决这样一种问题，测试每一个目标接口的调用时间和返回值。逻辑较为通用，所以想使用反射来实现这种功能。

2. 背景知识

网上查了下，发现反射有两种实现方式，分别为 runtime中的objc_msgSend 和 NSInvocation。

2.1 objc_msgSend

具体可参考这篇文章：https://www.jianshu.com/p/ce00728204ed?utm_campaign

试用了下，发现可以实现函数调用，但不太好额外加入其他逻辑，所以不太适用于解决我的问题。

2.2 NSInvocation

具体可参考这篇文章：https://juejin.cn/post/6844903542851895304 （虽然文中提到了NSProxy，但看实际代码里并未使用到） 和 https://ace.re/2017/objective-c-nsinvocation.html

试用了下，发现基本可以达成我的目标。但有一个问题，系统提供的performSelector函数，最多仅支持传两个参数。这显然是不够的。所以我们需要把这块优化下。我尝试了两种解决方式。

2.2.1 传参使用NSArray

具体可参考这个篇文章：https://blog.csdn.net/a158337/article/details/50915245

2.2.2 使用可变参数传参

具体可参考这篇文章：https://www.jianshu.com/p/1dc0d05cb766。基本可用，但文章中获取返回值处需要优化下。看网上的说法，文章中获取返回值处可能会造成崩溃。

3. 我的实现

3.1 使用可变参数传参

@implementation InterfaceProxy

//方法签名
- (id)methodSignatureForSelector:(SEL)sel {
  NSMethodSignature *signature = nil;
  if (self.obj) {
    return signature;
  }
  signature = [self.obj methodSignatureForSelector:sel];
  return signature ? signature : [super methodSignatureForSelector:sel];
}

- (NSDictionary *)performSelector:(NSObject *)obj sel:(SEL)sel withObject:(id)object,...NS_REQUIRES_NIL_TERMINATION {
  self.obj = obj;
  NSMethodSignature *signature = [self methodSignatureForSelector:sel];

  //根据类名以及SEL 获取方法签名的实例
  if (signature == nil) {
    NSLog(@"--- 未找到符合条件的方法 ---");
    return nil;
  }

  //NSInvocation是一个消息调用类，它包含了所有OC消息的成分：target、selector、参数以及返回值。
  NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:signature];
  invocation.target = self.obj;
  invocation.selector = sel;
  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(@"proxy --- 传参有误! please check it! ---");
     return nil;
  }
  //设置参数列表
  for (NSInteger i = 0; i < objs.count; i++) {
     id obj = objs[i];
     [invocation setArgument:&obj atIndex:i+2];
  }

  NSTimeInterval start = [TimeUtil currentTimeStr];
  [invocation invoke];
  NSTimeInterval stop = [TimeUtil currentTimeStr];

  NSString *selName = NSStringFromSelector(invocation.selector);
  NSLog(@"proxy selName: %@, start: %f stop:%f interval: %f", selName, start, stop, (stop - start));
  //获取返回值
  NSUInteger length = [signature methodReturnLength];
  const char *retType = signature.methodReturnType;
  id retVal;
  if (signature.methodReturnLength != 0 && signature.methodReturnLength) {
    if (!strcmp(retType, @encode(id))) {
      [invocation getReturnValue:&retVal];
    } else {
      void *buf = (void *)malloc(length);
      [invocation getReturnValue:buf];
      retVal = [self transRet:buf retType:retType];
    }
  }
  NSDictionary *dic = [[NSDictionary alloc] initWithObjectsAndKeys:selName,SELECTOR_NAME,
                       objs, INTERFACE_PARAMS,
                       [NSNumber numberWithDouble:start], START_TIME_STAMP,
                       [NSNumber numberWithDouble:stop], STOP_TIME_STAMP,
                       retVal, RET,
                       nil];
  return dic;
}

- (id)transRet:(void *)returnValue retType:(const char *)returnType {
  if (!strcmp(returnType, @encode(int))) {
    return [NSNumber numberWithInteger:*((NSInteger*)returnValue)];
  }
  if (!strcmp(returnType, @encode(BOOL))) {
    int tmp = *((BOOL *)returnValue) ? 0 : -1;
    return [NSNumber numberWithInt:tmp];
  }
  return nil;
}

@end

　　

调用方法：

InterfaceProxy proxy = [InterfaceProxy alloc];
NSDictionary *dic = [proxy performSelector:self->_member sel:@selector(funcName:param1:param2:) withObject:@"param0",@"param1",@"param2", nil];

 

4. 待优化的问题

此方法可适用于大多数场景，但有些地方仍不尽如人意。其中之一，就是传参必须是NSObject的子类。如果是C++的类，就传不进去。因为无法将其转为id。其二，不能方便的调用C语法的函数。只有成员方法或类方法，可以这样方便的调用。若开发中需调用的接口为C语法的接口，则调用起来会麻烦很多，不如将C语法的接口，包装成OC的风格。