Objective-C 动态生成中间类

有这样一个需求：

类Target，声明两个类Target的实例targetA，targetB；

当实例targetA调用方法oriMethod时，调用oriMethod;

当实例targetB调用方法oriMethod时，则实际调用的是hook之后的方法targetMethod

Target *targetA = [[Target alloc] init];

Target *targetB = [[Target alloc] init];

[targetA oriMethod];    //输出：原方法oriMethod

[targetB oriMethod];    //输出：新方法targetMethod

当面对上述需求时，Objective-C的动态性优势就显示出来啦；

具体实现原理类同KVO底层原理，给targetB类添加观察者，在观察方法中动态的生成一个中间类，然后在中间类中将方法oriMethod和targetMethod进行methodSwizzing;

具体实现步骤：

创建一个NSObject的Category，(NSObject+Swizzing.h),在类中添加两个方法

/// 添加观察方法

/// @param observer 目标类

/// @param oriMethod 原始方法

/// @param targetMethod 目标替换方法

- (void)ltAddObserver:(NSObject*)observer forOriMethod:(NSString*)oriMethod byNewMethod:(NSString*)targetMethod;

/// 移除观察方法

/// @param observer 目标类

/// @param oriMethod 原始方法

- (void)ltRemoveObserver:(NSObject*)observer forOriMethod:(NSString*)oriMethod;

判断原始方法(oriMethod)是否存在，如果不存在，那么直接返回，没必要进行后续操作

- (BOOL)judgeMethodIsExist:(NSString*)method{

    Class curClass =object_getClass(self);

    SEL oriSel =NSSelectorFromString(method);

    Method oriMethod =class_getInstanceMethod(curClass, oriSel);

    if(!oriMethod) {

        return false;

    }

    return true;

}

动态生成中间类

- (Class)dynamicCreateClass{

    NSString*oriClassName =NSStringFromClass([selfclass]);

    NSString*newClassName = [NSString stringWithFormat:@"%@%@",methodPrefix, oriClassName];

    Class newClass =NSClassFromString(newClassName);

    //申请类

    newClass =objc_allocateClassPair([self class], newClassName.UTF8String,0);

    //注册

    objc_registerClassPair(newClass);

    //添加class方法(class用来修改isa)

    SEL classSel =@selector(class);

    Method classMethod =class_getInstanceMethod([self class], classSel);

    const char *classType =method_getTypeEncoding(classMethod);

    class_addMethod(newClass, classSel, (IMP)lt_class, classType);

    //添加dealloc方法

    SEL deallocSel =NSSelectorFromString(@"dealloc");

    Method deallocMethod =class_getInstanceMethod([self class], deallocSel);

    const char*deallocType =method_getTypeEncoding(deallocMethod);

    class_addMethod(newClass, deallocSel, (IMP)lt_dealloc, deallocType);

    //添加方法

    return newClass;

}

lt_class方法实现，用来将iSA返回父类

Class lt_class(id self,SEL _cmd){

    return class_getSuperclass(object_getClass(self));

}

lt_dealloc方法，对象释放会自动调用

void lt_dealloc(id self,SEL _cmd){

    Class superClass = [self class];

    object_setClass(self, superClass);

}

交换的核心方法

+ (BOOL)lt_hookOriInstanceMethod:(Class)cls oriSel:(SEL)oriSel newInstanceMethod:(SEL)newSel{

    Method oriMethod =class_getInstanceMethod(cls, oriSel);

    Method newMethod =class_getInstanceMethod(cls, newSel);

    //如果不存在原始方法，那么添加一个空的方法，什么也不做，防止死循环

    if(!oriMethod) {

        class_addMethod(cls,

oriSel,method_getImplementation(newMethod),method_getTypeEncoding(newMethod));

           method_setImplementation(newMethod,imp_implementationWithBlock(^(id self,SEL _cmd){}));

    }

    //交换方法

    //步骤：

    //1、先添加方法，如果添加不成功，那么说明原始类中存在方法，直接交换即可

    //2、如果添加成功，说明原始类中不存在方法，那么直接replace

    BOOL addMethod =class_addMethod(cls, oriSel,method_getImplementation(newMethod),method_getTypeEncoding(newMethod));

    if(!addMethod) {

        method_exchangeImplementations(oriMethod, newMethod);

    }else{

        class_replaceMethod(cls, newSel,method_getImplementation(oriMethod),method_getTypeEncoding(oriMethod));

    }

    return true;

}

以上就是动态生成类以及类中添加方法的核心代码；

需要注意的是：

在ltAddObserver方法中，中间类创建出来后，一定要将修改iSA指向，否则调不到；

在ltRemoveObserver方法中，主要实现在移除观察之后，将iSA指针指回去；

还里利用associatedObject将被观察类和方法添加到全局表中(存储的是model信息)

具体代码实现 喜欢记得点星星哦