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等方法)
当然还有很多其他的用途,这些只是较为常见的






posted @ 2016-06-07 09:50  LiLM  阅读(1587)  评论(0编辑  收藏  举报