問題:“如何給OC對象(如 NSArray、UIViewController等)添加一個屬性,且不使用繼承的情況下。”
兩種方式:類擴展、分類+關(guān)聯(lián)對象(runtime)
第一種 類擴展 (Class Extension也有人稱為匿名分類)
- 作用:
- 能為某個類附加額外的屬性,成員變量,方法聲明
- 一般的類擴展寫到.m文件中
- 一般的私有屬性寫到類擴展
- 使用格式:
@interface NSArray()
//屬性
//方法
@end
第二種 分類+關(guān)聯(lián)對象(runtime)
- 先介紹下分類:
- 分類的小括號中必須有名字
@interface 類名(分類名字)
/*方法聲明*/
@end
@implementation 類名(分類名字)
/*方法實現(xiàn)*/
@end
- 分類只能擴充方法,不能擴展屬性和成員變量(如果包含成員變量會直接報錯)
- 如果分類中聲明了一個屬性,那么分類只會生成這個屬性的set、get方法聲明,也就是不會有實現(xiàn)。
- 舉例說明:如果我們分別在,類擴展與分類中添加了兩個屬性
//類別
@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 \\\\'
意思是說,我們所創(chuàng)建的對象中并沒有mOne這個屬性。也就是說雖然我們再類別中聲明屬性不會報錯,但是@property并沒有自動為我們設(shè)置的屬性生成set、get方法*。
現(xiàn)在我們結(jié)合關(guān)聯(lián)對象來實現(xiàn)
什么是關(guān)聯(lián)對象
關(guān)聯(lián)對象是指某個OC對象通過一個唯一的key連接到一個類的實例上。
舉個例子:xiaoming是Person類的一個實例,他的dog(一個OC對象)通過一根繩子(key)被他牽著散步,這可以說xiaoming和dog是關(guān)聯(lián)起來的,當然xiaoming可以牽著多個dog。
怎樣關(guān)聯(lián)對象
runtime提供給我們的方法:
//關(guān)聯(lián)對象
void objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
//獲取關(guān)聯(lián)的對象
id objc_getAssociatedObject(id object, const void *key)
//移除關(guān)聯(lián)的對象
void objc_removeAssociatedObjects(id object)
變量說明:
id object:被關(guān)聯(lián)的對象(如xiaoming)
const void *key:關(guān)聯(lián)的key,要求唯一
id value:關(guān)聯(lián)的對象(如dog)
objc_AssociationPolicy policy:內(nèi)存管理的策略
objc_AssociationPolicy policy的enum值有:
OBJC_ASSOCIATION_ASSIGN = 0, //弱引用關(guān)聯(lián)對象
//等價于@property (assign) , @property (unsafe_unretained)
OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, //強引用關(guān)聯(lián)對象,且為非原子操作
//等價于@property (strong, nonatomic)
OBJC_ASSOCIATION_COPY_NONATOMIC = 3, //復(fù)制關(guān)聯(lián)對象,且為非原子操作
//等價于@property (copy, nonatomic)
OBJC_ASSOCIATION_RETAIN = 01401, //強引用關(guān)聯(lián)對象,且為原子操作
//等價于@property (strong, atomic)
OBJC_ASSOCIATION_COPY = 01403 //復(fù)制關(guān)聯(lián)對象,且為原子操作
等價于@property (copy, atomic)
//其中,第 2 種與第 4 種、第 3 種與第 5 種關(guān)聯(lián)策略的唯一差別就在于操作是否具有原子性
當對象被釋放時,會根據(jù)這個策略來決定是否釋放關(guān)聯(lián)的對象,當策略是RETAIN/COPY時,會釋放(release)關(guān)聯(lián)的對象,當是ASSIGN,將不會釋放。
值得注意的是,我們不需要主動調(diào)用removeAssociated來接觸關(guān)聯(lián)的對象,如果需要解除指定的對象,可以使用setAssociatedObject置nil來實現(xiàn)。
關(guān)聯(lián)對象的應(yīng)用##
1、添加公共屬性
這是最常用的一個模式,通常我們會在類聲明里面添加屬性,但是出于某些需求(如前言描述的情況),我們需要在分類里添加一個或多個屬性的話,編譯器就會報錯,這個問題的解決方案就是使用runtime的關(guān)聯(lián)對象。
應(yīng)用舉例:
我們需要自定義一個tabbar,并暴露公共的屬性和方法。(讀者們可以思考下使用繼承和分類實現(xiàn)的優(yōu)點和不足之處)
@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、添加私有成員變量
有時候,需要在分類中添加不想暴露在公共聲明的成員變量。應(yīng)用舉例:給按鈕添加點擊時間的回調(diào)
@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:以上代碼是否會存在內(nèi)存管理的問題
3、關(guān)聯(lián)KVO觀察者
有時候我們在分類中使用KVO,推薦使用關(guān)聯(lián)的對象作為觀察者,盡量避免對象觀察自身。
此應(yīng)用模式不再舉例,有興趣的讀者可以自行深入研究,或者將代碼貼到評論處。