使用运行时库覆盖方法

Question

我需要覆盖- (void)viewWillAppear:(BOOL)animated，以便在此方法中添加NSLog(@"blabla") 的所有ViewController。 IE。在每次调用 viewWillAppear 调用实现 viewWillAppear + 我的 NSLog 消息的早期实现之后。是否可以？如果是的话，请给我一个建议。

目前我已经尝试过这段代码

@implementation RuntimeTest

IMP previusImp;
IMP newIMP;
- (void)ovverrideViewWillAppearInViewController:(Class)vcClass {
    newIMP = class_getMethodImplementation([self class], @selector(viewWillAppear:));
    Method viewWillAppearMethod = class_getInstanceMethod(vcClass, @selector(viewWillAppear:));
    previusImp = method_setImplementation(viewWillAppearMethod, newIMP);
}

- (void)viewWillAppear:(BOOL)animated {
    previusImp(self, @selector(viewWillAppear:), animated);
    NSLog(@"log2");
}

@end

那我有

@implementation IRViewController2

- (void)viewWillAppear:(BOOL)animated {
    [super viewWillAppear:animated];
    NSLog(@"log");
}

@end

我的自定义 viewWillAppear 首先调用，然后从 IRViewController2 调用 viewWillAppear。在此之后，我的应用程序因 EXC_BAD_ACCESS 而崩溃。怎么了？

Answer 1

如果你不能使用子类，因为你想全局使用它，但又不想丢失旧版本的方法（所以类别覆盖是不可能的），那么你需要swizzle away旧方法，放入你的新方法，然后调用旧方法...我过去曾使用名为 JRSwizzle 的东西来执行此操作，它可以轻松进行调配，而无需编写一堆运行时代码。

if (![self jr_swizzleMethod:@selector(originalInit) withMethod:@selector(init) error:&err])
{
    NSLog(@"unable to jr_swizzle methods, error: %@",err);
    exit(EXIT_FAILURE);
}
if (![self jr_swizzleMethod:@selector(init) withMethod:@selector(myInit) error:&err])
{
    NSLog(@"unable to jr_swizzle methods, error: %@",err);
    exit(EXIT_FAILURE);
}

那么您的-myInit 方法可以调用-originalInit 或您正在使用的任何方法。