method_exchangelmplementations

2017-07-19  本文已影响26人  无题007

图解

我们说过,方法的调用过程,归根结底是函数指针的调用,在oc的动态语言特性下,我们可以改变Method结构体中IMP函数实现指针的指向:(我们以实例方法为例)
方法交换前:

方法交换后:

创建一个Person类,实现+(void)eat;-(void)eat;方法。在VC中实现+(void)eat_VC;方法。在VC中的代码如下:

+(void)load{
    //交换类方法
    Method originalMethod_class = class_getClassMethod(self, NSSelectorFromString(@"eat_VC"));
    Method swizzledMethod_class = class_getClassMethod(NSClassFromString(@"Person"), NSSelectorFromString(@"eat"));
    method_exchangeImplementations(originalMethod_class, swizzledMethod_class);
    //交换实例方法
    Method originalMethod_id = class_getInstanceMethod(self, NSSelectorFromString(@"viewDidLoad"));
    Method swizzledMethod_id = class_getInstanceMethod(NSClassFromString(@"Person"), NSSelectorFromString(@"eat"));
    method_exchangeImplementations(originalMethod_id, swizzledMethod_id);
}

代码分享

NSObject (Runtime)
.h
#pragma mark - method_exchangeImplementations
/**
 *替换实例方法(在+load中调用)
 */
+ (void)swizzleSelector:(SEL)originalSelector withSelector:(SEL)swizzledSelector;
/**
 *替换类方法(在+load中调用)
 */
+ (void)swizzleClassSelector:(SEL)originalSelector withClassSelector:(SEL)swizzledSelector;
----------------------------------------------------------------
.m
#pragma mark - 交换方法
+ (void)swizzleSelector:(SEL)originalSelector withSelector:(SEL)swizzledSelector {
    Class class = [self class];

    Method originalMethod = class_getInstanceMethod(class, originalSelector);
    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);

    BOOL didAddMethodInit=class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));

    if (didAddMethodInit) {
        class_addMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
    }else{
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}

+ (void)swizzleClassSelector:(SEL)originalSelector withClassSelector:(SEL)swizzledSelector {
    Class class = [self class];

    Method originalMethod = class_getClassMethod(class, originalSelector);
    Method swizzledMethod = class_getClassMethod(class, swizzledSelector);
    if ((int)originalMethod != 0 && (int)swizzledMethod != 0) {
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}
###使用场景举🌰

- 当我们一个功能模块功能点很多时,我们可以按照功能点创建分类,需要调用相同的方法(比如:`-(void)viewDidLoad;`)我们可以在每个扩展类中都写一个对应的然后进行交换。

- 对于需求修改的扩展,比如你们公司有个超级牛逼的应用,100个`ViewController`,展示100张`Image`,迫不得已你使用了100次`+(nullable UIimage *)imageName:(NSString *)name`方法。然后你们产品要做一个大的迭代,在展示图片的同时,打印图片的大小。你需要去掉的100个地方?要是我,我会在原有的基础上添加打印大小的代码。

- 对于通用模块基于系统方法的扩展,比如我们要统计页面停留时长,我们需要在`- (void)viewDidApp:(BOOL)anmated;`以及`- (void)viewDidDisappear:(BOOL)animated;`中进行埋点。那么我们就可以先找到一个特定字段标志当前页面(比如用扩展类加一个特定埋点属性),然后写一个方法与系统方法进行交换,调回原方法,后加埋点实现。
上一篇下一篇

猜你喜欢

热点阅读