iOS性能优化实战篇-自动检测VC内存泄漏
一、内存泄漏的原因
1、iOS中引入ARC(Automatic Reference Counting)机制后还是会内存泄漏。主要原因是程序中存在循环强引用，导致对象自动释放时出现相互等待不能及时释放内存（不进行及时处理一直会等到整个程序结束时才释放）。解决办法：打破循环引用链。
2、ARC只会对Foundation\UIKit(前缀NS__、UI__)框架下的对象进行自动管理。在使用Core Foundation(前缀CF__)框架下的对象时还是需要去手动调用CFRelease(CF__、CG__)管理。
二、内存泄漏的几种情况
1、Block:
在 ARC 下，当 block 获取到外部变量时，由于编译器无法预测获取到的变量何时会被突然释放，为了保证程序能够正确运行，让 block 持有获取到的变量，向系统声明：我要用它，你们千万别把它回收了！然而，也正因 block 持有了变量，容易导致变量和 block 的循环引用，造成内存泄露。
类似循环引用系统是不会提示哦
2、NSTimer:
NSTimer在释放前，一定要调用[timer invalidate]，不调用的后果就是NSTimer无法释放其target，如果target正好是self，则会导致引用循环。
3、Delegate：
我们在使用代理设计模式的时候，一定要注意将 delegate 变量声明为 weak 类型，像这样如使用strong或别的类型修饰的话将会导致循环引用。
三、如何快速检测VC中的内存泄漏 ---(重点！！！)
一般我们都是通过重写dealloc方法来检测当前控制器内是否出现内存泄漏，如果每一个控制器中都去写一遍dealloc是非常较麻烦的。所以我准备给UIViewController、UINavigationController添加分类的方式来实现，先上代码（可先看后面的总结）：
先给UIViewController加分类：
@interface UIViewController (LeaksTest)
@end
#import "UIViewController+LeaksTest.h"
#import <objc/message.h>
const char * ZJPOPFLAG;
@implementation UIViewController (LeaksTest)
+ (void)load{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
//替换类方法
[self zj_swizzleOriginSEL:@selector(viewWillAppear:) currentSEL:@selector(zj_viewWillAppear:)];
[self zj_swizzleOriginSEL:@selector(viewDidDisappear:) currentSEL:@selector(zj_viewDidDisappear:)];
});
}
/**方法交换*/
+ (void)zj_swizzleOriginSEL:(SEL)originSEL currentSEL:(SEL )currentSEL{
Method originMethod = class_getInstanceMethod([self class], originSEL);
Method currentMethod = class_getInstanceMethod([self class], currentSEL);
method_exchangeImplementations(originMethod, currentMethod);
}
//自定义viewWillAppear
- (void)zj_viewWillAppear:(BOOL)animated{
[self zj_viewWillAppear:animated];
objc_setAssociatedObject(self, &ZJPOPFLAG, @(NO), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
//自定义viewDidDisappear
- (void)zj_viewDidDisappear:(BOOL)animated{
[self zj_viewDidDisappear:animated];
if ([objc_getAssociatedObject(self, &ZJPOPFLAG) boolValue]) {
[self willDealloc];
}
}
/**即将都用dealloc*/
- (void)willDealloc{
__weak typeof(self) weakSelf = self;
//延时4s 留足释放内存时间
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)( 4 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
__strong typeof(self) strongSelf = weakSelf;
/* !!!重要点 !!!
objc_msgSend(strongSelf,@selector(isNotDealloc));
如果控制器销毁就不会执行，如果执行说明控制器没有被释放(可能出现内存泄漏)
*/
[strongSelf isNotDealloc];
});
}
/**控制器未释放，内存泄漏*/
- (void)isNotDealloc{
NSLog(@"%@ is not dealloc",self);
}
@end
再给UINavigationController加分类：
@interface UINavigationController (LeaksTet)
@end
#import "UINavigationController+LeaksTet.h"
#import <objc/message.h>
@implementation UINavigationController (LeaksTet)
+ (void)load{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
Method originMethod = class_getInstanceMethod([self class], @selector(popViewControllerAnimated:));
Method currentMethod = class_getInstanceMethod([self class], @selector(zj_popViewControllerAnimated:));
//替换POP方法
method_exchangeImplementations(originMethod, currentMethod);
});
}
/**自定义popViewControllerAnimated*/
- (UIViewController *)zj_popViewControllerAnimated:(BOOL)animated{
UIViewController * popVC = [self zj_popViewControllerAnimated:animated];
extern const char * ZJPOPFLAG;
//给ZJPOPFLAG标识赋值
objc_setAssociatedObject(popVC, &ZJPOPFLAG, @(YES), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
return popVC;
}
@end
总结：
-
1、自定义控制器三个系统方法 ：zj_viewWillAppear、zj_viewDidDisappear、zj_popViewControllerAnimated;
-
2、通过Runtime在+load方法中替换系统中的三个方法：viewWillAppear、viewDidDisappear、popViewControllerAnimated
替换方法：method_exchangeImplementations(Method _Nonnull m1, Method _Nonnull m2)
-
3、在3个自定义的方法中给当前VC添加一个关联标识(==添加属性)。
关联方法：
设值：objc_setAssociatedObject(id _Nonnull object, const void * _Nonnull key, id _Nullable value, objc_AssociationPolicy policy)
取值：objc_getAssociatedObject(id _Nonnull object, const void * _Nonnull key)
然后通过该标识值来判断当前控制器是否要退出(销毁)，在即将退出时调用-willDealloc来执行延时任务。
(1)、若控制器销毁 [strongSelf isNotDealloc]将不会执行打印任务;
(2)、若控制器未销毁 [strongSelf isNotDealloc]将打印出：
<ViewControllerB: 0x7fbc96415bf0> is not dealloc
到此一个简单的自动检测控制器内存泄漏的工具就完成了，使用时只需要添加分类头文件即可，具体代码我就不写了。（见谅！）
