做開發(fā)時(shí)我們常常會(huì)需要在已經(jīng)實(shí)現(xiàn)了的類中增加一些方法主慰,這時(shí)候我們一般會(huì)用Category的方式來做。但是這樣做我們也只能擴(kuò)展一些方法鲫售,而有時(shí)候我們更多的是想給它增加一個(gè)屬性共螺。由于類已經(jīng)是編譯好的了,就不能靜態(tài)的增加成員了情竹,這樣我們就需要自己來實(shí)現(xiàn)getter和setter方法了藐不,在這些方法中動(dòng)態(tài)的讀寫屬性變量來實(shí)現(xiàn)屬性。一種比較簡單的做法是使用Objective-C運(yùn)行時(shí)的這兩個(gè)方法:
void objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy);
id objc_getAssociatedObject(id object, const void *key);
這兩個(gè)方法可以讓一個(gè)對(duì)象和另一個(gè)對(duì)象關(guān)聯(lián)秦效,就是說一個(gè)對(duì)象可以保持對(duì)另一個(gè)對(duì)象的引用佳吞,并獲取那個(gè)對(duì)象。有了這些棉安,就能實(shí)現(xiàn)屬性功能了。 policy可以設(shè)置為以下這些值:
enum {
OBJC_ASSOCIATION_ASSIGN = 0,
OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1,
OBJC_ASSOCIATION_COPY_NONATOMIC = 3,
OBJC_ASSOCIATION_RETAIN = 01401,
OBJC_ASSOCIATION_COPY = 01403
};
這些值跟屬性定義中的nonatomic铸抑,copy贡耽,retain等關(guān)鍵字的功能類似。
Example
下面是一個(gè)屬性自定義getter和setter的例子:
NSString const * kExposeController = @"exposeController";
- (UIViewController *)exposeController {
return (UIViewController *)objc_getAssociatedObject(self, kExposeController);
}
- (void)setExposeController:(UIViewController *)exposeController {
objc_setAssociatedObject(self, kExposeController, exposeController, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
可以看出使用objc_setAssociatedObject和objc_getAssociatedObject函數(shù)可以很方便的實(shí)現(xiàn)屬性的getter和setter鹊汛。
一個(gè)很方便的宏
為此蒲赂,我特意寫了一個(gè)Synthesize宏,可以提供@synthesize類似的功能刁憋±淖欤可以支持兩種最常用的屬性:非原子retain和assign屬性(如果需要其他類型的屬性可自行修改)。
#import <objc/runtime.h>
#define SYNTHESIZE_CATEGORY_OBJ_PROPERTY(propertyGetter, propertySetter)
- (id) propertyGetter {
return objc_getAssociatedObject(self, @selector( propertyGetter ));}
- (void) propertySetter (id)obj{
objc_setAssociatedObject(self, @selector( propertyGetter ), obj, OBJC_ASSOCIATION_RETAIN_NONATOMIC); }
#define SYNTHESIZE_CATEGORY_VALUE_PROPERTY(valueType, propertyGetter, propertySetter)
- (valueType) propertyGetter {
valueType ret = {0};
[objc_getAssociatedObject(self, @selector( propertyGetter )) getValue:&ret];
return ret;
}
- (void) propertySetter (valueType)value{
NSValue *valueObj = [NSValue valueWithBytes:&value objCType:@encode(valueType)];
objc_setAssociatedObject(self, @selector( propertyGetter ), valueObj, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
用這個(gè)宏只需要指定相關(guān)屬性的類型至耻,getter和setter就可以快速的實(shí)現(xiàn)一個(gè)屬性若皱。比如在UIAlert的Category實(shí)現(xiàn)一個(gè)非原子retain屬性u(píng)serInfo,以及一個(gè)assign的類型為CGRect的customArea屬性:
@interface UIAlertView (Ex)
@property(nonatomic, retain) id userInfo;
@property(nonatomic) CGRect customArea;
@end
@implementation UIAlertView (Ex)
SYNTHESIZE_CATEGORY_OBJ_PROPERTY(userInfo, setUserInfo:)SYNTHESIZE_CATEGORY_VALUE_PROPERTY(CGRect, customArea, setCustomArea:)
@end