利用runtime替换系统方法

对于个人来说，开发中用到runtime的情况很少，而且个人也不建议过多使用runtime甚至滥用runtime，如果项目中用了较多运行时的高级用法，接手项目的人估计会很头疼吧。无奈，runtime确实有着它的魅力，或多或少都想拿它装扮一下自己的门面。

场景: 判断url是否正确

可以有两种方式

• 给URL新建个category，公开一个检查url的自定义方法

#import <Foundation/Foundation.h>

@interface NSURL (Exchange)

+ (instancetype)CCC_URLWithString:(NSString *)str;

@end

#import "NSURL+Exchange.h"

@implementation NSURL (Exchange)

+ (instancetype)CCC_URLWithString:(NSString *)str
{
    NSURL *url = [NSURL URLWithString:str];
    if (!url) {
        NSLog(@"这个url有问题");
    }
    
    return url;
}

@end
复制代码

• 使用的地方，引入#import "NSURL+Exchange.h"

NSURL *url = [NSURL CCC_URLWithString:@"https://www.baidu.com/中文"];
复制代码

• 链接带中文时

NSURLRequest *request = [NSURLRequest requestWithURL:url];
NSLog(@"%@", request);

output:  
CCRuntime_Exchange[62692:3081179] 这个url有问题
CCRuntime_Exchange[62692:3081179] <NSURLRequest: 0x608000005820> { URL: (null) }
复制代码

这种方式是可以的，不过当你已经写了大量代码后，再想更换新自定义的方法会比较繁琐。

---

• 利用runtime替换系统方法

利用runtime 将系统的方法和我们自定义的方法的实现进行交换
    - method_exchangeImplementations交换两个方法的实现
    - class_getClassMethod 获取类方法
    - class_getInstanceMethod 获取实例方法
复制代码

• 在category中利用runtime替换系统的实现

#import "NSURL+Exchange.h"

#import <objc/runtime.h>

@implementation NSURL (Exchange)

//load早于main函数
+ (void)load
{
    //1.拿到两个Method
    Method URLWithStr = class_getClassMethod([NSURL class], @selector(URLWithString:));
    
    Method CCCURLWithStr = class_getClassMethod([NSURL class], @selector(CCC_URLWithString:));
    
    //2.进行方法的交换
    method_exchangeImplementations(URLWithStr, CCCURLWithStr);
    
}

+ (instancetype)CCC_URLWithString:(NSString *)str
{
//    NSURL *url = [NSURL URLWithString:str];
    
    NSURL *url = [NSURL CCC_URLWithString:str];//走到了系统的实现
    
    if (!url) {
        NSLog(@"这个url有问题");
    }
    
    return url;
}

@end
复制代码

• 使用时 仍直接使用系统的方法即可

NSURL *url = [NSURL URLWithString:@"https://www.baidu.com/中文"];
复制代码

• 结果:

NSURLRequest *request = [NSURLRequest requestWithURL:url];
    
NSLog(@"%@", request);

output:
CCRuntime_Exchange[62872:3088854] 这个url有问题
CCRuntime_Exchange[62872:3088854] <NSURLRequest: 0x6000000115a0> { URL: (null) }

复制代码

ps: 如果使用了runtime的高级功能，一定要写好注释，接手的小伙伴会感激你的

代码