Runtime替换实现方法

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);