您的位置：首页 > 其它

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

2016-07-15 19:28 253 查看



问题引入

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


暂时这么多，如果文章中有问题请指出~~欢迎讨论
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签： 
相关文章推荐
新的分享
章节导航