Runtime之方法交换

在没有一个类的实现源码的情况下，想改变其中一个方法的实现，除了继承它重写、和借助类别重名方法暴力抢先之外，还有就是方法交换

方法交换的原理：在OC中调用一个方法其实是向一个对象发送消息，查找消息的唯一依据是selector的名字。利用OC的动态特性，可以实现在运行时偷换selector方法的实现，达到和方法挂钩的目的。

每一个类都有一个方法列表，存放在selector的名字和方法实现的映射关系，imp有点像函数指针，指向具体的方法实现。

可以利用method_exchanggeimplementations来交换两个方法的imp

可以利用class_replaceMethod来替换方法的imp

可以利用method_setimplementation来直接设置某个方法的imp

归根结底方法交换就是偷换了selector的imp

Method Swizzling 实践

举个例子好了，就替换NSArray的objectAtIndex:方法吧，只需两个步骤。

第一步：自定义一个objectAtIndex:方法，方法名不能和系统的一样

#import <objc/runtime.h>

#import "NSArray+Swizzle.h"

@implementation NSArray (Swizzle)

-(id)ll_ObjectAtIndex:(NSUInteger)index{

    if (index < [self count]) {

        return [self ll_ObjectAtIndex:index];

    }else{

        return nil;

    }

}

乍一看，这不递归了么？别忘记这是我们准备调换IMP的selector，[self ll_ObjectAtIndex:index] 将会执行真的 [self objectAtIndex:index] 。

第二步：调换IMP

+(void)initialize{

    static dispatch_once_t once;

    dispatch_once(&once, ^{

        Class class = NSClassFromString(@"__NSArrayI");

        [self swizzleMethods:class originalSelector:@selector(objectAtIndex:) swizzledSelector:@selector(ll_objectAtIndex:)];

    });

}

//方法交换的具体实现

+ (void)swizzleMethods:(Class)class originalSelector:(SEL)origSel swizzledSelector:(SEL)swizSel

{

    Method origMethod = class_getInstanceMethod(class, origSel);

    Method swizMethod = class_getInstanceMethod(class, swizSel);

    //class_addMethod will fail if original method already exists

    BOOL didAddMethod = class_addMethod(class, origSel, method_getImplementation(swizMethod), method_getTypeEncoding(swizMethod));

    if (didAddMethod) {

        class_replaceMethod(class, swizSel, method_getImplementation(origMethod), method_getTypeEncoding(origMethod));

    } else {

        //origMethod and swizMethod already exist

        method_exchangeImplementations(origMethod, swizMethod);

    }

}

@end

定义完毕新方法后,需要弄清楚什么时候实现与系统的方法交互?

既然是给系统的方法添加额外的功能,换句话说,我们以后在开发中都是使用自己定义的方法,取代系统的方法,所以,当程序一启动,就要求能使用自己定义的功能方法.说到这里:我们必须要弄明白一下两个方法 :

+(void)initialize(当类第一次被调用的时候就会调用该方法,整个程序运行中只会调用一次)

+ (void)load(当程序启动的时候就会调用该方法,换句话说,只要程序一启动就会调用load方法,整个程序运行中只会调用一次)
方法交换可以做什么呢
1、防止数组越界
2、统计页面的访问次数（通过交换viewdidload 或者 viewwillappear等方法）
当然还有很多其他的用途，这些只是较为常见的

巴特西

Runtime之方法交换

最新文章

热门文章