[个人博客搬运]Method Swizzling的简单应用场景

问题引入

前几天和同事一起看一个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

 

还可以做什么？

1. 记得去年实习期快结束的时候，恰逢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

 

2. 可以应用到埋点或者日志上，在控制器一些生命周期方法中需要添加埋点和日志，我们可以自定义这些生命周期方法然后动态替换它们.

@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

 

暂时这么多，如果文章中有问题请指出~~欢迎讨论