iOS 自定义一个简单的内存检测工具

原理: 给当前对象延迟发送消息 例如:当viewcontroller调用pop方法后它应该会被释放,此时在两秒后给它发送一个消息如果它还能响应则表示它未被正常释放.

1. 给UIViewController添加一个分类

在 + (void)load 通过methodswizzle 交换 viewWillAppear:(BOOL)animated 方法,并在新的方法里通过属性绑定添加一个标识默认为NO(如果为NO就不给它延迟发送消息,如果为YES则延迟发送消息).

Method appearMethod = class_getInstanceMethod([self class], @selector(viewWillAppear:));
    Method md_appearMethod = class_getInstanceMethod([self class], @selector(md_viewWillAppear:));
method_exchangeImplementations(appearMethod, md_appearMethod);

- (void)md_viewWillAppear:(BOOL)animated{
    [self md_viewWillAppear:animated];
    objc_setAssociatedObject(self, "md_mark", @(NO), 0);
}
2. 给UINavigationController添加一个分类

在 + (void)load 通过methodswizzle 交换 popViewControllerAnimated 方法,并在新的方法里, 将第一步添加的属性标识赋值为YES.

+ (void)load{
    Method popMethod = class_getInstanceMethod([self class], @selector(popViewControllerAnimated:));
    Method md_popMethod = class_getInstanceMethod([self class], @selector(md_popViewControllerAnimated:));
    method_exchangeImplementations(popMethod, md_popMethod);
}

- (UIViewController *)md_popViewControllerAnimated:(BOOL)animated{
    UIViewController *poppedController = [self md_popViewControllerAnimated:animated];
    objc_setAssociatedObject(poppedController, "md_mark", @(YES), 0);
    return poppedController;
}
3. 延迟发送消息

在第一步创建的分类里再通过methodswizzle 交换viewDidDisappear方法(或者viewWillDisappear也行)在新的方法里通过判断属性标识,再延迟发送消息(如果strongSelf为nil表示正常释放,否则表示存在内存泄漏).

+ (void)load{
   Method disappearMethod = class_getInstanceMethod([self class], @selector(viewDidDisappear:));
    Method md_disappearMethod = class_getInstanceMethod([self class], @selector(md_viewDidDisappear:));
 method_exchangeImplementations(disappearMethod, md_disappearMethod);
}
- (void)md_viewDidDisappear:(BOOL)animated{
    [self md_viewDidDisappear:animated];
    BOOL md_mark = [objc_getAssociatedObject(self, "md_mark") boolValue];
    if (md_mark) {
        __weak typeof(self)weakSelf = self;
        dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(2 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
            __strong typeof(self)strongSelf = weakSelf;
            if (strongSelf) {
                NSLog(@"👉👉👉👉👉:%@",strongSelf);
            }else{
               // NSLog(@"ok!!!!!");
            }
        });
    }
}
4. 这时候就可以测试了

在ViewController2里故意写一个block循环引用

 self.block = ^{
        self.index = 5;
    };

在pop回去2秒后,控制台会打印未释放的控制器:

2019-07-09 15:57:46.930865+0800 NeiCun[4928:154041] 👉👉👉👉👉:<ViewController2: 0x7ffb76708150>
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Swift1> Swift和OC的区别1.1> Swift没有地址/指针的概念1.2> 泛型1.3> 类型严谨 对...
    cosWriter阅读 11,148评论 1 32
  • 面向对象的三大特性:封装、继承、多态 OC内存管理 _strong 引用计数器来控制对象的生命周期。 _weak...
    运气不够技术凑阅读 1,137评论 0 10
  • 国家电网公司企业标准(Q/GDW)- 面向对象的用电信息数据交换协议 - 报批稿:20170802 前言: 排版 ...
    庭说阅读 11,186评论 6 13
  • 尊敬的李老师,智慧的班主任,亲爱的跃友们: 大家好!我是来自潍坊先农的何军永 今天是2019年6月26日,是我的日...
    93803f718ea8阅读 126评论 0 0
  • 上一章 目录 下一章 当蛛儿前进了大约半个时辰的时候,它突然停了下来,我正奇怪,突然发现身下的草原上,有着...
    铁面君阅读 200评论 0 2