objc运行时，方法交换 模板示例

2019独角兽企业重金招聘Python工程师标准>>>

模板代码：

    

/**
 1. 将 AFN 中的两个静态函数复制过来
 2. 添加运行时的头文件
 3. 在 +load 方法中交换方法，是在类被添加到运行时的时候，被调用，而且只会被执行一次
 
 内联函数，会在编译的时候，将整个函数体，复制到调用函数的位置，不会创建单独的函数
 通常适合非常短小的函数，内联函数在 C++ 中使用的非常多！
 */
static inline void af_swizzleSelector(Class theClass, SEL originalSelector, SEL swizzledSelector) {
    // 1. 从一个类中获取到原始的方法，例如 setImage:
    Method originalMethod = class_getInstanceMethod(theClass, originalSelector);
    // 2. 从一个类中获取到交换的方法，例如 me_setImage:
    // 使用此函数之前，需要添加方法 me_setImage: 添加到 UIImageView 的 class 上
    Method swizzledMethod = class_getInstanceMethod(theClass, swizzledSelector);
    
    // 3. 交换方法的实现
    method_exchangeImplementations(originalMethod, swizzledMethod);
}

static inline BOOL af_addMethod(Class theClass, SEL selector, Method method) {
    // 给指定的类，添加一个方法
    return class_addMethod(theClass, selector,  method_getImplementation(method),  method_getTypeEncoding(method));
}

使用时只要将代码拷贝到做方法交换的地方在load中使用即可。

代码示例：

    

//
//  ViewController.m
//  06-交换方法
//
//  Created by xxxx on 16/1/14.
//  Copyright © 2016年 xxxx. All rights reserved.
//

#import "ViewController.h"
#import <objc/runtime.h>

@interface ViewController ()
@property (weak, nonatomic) IBOutlet UIImageView *imageView;

@end

@implementation ViewController

/**
 1. 将 AFN 中的两个静态函数复制过来
 2. 添加运行时的头文件
 3. 在 +load 方法中交换方法，是在类被添加到运行时的时候，被调用，而且只会被执行一次
 
 内联函数，会在编译的时候，将整个函数体，复制到调用函数的位置，不会创建单独的函数
 通常适合非常短小的函数，内联函数在 C++ 中使用的非常多！
 */
static inline void af_swizzleSelector(Class theClass, SEL originalSelector, SEL swizzledSelector) {
    // 1. 从一个类中获取到原始的方法，例如 setImage:
    Method originalMethod = class_getInstanceMethod(theClass, originalSelector);
    // 2. 从一个类中获取到交换的方法，例如 me_setImage:
    // 使用此函数之前，需要添加方法 me_setImage: 添加到 UIImageView 的 class 上
    Method swizzledMethod = class_getInstanceMethod(theClass, swizzledSelector);
    
    // 3. 交换方法的实现
    method_exchangeImplementations(originalMethod, swizzledMethod);
}

static inline BOOL af_addMethod(Class theClass, SEL selector, Method method) {
    // 给指定的类，添加一个方法
    return class_addMethod(theClass, selector,  method_getImplementation(method),  method_getTypeEncoding(method));
}

+ (void)load {
    
    // 交换方法有一个特点，一旦交换，全局变化！
    // see: http://nshipster.cn/swift-objc-runtime/
    // 1. 从当前类中 - 获得准备交叉的方法
    Method swizzledMethod = class_getInstanceMethod([self class], @selector(me_setImage:));
    
    // 2. 将方法添加到 `UIImageView` 的类上
    if (af_addMethod([UIImageView class], @selector(me_setImage:), swizzledMethod)) {
        // 3. 方法添加成功，交换 UIImageView 类中的对象方法
        af_swizzleSelector([UIImageView class], @selector(setImage:), @selector(me_setImage:));
    }
}

- (void)viewDidLoad {
    [super viewDidLoad];
    
    // 目标 -> 拦截 UIImageView 的 setImage: 方法
    self.imageView.image = [UIImage imageNamed:@"tabbar_compose_camera"];
}

// 方法一旦交换完成，代码本身虽然还在 viewController 中，但是已经被添加到了 UIImageView 上，
// 1> 名字被改成了 setImage:
// 2> 原有的 setImage: 被改成了 me_setImage:
// 注意：ViewController 自己千万不要调用 hm_serImage:，否则死循环！
- (void)me_setImage:(UIImage *)image {
    // self == UIImageView
    NSLog(@"%@ %@", image, self);
    
    // 调用系统的 setImage:
    [self me_setImage:image];
}

@end

转载于:https://my.oschina.net/quntion/blog/614373