#import <Foundation/Foundation.h>
@interface NSObject (Add)
/* 调换实例方法 */
+ (BOOL)swizzleInstanceMethod:(SEL)originalSel with:(SEL)newSel;
/* 调换类方法 */
+ (BOOL)swizzleClassMethod:(SEL)originalSel with:(SEL)newSel;
/* 给类添加属性 */
- (void)setAssociateValue:(id)value withKey:(void *)key;
/* 给类添加弱引用属性,上面的就可以理解为强引用属性 */
- (void)setAssociateWeakValue:(id)value withKey:(void *)key;
/* 获取某个类的属性 */
- (id)getAssociatedValueForKey:(void *)key;
/* 删除一个类的所有属性 */
- (void)removeAssociatedValues;
/* 获取类的所有属性 */
+ (NSArray *)getAllProperties;
@end
#import "NSObject+Add.h"
#import <objc/runtime.h>
@implementation NSObject (Add)
+ (BOOL)swizzleInstanceMethod:(SEL)originalSel with:(SEL)newSel {
//获取某个类的对象方法： class_getInstanceMethod
Method originalMethod = class_getInstanceMethod(self, originalSel);
Method newMethod = class_getInstanceMethod(self, newSel);
if (!originalMethod || !newMethod) return NO;
//给某个类添加方法 class_addMethod，获取一个方法的实现IMP：class_getMethodImplementation，
//获取方法的编码方式 method_getTypeEncoding
class_addMethod(self,
originalSel,
class_getMethodImplementation(self, originalSel),
method_getTypeEncoding(originalMethod));
BOOL didAddMethod = class_addMethod(self,
newSel,
class_getMethodImplementation(self, newSel),
method_getTypeEncoding(newMethod));
/*
如果didAddMethod返回YES，说明当前类中没有要替换方法的实现，我们需要在父类中去寻找。这个时候就需要用到method_getImplementation去获取class_getInstanceMethod里面的方法实现。然后再进行class_replaceMethod来实现Swizzling。
*/
if (didAddMethod) {
class_replaceMethod(self, newSel, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
}else{
method_exchangeImplementations(class_getInstanceMethod(self, originalSel),
class_getInstanceMethod(self, newSel));
}
//交换两个方法的实现IMP：method_exchangeImplementations
return YES;
}
+ (BOOL)swizzleClassMethod:(SEL)originalSel with:(SEL)newSel {
/*
对象的实例方法调用时，通过对象的 isa 在类中获取方法的实现。
类对象的类方法调用时，通过类的 isa 在元类中获取方法的实现。
上面这两句话是简书 神经病院Objective-C Runtime入院第一天——isa和Class 里面的，
所以 object_getClass(self) 其实是获取元类,然后在元类里找类方法的实现
*/
//获取类：object_getClass
Class class = object_getClass(self);
Method originalMethod = class_getInstanceMethod(class, originalSel);
Method newMethod = class_getInstanceMethod(class, newSel);
if (!originalMethod || !newMethod) return NO;
//交换两个方法的实现IMP：method_exchangeImplementations
method_exchangeImplementations(originalMethod, newMethod);
return YES;
}
- (void)setAssociateValue:(id)value withKey:(void *)key{
//添加属性objc_setAssociatedObject,
// OBJC_ASSOCIATION_RETAIN_NONATOMIC ： (nonatomic,retain)
objc_setAssociatedObject(self, key, value, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (void)setAssociateWeakValue:(id)value withKey:(void *)key {
//OBJC_ASSOCIATION_ASSIGN: (nonatomic,assign)
objc_setAssociatedObject(self, key, value, OBJC_ASSOCIATION_ASSIGN);
}
- (id)getAssociatedValueForKey:(void *)key {
//根绝key获取属性
return objc_getAssociatedObject(self, key);
}
- (void)removeAssociatedValues {
//删除当前类的所有属性
objc_removeAssociatedObjects(self);
}
+ (NSArray *)getAllProperties {
//用于存入属性数量
unsigned int outCount = 0;
//获取属性数组
objc_property_t *propertyList = class_copyPropertyList([self class], &outCount);
NSMutableArray *arrM = [NSMutableArray arrayWithCapacity:outCount];
//遍历数组
for (int i = 0; i < outCount; ++i) {
objc_property_t property = propertyList[i];
//获取属性名
const char *cName = property_getName(property);
//将其转换成c字符串
NSString *propertyName = [NSString stringWithCString:cName encoding:NSUTF8StringEncoding];
// 加入数组
[arrM addObject:propertyName];
}
//在使用了c函数的creat, copy等函数是记得手动释放,要不然会引起内存泄露问题
free(propertyList);
return arrM.copy;
}
@end
对象的实例方法调用时，通过对象的 isa 在类中获取方法的实现。
类对象的类方法调用时，通过类的 isa 在元类中获取方法的实现。
上面这两句话是简书 神经病院Objective-C Runtime入院第一天——isa和Class 里面的，
SwizzleMethod(object_getClass([Foo class]), @selector(bar), @selector(swz_bar));
所以 object_getClass([Foo class]) 其实是获取元类,然后在元类里找类方法的实现
// SwizzleMethod(object_getClass([Foo class]), @selector(bar), @selector(swz_bar));
//如果要替换某个类的方法，可以给指定类建一个分类，并将替换方法的实现写在分类里
#import "Foo+Swizzle.h"
#import "NSObject+Add.h"
@implementation Foo (Swizzle)
/*
1, Swizzling应该总在+load中执行
Objective-C在运行时会自动调用类的两个方法+load和+initialize。+load会在类初始加载时调用， +initialize方法是以懒加载的方式被调用的，如果程序一直没有给某个类或它的子类发送消息，那么这个类的 +initialize方法是永远不会被调用的。所以Swizzling要是写在+initialize方法中，是有可能永远都不被执行
2.Swizzling应该总是在dispatch_once中执行
Swizzling会改变全局状态，所以在运行时采取一些预防措施，使用dispatch_once就能够确保代码不管有多少线程都只被执行一次。这将成为Method Swizzling的最佳实践。
这里有一个很容易犯的错误，那就是继承中用了Swizzling。如果不写dispatch_once就会导致Swizzling失效！
3.Swizzling在+load中执行时，不要调用[super load]
*/
+(void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
[Foo swizzleClassMethod:@selector(bar) with:@selector(swz_bar)];
});
}
+ (void)swz_bar {
NSLog(@"Before calling ----");
[self swz_bar];
NSLog(@"After calling ----");
}
@end
