iOS runtime 动态添加属性,方法,方法交换案例

动态添加属性

我之前写过分类的文章，说过，分类不能添加成员变量，虽然我们这样去写了@interface TestClass (runtime) @property(nonatomic,copy)NSString *name; @end

但是只是生成了对应的set和get方法，并没有生成对应的成员变量，需要通过runtime的关联对象的方式给对象添加成员变量, objc_setAssociatedObject,并没有直接添加到宿主类上面，而是由一个 AssociationManager ， 管理在一个 hashMap 当中，他是一个全局的容器，为不同的类添加关联对象，也都放在这个map当中，

/*
     加入下面是为某个类添加的成员变量
     */
    
    {
        /*
         类1：
         @property(copy)NSString *name;
         */
        // 类1
        // 最外面的 key 为被关联的对象，value 是一个 ObjcAssociationMap ,ObjcAssociationMap 中，@selector 为被关联的值得 key，policy 为策略，是 retain还是copy 等
        "0x1242342432":{
            "@selector(name)":{
                "value":"sunchengxiu",
                "policy":"copy"
            }
        },
        // 类2
        "0x54354253223":{
            
            "@selector(age)":{
                "value":"25"
                "policy":"retain"
            }
        }
    }
    

类似于上面的json，最外层表示为哪个对象添加了关联对象，然后用关联对象的名字作为key，value是一个对象，里面存放着值和修饰符。

@implementation TestClass (runtime)
static const char *key = "name";
-(NSString *)school{
    return objc_getAssociatedObject(self, key);
    
}
-(void)setSchool:(NSString *)school{
    
    objc_setAssociatedObject(self, key, school, OBJC_ASSOCIATION_COPY);
}
@end

那么以上的关联对象正是用了 runtime 的动态添加属性的方式。

动态添加方法

上篇讲过，runtime 消息转发的第一步，动态方法解析，我们可以动态添加一个方法，通过 runtime

void  toPrint(id obj,SEL  _cmd){
    printf("to test resolveInstanceMethod");
}
+(BOOL)resolveInstanceMethod:(SEL)sel{
    if (sel == @selector(print)) {
        class_addMethod([self class], sel, (IMP)toPrint, "v@:");
        return YES;
    }
    return [super resolveInstanceMethod:sel];
}

这里就是动态添加了一个方法

方法交换

比如我们经常做 http 请求，那么我们想每次都打印这个url的值和判断是否为空应该怎么做呢？这时候就用到了，动态方法交换

@implementation NSURL (runtime)
+(void)load{
    Method m1 = class_getClassMethod([NSURL class], @selector(URLWithString:));
    Method m2 = class_getClassMethod([NSURL class], @selector(scx_URLWithString:));
    method_exchangeImplementations(m1, m2);
}
+(instancetype)scx_URLWithString:(NSString *)URLString{
    NSURL *url = [NSURL scx_URLWithString:URLString];
    
    if (!url) {
        NSLog(@"url 空了");
    } else {
        NSLog(@"%@",url);
    }
    return url;
}
@end

如上面的代码，这样我们就做到了，动态方法交换，就可以每次都判断这个url是否为空了。我们要在系统调用方法之前进行交换，所以要卸载 +load 里面。当我们调用系统的 URLWithString 其实就是调用了我们的 scx_URLWithString， 而我们在这个方法里面有调用scx_URLWithString了这个，其实就是调用系统的URLWithString，所以不会产生死循环。