oc中分类作为类扩展的一种方式,可以允许在不知道和不改动原有类的基础上为该类扩充一些方法,但是分类本身是不允许扩充数据成员的。那么如何给分类扩充数据成员呢?这里利用runtime运行时的objc_方法,就可以分类动态的扩充成员变量。
1.首先包含头文件:
#import <objc/runtime.h>
2.在类声明中添加要扩充的属性:
@interface NSDictionary (extension)
@property (nonatomic, assign, readonly) FaceModelParam *faceModel;
@property (nonatomic, strong, readonly) NSMutableDictionary *dict;
@end
3.在类实现中,重写该属性的set方法 ,在set方法中将该类与传入的值动态绑定,并指定一个key(这里的key可以随意指定,一般是const static char*类型):
const static char *FaceModelKey = "faceModelKey";
const static char *DictKey = "dictKey";
- (void)setFaceModel:(FaceModelParam *)faceModel
{
objc_setAssociatedObject(self, FaceModelKey, [NSValue valueWithPointer:faceModel], OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (void)setDict:(NSMutableDictionary *)dict
{
objc_setAssociatedObject(self, DictKey, dict, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
4.重写get方法,根据key获取到动态绑定的数据:
- (FaceModelParam *)faceModel
{
NSValue *value = objc_getAssociatedObject(self, FaceModelKey);
FaceModelParam *faceModelPoint = [value pointerValue];
return faceModelPoint;
}
- (NSMutableDictionary *)dict
{
return objc_getAssociatedObject(self, DictKey);
}
这样即可以像之前一样,利用点语法来获取和修改添加的数据成员了。
注意:
在我理解来看,这种方法看似添加了数据成员,其实并没有,只是在分类内部可以使用传入的值而已。其实分类本身并没有该数据成员,在外部初始化一个类对象后,调用点语法也无法获取到之前传入的值,所以这种做法还是有一定局限性的。