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

参考资料:

iOS 关于 Category

MJRefresh

BlocksKit