Method Swizzling
//
// UIViewController+Tracking.m
// tesssss
//
// Created by iOS on 23/11/17.
// Copyright © 2017年 iOS. All rights reserved.
//
#import "UIViewController+Tracking.h"
#import <objc/runtime.h>
@implementation UIViewController (Tracking)
// 放在load方法中，当类被加载之后就执行以下方法。
+ (void)load {
// 防止手动调用 class 方法，保证替换方法只执行一次
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
Class class = [self class];
SEL originalSelector = @selector(viewWillAppear:);
SEL swizzledSelector = @selector(ICE_viewWillAppear:);
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
/*
class_addMethod(Class _Nullable cls, SEL _Nonnull name, IMP _Nonnull imp,
const char * _Nullable types)
* Class cls cls 参数表示需要添加新方法的类。
* SEL name name 参数表示 selector 的方法名称，可以根据喜好自己进行命名。
* IMP imp imp 即 implementation ，表示由编译器生成的、指向实现方法的指针。也就是说，这个指针指向的方法就是我们要添加的方法。
* const char *types 最后一个参数 *types 表示我们要添加的方法的返回值和参数。
如果发现方法已经存在，会失败返回，也可以用来做检查是否已经添加过方法了,我们这里是为了避免源方法没有实现的情况;
如果方法没有存在,我们则先尝试添加被替换的方法的实现
*/
BOOL didAddMethod = class_addMethod(
class,
originalSelector,
method_getImplementation(swizzledMethod),
method_getTypeEncoding(swizzledMethod)
);
/*
判断class_addMethod是否已经添加成功了
YES 则说明被替换方法不存在.也就是被替换的方法没有被实现,我们需要先把这个方法实现,然后再执行我们想要的效果,用我们自定义的方法去替换被替换的方法. 这里使用到的是class_replaceMethod这个方法. class_replaceMethod本身会尝试调用class_addMethod和method_setImplementation，所以直接调用class_replaceMethod就可以了)
NO 则说明被替换方法已经存在.直接将两个方法的实现交换即
*/
if (didAddMethod) {
class_replaceMethod(
class,
swizzledSelector,
method_getImplementation(originalMethod),
method_getTypeEncoding(originalMethod)
);
}
else {
method_exchangeImplementations(originalMethod, swizzledMethod);
}
});
}
- (void)ICE_viewWillAppear:(BOOL)animated {
[self ICE_viewWillAppear:animated];
NSLog(@"ICEviewWillAppear:%@",self);
}
@end
//如果替换的是类方法
Class classB = Object_getClass ((id) self);
SEL originalSelector = @selector (想要替换的类方法);
SEL swizzledSelector = @selector (替换的新方法名) ;
Method originalMethod = class_getClassMethod (classB, originalSelector);
Method swizzlingMethod = class_getClassMethod (classB, swizzledSelector);
