iOS runtime實戰應用：關聯對象

標籤：模式   擷取   副作用   作用   self   array   nat   call   run   

在開始之前建議先閱讀iOS runtime的基礎理解篇：iOS內功篇：runtime

有筒子在面試的時候，遇到這樣一個問題：“如何給NSArray添加一個屬性（不能使用繼承）”，筒子立馬蒙逼了，不能用繼承，難道用分類？但是分類貌似只能添加方法不能添加屬性啊，筒子百思不得其解，直到後來接觸到了runtime才恍然大悟。

什麼是關聯對象

關聯對象是指某個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,          OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, OBJC_ASSOCIATION_COPY_NONATOMIC = 3,  OBJC_ASSOCIATION_RETAIN = 01401,       OBJC_ASSOCIATION_COPY = 01403

當對象被釋放時，會根據這個策略來決定是否釋放關聯的對象，當策略是RETAIN/COPY時，會釋放（release）關聯的對象，當是ASSIGN，將不會釋放。
值得注意的是，我們不需要主動調用removeAssociated來接觸關聯的對象，如果需要解除指定的對象，可以使用setAssociatedObject置nil來實現。

關聯對象的應用

1、添加公用屬性

這是最常用的一個模式，通常我們會在類聲明裡面添加屬性，但是出於某些需求（如前言描述的情況），我們需要在分類裡添加一個或多個屬性的話，編譯器就會報錯，這個問題的解決方案就是使用runtime的關聯對象。
應用舉例：
我們需要自訂一個tabbar，並暴露公用的屬性和方法。（讀者們可以思考下使用繼承和分類實現的優點和不足之處）

@interface UITabBarController (Custom)@property (nonatomic, strong) SUCustomTabbar * customTabbar;@end

#import "UITabBarController+Custom.h"#import <objc/runtime.h>@implementation UITabBarController (Custom)- (void)setCustomTabbar:(UIView *)customTabbar {    //這裡使用方法的指標地址作為唯一的key    objc_setAssociatedObject(self, @selector(customTabbar), customTabbar, OBJC_ASSOCIATION_RETAIN_NONATOMIC);}- (UIView *)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

讀者思考：以上代碼是否會存在記憶體管理的問題

3、關聯KVO觀察者
有時候我們在分類中使用KVO，推薦使用關聯的對象作為觀察者，盡量避免對象觀察自身。
此應用模式不再舉例，有興趣的讀者可以自行深入研究，或者將代碼貼到評論處。

思考

1、關聯對象更多的應用模式？
2、關聯對象是否是解決問題的首選？
3、關聯對象的副作用？

歡迎大家交流探討。

 

轉自：http://www.jianshu.com/p/c68cc81ef763

iOS runtime實戰應用：關聯對象