iOS Swizzled 方法的替换、给原方法修改、添加代码

在iOS的开发当中、我们需要对原有的方法进行改进时,我们可以通过重写父类实现。但是,当我们需要在所有的子类都要修改父类方法,如果父类是我们自定义的API当然可以直接修改API,通常需要修改系统API活着三方库,就需要用到runtime时机制进行方法替换活着补救。
本文章以hook原系统的dealloc方法添加一句NSLog来打印当前页面的释放状态

采用category重写 load 方法,导入工程即可实现方法的替换

UIViewController+Swizzled.h

//
//  UIViewController+Swizzled.h
//  FtxBookViaPhone
//
//  Created by Jone on 2017/2/13.
//  Copyright © 2017年 FTXJOY. All rights reserved.
//

#import <UIKit/UIKit.h>

@interface UIViewController (Swizzled)

@end

UIViewController+Swizzled.m

//
//  UIViewController+Swizzled.m
//  FtxBookViaPhone
//
//  Created by Jone on 2017/2/13.
//  Copyright © 2017年 FTXJOY. All rights reserved.
//

#import "UIViewController+Swizzled.h"

@implementation UIViewController (Swizzled)

+(void)load {
    [super load];
    
    __weak typeof(self) weakSelf = self;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [weakSelf class];
        Method oldMethod = class_getInstanceMethod(weakSelf, NSSelectorFromString(@"dealloc"));
        Method newMethod = class_getInstanceMethod(weakSelf, @selector(swizzledDealloc));
        BOOL didAddMethod = class_addMethod(class, NSSelectorFromString(@"dealloc"), method_getImplementation(newMethod), method_getTypeEncoding(newMethod));
        if (didAddMethod) {
            class_replaceMethod(class, @selector(swizzledDealloc), method_getImplementation(oldMethod), method_getTypeEncoding(oldMethod));
        } else {
            method_exchangeImplementations(oldMethod, newMethod);
        }
    });
}

- (void)swizzledDealloc {
    NSLog(@"print:dealloc-%@",NSStringFromClass([self class]));
    
    [self swizzledDealloc];
}

@end

Demo

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 转至元数据结尾创建: 董潇伟,最新修改于: 十二月 23, 2016 转至元数据起始第一章:isa和Class一....
    40c0490e5268阅读 1,853评论 0 9
  • 本文由币乎社区(bihu.com)内容支持计划奖励 我的微信圈的朋友以前跟他们介绍比特币认为是骗局,现在看着不断...
    区块链台灯阅读 249评论 0 0
  • 红垂阅读 325评论 0 0
  • 与袁老师的初次见面在国粹堂开业当天,那时的匆匆一面感觉到雷厉风行的气场。 二次见面是袁老师和同事一起来体验,到店以...
    Wan9sha阅读 636评论 0 1