问题引入
前几天和同事一起看一个iOS客户端工程的bug，找出来是向字典插入空值的原因，这类问题如果要预防，需要在每次插入之前做一次空值判断，虽然严谨但是稍微有些麻烦。于是周末我调研了其他的方法，就是Method Swizzling，用到Objective-C的底层Runtime机制。
原理
简单来说，每个类里都有一个 Dispatch Table ，将方法的名字（SEL）跟方法的实现（IMP，指向 C 函数的指针）一一对应。Swizzle 一个方法其实就是在程序运行时在 Dispatch Table 里做点改动，让这个方法的名字（SEL）对应到另个 IMP ，做一次交叉互换，这就可以让我们用自定义的方法来替代系统实现。
@implementation NSObject (Feifu)
(void)swizzleClassMethodOfClass:(Class)clazz firstMethod:(SEL)firstMethod secondMethod:(SEL)secondMethod
{
Method m1 = class_getClassMethod(clazz, firstMethod);
Method m2 = class_getClassMethod(clazz, secondMethod);
method_exchangeImplementations(m1, m2);
}(void)swizzleInstanceMethodOfClass:(Class)clazz firstMethod:(SEL)firstMethod secondMethod:(SEL)secondMethod
{
Method m1 = class_getInstanceMethod(clazz, firstMethod);
Method m2 = class_getInstanceMethod(clazz, secondMethod);
method_exchangeImplementations(m1, m2);
}
@end
上面是类方法和实例方法的Method Swizzling实现。为了让所有的类都可以方便调用，将上面的代码做成NSObject的分类。
实现
接着我们再新建一个NSMutableDictionary的分类
@implementation NSMutableDictionary (Feifu)
- (void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
[self swizzleInstanceMethodOfClass:NSClassFromString(@"__NSDictionaryM") firstMethod:@selector(setObject:forKey:) secondMethod:@selector(ff_setObject:forKey:)];
});
}
- (void)ff_setObject:(id)anObject forKey:(id<NSCopying>)aKey
{
if (anObject == nil) {
NSLog(@"ff_setObject: for key: object can't be nil!!!");
} else {
[self ff_setObject:anObject forKey:aKey];
}
}
@end
上面的+（void）load方法，只要这个类的符号被编译到最后的可执行文件中，load方法就会被调用。这省去我们#import这个分类的麻烦。dispatch_once可以让MethodSwizzling操作只执行一次，防止多次修改
上面的ff_setObject:forKey:的实现和setObject:forKey:的实现互相替换，注意ff_setObject:forKey:方法里面的ff_setObject:forKey:并不是递归调用，而是此时的ff_setObject:forKey:具体实现是setObject:forKey:（有点绕）
相应的还有NSMutableArray的安全插入实现：
@implementation NSMutableArray (Feifu)
- (void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
[self swizzleInstanceMethodOfClass:NSClassFromString(@"__NSArrayM") firstMethod:@selector(addObject:) secondMethod:@selector(ff_addObject:)];
[self swizzleInstanceMethodOfClass:NSClassFromString(@"__NSArrayM") firstMethod:@selector(insertObject:atIndex:) secondMethod:@selector(ff_insertObject:atIndex:)];
});
}
(void)ff_addObject:(id)anObject
{
if (anObject == nil) {
NSLog(@"ff_addObject:object can't be nil!!!");
} else {
[self ff_addObject:anObject];
}
}(void)ff_insertObject:(id)anObject atIndex:(NSUInteger)index
{
// NSLog(@"ff_insertObject");
if (anObject == nil) {
NSLog(@"ff_insertObject: object can't be nil!!!");
} else {
[self ff_insertObject:anObject atIndex:index];
}
}
@end
还可以做什么？
- 记得去年实习期快结束的时候，恰逢iPhone6和iPhone6s的适配工作，解决方法是根据手机的屏幕尺寸不同来设置frame对应的宽度乘上一个系数，但是这样工程量会非常大
其实可以给UIView做一个分类，然后用Method Swizzling做一次替换，替换掉系统的setFrame:方法，将拿到的frame的宽度等数值直接乘上系数.不过这个做法有个问题，就是粒度太大，把一些不需要适配宽度也会影响。系统自带的控件要显示，也会调用setFrame方法，如statusBar啊，这个怎么过滤掉是接下来要思考的问题。下面以UIButton做例子
@implementation UIView (Feifu)
- (void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
[self swizzleInstanceMethodOfClass:NSClassFromString(@"UIView") firstMethod:@selector( setFrame:) secondMethod:@selector(ff_setFrame:)];
});
}
-
(void)ff_setFrame:(CGRect)frame
{
if ([self isKindOfClass:[UIButton class]]) {
CGFloat scale = 1.0;
if ([[UIScreen mainScreen]bounds].size.width == 375.f) {
scale = 1.171875;
} else if ([[UIScreen mainScreen]bounds].size.width == 414.f) {
scale = 1.29375;
} else {
scale = 1;
}frame.size.width *= scale;}
[self ff_setFrame:frame];
}
@end
- 可以应用到埋点或者日志上，在控制器一些生命周期方法中需要添加埋点和日志，我们可以自定义这些生命周期方法然后动态替换它们.
@implementation UIViewController (Feifu)
- (void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
[self swizzleInstanceMethodOfClass:NSClassFromString(@"UIViewController") firstMethod:@selector(viewDidLoad) secondMethod:@selector(ff_viewDidLoad)];
});
}
- (void)ff_viewDidLoad
{
NSLog(@"Log here");
[self ff_viewDidLoad];
}
@end
暂时这么多，如果文章中有问题请指出~~欢迎讨论
