小码哥底层原理笔记:Catgory关联对象的原理

我们知道Catgory可以定义属性,但是不能定义成员变量。比如我们这样写:

@interface Person (Test)
{
   int _weight;
}

@property (nonatomic, assign) int weight;//这句代码只会做一件事
@end

编译一下,是会报错的,提示分类不能添加成员变量。我们只能像下面这样定义属性,
我们先创建一个Person的分类Test:

@interface Person (Test)

@property (nonatomic, copy) NSString *name;

@property (nonatomic, assign) int weight;//这句代码只会做一件事
//就是只会生成这两句声明
/*
 - (void)setWeight:(int)weight;
 - (int)weight;
 */

@end

#import "Person+Test.h"
#import <objc/runtime.h>

@implementation Person (Test)

@end

如上代码所示,我们给这个Person分类Test定义了两个属性,一个是@property (nonatomic, copy) NSString *name;,一个是@property (nonatomic, assign) int weight;
现在我们执行以下代码:

Person *person = [[Person alloc] init];
person.weight = 20;

发现闪退了,报错是提示没有找到setWeight:方法。这是因为通过@property (nonatomic, copy) NSString *name;定义的属性只是声明了一个get和set方法,但是并没有实现get,set方法。那么我们要像Person类那样正常去使用name和weight属性,我们就必须在.m文件里面实现对应的set和get方法。在set方法中保存weight的值20,然后在get方法中取出20。

那么我们之间跟之前那样写行不行呢,像这样,

@implementation Person (Test)

- (void) setWeight:(int) weight{
   _weight = weight;
}

- (int)weight{
 return _weight;
}
@end

我们运行发现这样行不通,打印weight的值是0,并不是我们赋值的20。那是因为在正常Person类里面是把值保存在成员变量里面的。现在分类是没有成员变量的,所以无法保存值20。

对此我们的主要问题就是在没有成员变量的情况下如何把赋给weight的值20保存起来,我们有以下几种方案,在.m里面定义一个全局变量保存赋给weight的值,或者在.m里面定义一个全局字典,通过key-value来保存对应的属性的赋值。还有就是通过runtime的API方法来实现,如下:

#import "Person+Test.h"
#import <objc/runtime.h>

@implementation Person (Test)

- (void)setWeight:(int)weight{
    objc_setAssociatedObject(self, @selector(weight), @(weight), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (int)weight{
    return [objc_getAssociatedObject(self, _cmd) intValue];
}

- (void)setName:(NSString *)name{//_cmd =@selector(setName:)
    objc_setAssociatedObject(self, @selector(name), name, OBJC_ASSOCIATION_COPY_NONATOMIC);
}

- (NSString *)name{//_cmd代表当前方法的selector,_cmd =@selector(name);,但是set方法不能这样写
    return objc_getAssociatedObject(self, _cmd);
}

//static const char MJNameKey;//定义一个只占一个字节的字符串。比下面这个方法要更好
//static const char MJWeightKey;
//
//- (void)setWeight:(int)weight{
//    objc_setAssociatedObject(self, &MJWeightKey, @(weight), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
//}
//
//- (int)weight{
//    return [objc_getAssociatedObject(self, &MJWeightKey) intValue];
//}
//
//- (void)setName:(NSString *)name{
//    objc_setAssociatedObject(self, &MJNameKey, name, OBJC_ASSOCIATION_COPY_NONATOMIC);
//}
//
//- (NSString *)name{
//    return objc_getAssociatedObject(self, &MJNameKey);
//}

//static const void *MJNameKey = &MJNameKey;
//static const void *MJWeightKey = &MJWeightKey;
//
//- (void)setWeight:(int)weight{
//    objc_setAssociatedObject(self, MJWeightKey, @(weight), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
//}
//
//- (int)weight{
//    return [objc_getAssociatedObject(self, MJWeightKey) intValue];
//}
//
//- (void)setName:(NSString *)name{
//    objc_setAssociatedObject(self, MJNameKey, name, OBJC_ASSOCIATION_COPY_NONATOMIC);
//}
//
//- (NSString *)name{
//    return objc_getAssociatedObject(self, MJNameKey);
//}

@end

这几种方法都可以,其中第一种是最好的。还是回到上一个问题,这样用runtime实现的话,那这个成员变量是保存在哪里呢?是合并到Person类里面吗?答案不是的。runtime为这个分类的关联对象单独进行管理,并不是合并到Person类里面了。

其实runtime是生成一个AssociationsManager类用来管理分类的关联对象,让分类能正常使用成员变量,有点类似于前面说到的用字典去保存的原理。通过查看源码,我们可以知道AssociationsManager类有一个AssociationsHashMap属性,这个属性是相当于一个字典,用来存储对象-关联对象的,也就是它的key是我们关联对象时传的self,也就是这个Person分类,以这个为key,然后value是一个ObjectAssociationMapObjectAssociationMap对象也相当于是一个字典,这个字典的key是我们关联对象时传进去的那个key,value是ObjcAssociationObjcAssociation对象里面有两个属性_value_policy,这两个就是我们关联对象的值和关联策略了。

简单说AssociationsHashMap存储的是项目中所有分类的关联对象,里面应该是长这样{Person分类: AssociationsHashMap,Student分类: AssociationsHashMap},我们项目中有几个分类有关联对象,那AssociationsHashMap里面就有多少个元素。而AssociationsHashMap里面存放的就是关联对象的keyObjcAssociation,里面应该长这样{@selector(weight):@(weight),@selector(name):name},一个分类里面有多少个关联对象AssociationsHashMap里面就有多少个元素。最后ObjcAssociation就是保存着关联对象的值和关联策略了。ObjcAssociation{unitptr_t _policy= OBJC_ASSOCIATION_RETAIN_NONATOMIC; id _value=@(weight)}。以下图很好的展示了其中的原理。

分类关联对象的原理

注意:关联策略是没有weak类型的。

对应的关联策略

面试题

1、Category能否添加成员变量?如果可以,如何给Category添加成员变量?
答:不能直接给Category添加成员变量,但是可以间接实现Category有成员变量的效果

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,126评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,254评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,445评论 0 341
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,185评论 1 278
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,178评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,970评论 1 284
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,276评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,927评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,400评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,883评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,997评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,646评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,213评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,204评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,423评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,423评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,722评论 2 345

推荐阅读更多精彩内容