iOS-Swizzle

最后更新:2017-06-21

一、先说结论

void swizzleMethod(Class cls, SEL originalSelector, SEL swizzledSelector)
{
    Method originalMethod = class_getInstanceMethod(cls, originalSelector);
    Method swizzledMethod = class_getInstanceMethod(cls, swizzledSelector);

    BOOL didAddMethod =
    class_addMethod(cls,
                    originalSelector,
                    method_getImplementation(swizzledMethod),
                    method_getTypeEncoding(swizzledMethod));

    if (didAddMethod) {
        class_replaceMethod(cls,
                            swizzledSelector,
                            method_getImplementation(originalMethod),
                            method_getTypeEncoding(originalMethod));
    } else {
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}

二、代码分析

2.1 class_getInstanceMethod()

获取某个类实例的方法, 如果该类实例没有此方法, 则返回NULL

Method swizzleMethod = class_getInstanceMethod([Person class], @selector(run));
if (swizzleMethod == NULL) {
    NSLog(@"NULL");
}

参数解释

class_getInstanceMethod(Class cls, SEL name)

cls: 获取方法的类

name: 方法的名称

2.2 class_addMethod()

参数解释

BOOL class_addMethod(Class cls, SEL name, IMP imp, const char *types)

cls: 获取方法的类

name: 添加的方法方法的名称

imp: 方法的实现,也就一个指向方法的指针

const char *types: 定义了返回值类型和参数类型的字符串(下面会提到)

返回值

YES: 增加方法成功

NO: 增加方法失败，例如 (如果目标类(Person) 实现了该方法,那么会返回 NO)

注意点

• class_addMethod 能够覆盖父类的实现的;
  
  如果目标类有实现了该方法，class_addMethod就会失败

class_addMethod will add an override of a superclass's implementation

• 处理警告问题
  
  参考: https://stackoverflow.com/questions/6224976/how-to-get-rid-of-the-undeclared-selector-warning

  void sayHello(id self, SEL _cmd, NSString *word)
  {
      NSLog(@"%@", word);
  }
  
  - (void)viewDidLoad {
      [super viewDidLoad];
  
  #pragma clang diagnostic push
  #pragma clang diagnostic ignored "-Wundeclared-selector"
      class_addMethod([Person class], @selector(resolveThisMethodDynamically:), (IMP)sayHello, "v@:@");
      Person *p = [[Person alloc] init];
      [p performSelector:@selector(resolveThisMethodDynamically:) withObject:@"hello"];
  #pragma clang diagnostic pop
  
  }
  
  

• 需要动态调用,因为通过运行时添加的方法,直接调用 编译不过的

  正确做法:
  [p performSelector:@selector(resolveThisMethodDynamically:) withObject:@"hello"];
  
  错误做法，编译不过
  [p resolveThisMethodDynamically:@"hello"];
  

• 参 imp 默认自带两个参数, id类型 以及 SEL 类型

  void sayHello(id self, SEL _cmd, ...)
  

2.3 参数 const char *types 解释

v 表示的是void 类型

i 表示整数类型

@ 表示一个对象

: 表示一个方法

v@: 表示的是返回值类型是void, 一个参数是对象(id self),另一个参数为方法 (SEL _cmd)

v@