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)kz_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)kz_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 KZTapActionBlockKey = '\0';
- (TapActionBlock)tapBlock {
return objc_getAssociatedObject(self, &KZTapActionBlockKey);
}
- (void)setTapBlock:(TapActionBlock)tapBlock {
objc_setAssociatedObject(self, &KZTapActionBlockKey, tapBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);
}
@end
参考资料: