iOS runtime应用:关联对象

问题:“如何给OC对象(如 NSArray、UIViewController等)添加一个属性,且不使用继承的情况下。”
两种方式:类扩展、分类+关联对象(runtime)

第一种 类扩展 (Class Extension也有人称为匿名分类)

  • 作用:
    • 能为某个类附加额外的属性,成员变量,方法声明
    • 一般的类扩展写到.m文件中
    • 一般的私有属性写到类扩展
  • 使用格式:
@interface NSArray()
//属性
//方法
@end

第二种 分类+关联对象(runtime)

  • 先介绍下分类:
    • 分类的小括号中必须有名字
@interface 类名(分类名字)
/*方法声明*/
@end
@implementation 类名(分类名字)
/*方法实现*/
@end
  • 分类只能扩充方法,不能扩展属性和成员变量(如果包含成员变量会直接报错)
  • 如果分类中声明了一个属性,那么分类只会生成这个属性的set、get方法声明,也就是不会有实现。
  • 举例说明:如果我们分别在,类扩展与分类中添加了两个属性
//类别
@interface UIViewController(My)
@property(nonatomic,copy)NSString *mOne;
@end
//类扩展
@interface UIViewController()
@property(nonatomic,copy)NSString *mTwo;
@end
-(instancetype)init{
  if (self = [super init]) {
      self.mTwo = @"two two";
      self.mOne = @"one one";//这行报错 崩溃
  }
}

po self.mOne
error: property 'mOne' not found on object of type 'UIViewController \\\\'
意思是说,我们所创建的对象中并没有mOne这个属性。也就是说虽然我们再类别中声明属性不会报错,但是
@property并没有自动为我们设置的属性生成set、get方法*。

现在我们结合关联对象来实现
什么是关联对象
关联对象是指某个OC对象通过一个唯一的key连接到一个类的实例上。
举个例子:xiaoming是Person类的一个实例,他的dog(一个OC对象)通过一根绳子(key)被他牵着散步,这可以说xiaoming和dog是关联起来的,当然xiaoming可以牵着多个dog。
怎样关联对象
runtime提供給我们的方法:

//关联对象
void objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
//获取关联的对象
id objc_getAssociatedObject(id object, const void *key)
//移除关联的对象
void objc_removeAssociatedObjects(id object)

变量说明:

id object:被关联的对象(如xiaoming)
const void *key:关联的key,要求唯一
id value:关联的对象(如dog)
objc_AssociationPolicy policy:内存管理的策略

objc_AssociationPolicy policy的enum值有:

OBJC_ASSOCIATION_ASSIGN = 0, //弱引用关联对象
//等价于@property (assign) , @property (unsafe_unretained)

OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, //强引用关联对象,且为非原子操作
//等价于@property (strong, nonatomic)

OBJC_ASSOCIATION_COPY_NONATOMIC = 3, //复制关联对象,且为非原子操作
//等价于@property (copy, nonatomic)

OBJC_ASSOCIATION_RETAIN = 01401, //强引用关联对象,且为原子操作
//等价于@property (strong, atomic)

OBJC_ASSOCIATION_COPY = 01403 //复制关联对象,且为原子操作
等价于@property (copy, atomic)

//其中,第 2 种与第 4 种、第 3 种与第 5 种关联策略的唯一差别就在于操作是否具有原子性

当对象被释放时,会根据这个策略来决定是否释放关联的对象,当策略是RETAIN/COPY时,会释放(release)关联的对象,当是ASSIGN,将不会释放。
值得注意的是,我们不需要主动调用removeAssociated来接触关联的对象,如果需要解除指定的对象,可以使用setAssociatedObject置nil来实现。

关联对象的应用##

1、添加公共属性

这是最常用的一个模式,通常我们会在类声明里面添加属性,但是出于某些需求(如前言描述的情况),我们需要在分类里添加一个或多个属性的话,编译器就会报错,这个问题的解决方案就是使用runtime的关联对象。
应用举例:
我们需要自定义一个tabbar,并暴露公共的属性和方法。(读者们可以思考下使用继承和分类实现的优点和不足之处)

@interface UITabBarController (Custom)
@property (nonatomic, strong) MyCustomTabbar * customTabbar;
@end
#import "UITabBarController+Custom.h"
#import <objc/runtime.h>
@implementation UITabBarController (Custom)
- (void)setCustomTabbar:(MyCustomTabbar *)customTabbar {
//这里使用方法的指针地址作为唯一的key
    objc_setAssociatedObject(self, @selector(customTabbar), customTabbar, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (MyCustomTabbar *)customTabbar { 
    return objc_getAssociatedObject(self, @selector(customTabbar));
}
//其他方法...
@end

这样,我们就可以像原生的tabbar一样使用自定义的tabbar:

[self.tabBarController.customTabbar doSomgthig];

2、添加私有成员变量
有时候,需要在分类中添加不想暴露在公共声明的成员变量。应用举例:給按钮添加点击时间的回调

@interface UIButton (Callback)
- (instancetype)initWithFrame:(CGRect)frame callback:(void (^)(UIButton *))callbackBlock;
@end
@interface UIButton ()
@property (nonatomic, copy) void (^callbackBlock)(UIButton * button);
@end
@implementation UIButton (Callback)
- (void (^)(UIButton *))callbackBlock { 
  return objc_getAssociatedObject(self, @selector(callbackBlock));
}
- (void)setCallbackBlock:(void (^)(UIButton *))callbackBlock {
  objc_setAssociatedObject(self, @selector(callbackBlock), callbackBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);
}
- (instancetype)initWithFrame:(CGRect)frame callback:(void (^)(UIButton *))callbackBlock {
  if (self = [super initWithFrame:frame]) 
  {
    self.callbackBlock = callbackBlock;
    [self addTarget:self action:@selector(didClickAction:) forControlEvents:UIControlEventTouchUpInside];
  } return self;
}
- (void)didClickAction:(UIButton *)button {
  self.callbackBlock(button);
}
@end

PS:以上代码是否会存在内存管理的问题

3、关联KVO观察者
有时候我们在分类中使用KVO,推荐使用关联的对象作为观察者,尽量避免对象观察自身。
此应用模式不再举例,有兴趣的读者可以自行深入研究,或者将代码贴到评论处。

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

推荐阅读更多精彩内容