#iOS底层探索 -- Method-Swizzling 的应用与坑点

在我们开始接触到runtime之后,我们经常能见到Method-Swizzling这个被称作 黑魔法 的东西,那么到底什么是Method-Swizzling 怎么使用、使用过程中又有哪些坑点,我们今天来探究一下

什么是Method-Swizzling

字面上: 方法调配 、方法交换
实现上: 我们常说的方法、方法名sel 通过其指向的IMP指针(方法的实现),通过这样的对应关系在使用时去调用对应的方法。
这里通过 objc/runtime.h中提供的api:method_exchangeImplementations(Method m1, Method m2)来交换IMP,通过这样的方法重新绑定selIMP对应关系。

Method-Swizzling原理.jpg

Method-Swizzling有什么作用

那么上面这样的方法交换在我们开发中到底有什么作用?

举个栗子🌰
我们在开发过程中一直都会有一个无法避开的需求--埋点比如记录用户在操作APP期间到底访问了哪些页面。

  1. 最基础的,我们在每个VC的ViewDidLoad中去预留方法
  2. 高一点的,我们实现一个VC的基类,在基类中我们对ViewDidLoad重写
  3. 再或者,我们可以通过category去实现
    但这些我们都需要写大量代码,如果2、3方法是在项目后期去做时,也需要去调整很多类信息。

甚至,我们有可能是需要再SDK中,在不接触页面代码的情况下去实现这个埋点的需求,我们就可以通过Method-Swizzling 去直接hook ViewDidLoad方法,直接实现埋点。
这也就是AOP(Aspect Oriented Programming) 面向切面编程 思想

通过预编译方式和运行期间动态代理实现程序功能的统一维护的一种技术。AOPOOP的延续,是软件开发中的一个热点,也是Spring框架中的一个重要内容,是函数式编程的一种衍生范型。利用AOP可以对业务逻辑的各个部分进行隔离,从而使得业务逻辑各部分之间的耦合度降低,提高程序的可重用性,同时提高了开发的效率。(摘自百度百科)

除了上面说的埋点,还有一下crash收集我们也可以通过这样的方法实现。

Method-Swizzling具体使用 及 坑点

接下来,我们实战看一下Method-Swizzling的具体应用

/// 实例方法交换
/// @param cls 方法交换的类
/// @param oriSEL 需要被交换的方法编号
/// @param swizzledSEL 用来交换的方法编号
+ (void)lg_methodSwizzlingWithClass:(Class)cls oriSEL:(SEL)oriSEL swizzledSEL:(SEL)swizzledSEL{
    
    if (!cls) NSLog(@"传入的交换类不能为空");

    Method oriMethod = class_getInstanceMethod(cls, oriSEL);
    Method swiMethod = class_getInstanceMethod(cls, swizzledSEL);
    method_exchangeImplementations(oriMethod, swiMethod);
}

这是最基础的Method-Swizzling应用,其中会有一些问题,比如

重复调用问题

因为+(void)load方法调用的最早,所以一般我们放在其中去做方法交换。而+(void)load并不能确定只调用一次,如果发生多次调用,那么方法交换也会发生多次,IMP就会反复的交换。

解决方案

通过单例的思路,我们通过 dispatch_once(&onceToken, ^{})去做数据保护,避免反复横跳的发生

或者我们可以通过手动触发,只调用一次

子类交换了继承于父类方法

- (void)viewDidLoad {
    [super viewDidLoad];

    ///LGStudent 继承于 LGPerson
    LGStudent *s = [[LGStudent alloc] init];
    [s personInstanceMethod];

    LGPerson *p = [[LGPerson alloc] init];
    [p personInstanceMethod];
}
#import "LGPerson.h"

@implementation LGPerson
- (void)personInstanceMethod{
    NSLog(@"person对象方法:%s",__func__);
}
+ (void)personClassMethod{
    NSLog(@"person类方法:%s",__func__);
}
@end
@implementation LGStudent (LG)

+ (void)load{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        [LGRuntimeTool lg_methodSwizzlingWithClass:self oriSEL:@selector(personInstanceMethod) swizzledSEL:@selector(lg_studentInstanceMethod)];
    });
}
- (void)lg_studentInstanceMethod{
    NSLog(@"LGStudent分类添加的lg对象方法:%s",__func__);
}

运行结果


Method-Swizzling坑点2.jpg

我们在之前的探索中就知道,当子类没有实现方法时,方法会遍历到父类的方法列表中返回IMP,而Method-Swizzling是直接修改的IMP,所以被交换的其实就是父类的方法

那么问题就来了,如果交换后的方法父类本身不存在,那就找不到对应方法,就会出现崩溃。

解决方案

+ (void)lg_betterMethodSwizzlingWithClass:(Class)cls oriSEL:(SEL)oriSEL swizzledSEL:(SEL)swizzledSEL{
    
    if (!cls) NSLog(@"传入的交换类不能为空");
    // oriSEL       personInstanceMethod
    // swizzledSEL  lg_studentInstanceMethod
    Method oriMethod = class_getInstanceMethod(cls, oriSEL);
    Method swiMethod = class_getInstanceMethod(cls, swizzledSEL);
    //加一层保护措施,如果添加成功,则表示该方法不存在于本类,而是存在于父类中,不能交换父类的方法,否则父类的对象调用该方法会crash;添加失败则表示本类存在该方法
    BOOL success = class_addMethod(cls, oriSEL, method_getImplementation(swiMethod), method_getTypeEncoding(oriMethod));
    if (success) {// 自己没有 - 交换 - 没有父类进行处理 (重写一个)
        //再将原有的实现替换到swizzledMethod方法上,从而实现方法的交换,并且未影响到父类方法的实现
        class_replaceMethod(cls, swizzledSEL, method_getImplementation(oriMethod), method_getTypeEncoding(oriMethod));
    }else{ // 自己有
        method_exchangeImplementations(oriMethod, swiMethod);
    }
}

在交换前判断下是否是自己的方法,如果不是,那就改为替换,不直接交换,避免影响父类的方法实现。

探究 在交换的方法中再次调用原方法,是否会发生递归

// 是否递归
- (void)lg_studentInstanceMethod{
    [self lg_studentInstanceMethod]; 
    NSLog(@"LGStudent分类添加的lg对象方法:%s",__func__);
}

答案是不会。


递归探寻.jpg

此时lg_studentInstanceMethod 内部实现是通过personInstanceMethod 方法名去调用的,而中间lg_studentInstanceMethod的方法,则指向了personInstanceMethodIMP,并不会行成一个递归循环。
但如果改为

// 是否递归
- (void)lg_studentInstanceMethod{
    [self personInstanceMethod]; 
    NSLog(@"LGStudent分类添加的lg对象方法:%s",__func__);
}

还是会一直在递归循环中,所以在这里要清楚理解SELIMP的关系和区别

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,921评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,635评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,393评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,836评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,833评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,685评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,043评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,694评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,671评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,670评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,779评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,424评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,027评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,984评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,214评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,108评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,517评论 2 343

推荐阅读更多精彩内容