category 是一个平常开发经常会用到的一个技术点,不过大多数情况下,也只是对某一个类添加一些实例方法 或者类方法。一般也足以满足需求。
不过如果在方法中,有需要传递变量时,仅仅靠方法就不够了。
例如,为 UIView 添加一个点击手势,传入一个 block。 就需要 UIView 持有一个 block。
在Objective-C提供的runtime函数中,确实有一个class_addIvar()函数用于给类添加成员变量,但是这个函数只能在“构建一个类的过程中”调用。一旦完成类定义,就不能再添加成员变量了。
那么在 category 的 .h 添加了 @property 的时候,只会生成对应的 getter 和 setter 方法,并不会有实例变量的产生。因为类分配的内存区域在编译时就确定了。
为什么可以在类别中添加方法和属性呢?
因为方法和属性并不“属于”类实例,而成员变量“属于”类实例。
方法定义是在objc_class中管理的,不管如何增删类方法,都不影响类实例的内存布局,已经创建出的类实例仍然可正常使用。
如果需要在 category 中添加实例变量怎么处理呢?
这时候就需要使用到黑魔法 Runtime ,因为 OC 的是一门动态语言,有运行时的特性。所以可以利用 Runtime 的关联方法,让两个对象关联起来。
代码如下,
// UIView+TapBlock.h
#import <UIKit/UIKit.h>
@interface UIView (TapBlock)
typedef void (^TapActionBlock)(void);
- (void)bs_whenTapped:(TapActionBlock)block;
@end
// UIView+TapBlock.m
#import "UIView+TapBlock.h"
#import <objc/runtime.h>
@interface UIView (TapBlockInternal)
@property (nonatomic, copy) TapActionBlock tapBlock;
@end
@implementation UIView (TapBlock)
- (void)bs_whenTapped:(TapActionBlock)block {
self.tapBlock = block;
self.userInteractionEnabled = YES;
UITapGestureRecognizer *tap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(tapAction)];
[self addGestureRecognizer:tap];
}
- (void)tapAction {
if (self.tapBlock) {
self.tapBlock();
}
}
static const char BSTapActionBlockKey = '\0';
- (TapActionBlock)tapBlock {
return objc_getAssociatedObject(self, &BSTapActionBlockKey);
}
- (void)setTapBlock:(TapActionBlock)tapBlock {
/*
objc_AssociationPolicy参数使用的策略:
OBJC_ASSOCIATION_ASSIGN; //assign策略
OBJC_ASSOCIATION_COPY_NONATOMIC; //copy策略
OBJC_ASSOCIATION_RETAIN_NONATOMIC; // retain策略
OBJC_ASSOCIATION_RETAIN;
OBJC_ASSOCIATION_COPY;
*/
/*
关联方法:
objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy);
参数:
* id object 给哪个对象的属性赋值
const void *key 属性对应的key
id value 设置属性值为value
objc_AssociationPolicy policy 使用的策略,是一个枚举值,和copy,retain,assign是一样的,手机开发一般都选择NONATOMIC
*/
objc_setAssociatedObject(self, &BSTapActionBlockKey, tapBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);
}
@end
文中借鉴的资料链接:
https://blog.csdn.net/mumuyinyin/article/details/72854579
https://github.com/CoderMJLee/MJRefresh
https://github.com/BlocksKit/BlocksKit