Runtime 运行时的方法交换

最近看了一个Runtime 的东西,”方法欺骗”(IMP 方法的交换).
使用好了非常厉害.

    NSString * str = @"http://baidu.com李";

    NSURL * url = [NSURL URLWithString:str];

    NSLog(@"1----%@",url);
打印:
2017-09-14 17:45:41.777 Runtime[4402:1147114] 1----(null)

这里出现nil 是因为 字符串包含中文,导致转成URL时不识别.
可以UTF-8 转码处理

NSString* string2 = [str stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];

但是每次出现一个URL 你都要这样处理,显然是很麻烦的.
就算你写在类别里面处理,使用的地方你要导入头文件,使用新加的类别方法.新项目还ok,但是你要改写别人的代码你会疯掉的.

---

下面要介绍使用runtime 来处理这个事情
还是要新建类别的 如:NSURL 的类别

 引用#import <objc/runtime.h>
  Runtime的头文件

#import "NSURL+url.h"
#import <objc/runtime.h>
@implementation NSURL (url)

这里是你新加的类别方法来做处理

+(instancetype)strURL:(NSString *)str {

   NSURL * url =  [NSURL URLWithString:str]; 
    if (url == nil) {
        NSLog(@"url 为空");
    }
    return url;
}

我们希望在外面还调用下面这个系统方法,但是希望内部走上面那个处理方法

[NSURL URLWithString:str];

ok,其实一个类被内存装载时,会调用一个方法: +(void)load
在这里做他们两个的方法交换处理

// 这个类被装载的时候调用 进入内存的时候调用
+(void)load {
    // 交换方法的IMP
    /*
     <#Method m1#>   要交换的方法1

     <#Method m2#>   要交换的方法2
      method_exchangeImplementations(<#Method m1#>, <#Method m2#>)

     得到一个类方法
     class_getClassMethod([self class], @selector(URLWithString:));

     得到一个实例方法
     class_getInstanceMethod(<#__unsafe_unretained Class cls#>, <#SEL name#>)
     */

    Method url = class_getClassMethod([self class], @selector(URLWithString:));

    Method strUrl = class_getClassMethod([self class], @selector(strURL:) );

    method_exchangeImplementations(url, strUrl);

}

这时候两个方法已经交换完了,但是你自己新加的类方法: +(instancetype)strURL:(NSString *)str
内部实现还是有 : NSURL * url = [NSURL URLWithString:str];
会造成死循环.你应该这样:

#import "NSURL+url.h"
#import <objc/runtime.h>
@implementation NSURL (url)

// 这个类被装载的时候调用 进入内存的时候调用
+(void)load {
    // 交换方法的IMP
    /*
     <#Method m1#>   要交换的方法1

     <#Method m2#>   要交换的方法2
      method_exchangeImplementations(<#Method m1#>, <#Method m2#>)

     得到一个类方法

     class_getClassMethod([self class], @selector(URLWithString:));

     得到一个实例方法
     class_getInstanceMethod(<#__unsafe_unretained Class cls#>, <#SEL name#>)
     */

    Method url = class_getClassMethod([self class], @selector(URLWithString:));

    Method strUrl = class_getClassMethod([self class], @selector(strURL:) );

    method_exchangeImplementations(url, strUrl);

}

+(instancetype)strURL:(NSString *)str {
   注意:这里一定要写好注释,不然别人看到...
   NSURL * url =  [NSURL strURL:str];  // 这里用到的IMP 方法交换 这个方法已经是 URLWithString
    if (url == nil) {
        NSLog(@"url 为空");
    }
    return url;
}

@end

这里已经ok了.外面依然是使用系统的方法,但内部是走的自己新加的方法.