Objective-C的分類(lèi)無(wú)法直接添加實(shí)例變量,但我們可以利用關(guān)聯(lián)對(duì)象(AssociatedObject)給分類(lèi)添加屬性。為了避免循環(huán)引用而需要使用weak修飾符,可以使用OBJC_ASSOCIATION_ASSIGN,OBJC_ASSOCIATION_RETAIN_NONATOMIC等關(guān)聯(lián)對(duì)象的內(nèi)存管理策略。以下是示例代碼:
#import <objc/runtime.h>
static char associatedObjectKey;
@interface NSObject (MyCategory)
@property (nonatomic, weak) id myAssociatedObject;
@end
@implementation NSObject (MyCategory)
- (id)myAssociatedObject {
return objc_getAssociatedObject(self, &associatedObjectKey);
}
- (void)setMyAssociatedObject:(id)object {
objc_setAssociatedObject(self, &associatedObjectKey, object, OBJC_ASSOCIATION_ASSIGN);
}
@end
在這個(gè)示例代碼中,我們使用關(guān)聯(lián)對(duì)象給NSObject分類(lèi)添加了一個(gè)名為myAssociatedObject的屬性,它是一個(gè)弱引用。我們首先定義一個(gè)靜態(tài)變量associatedObjectKey作為關(guān)聯(lián)對(duì)象的key。
使用objc_getAssociatedObject函數(shù)獲取對(duì)象的關(guān)聯(lián)對(duì)象值,使用objc_setAssociatedObject函數(shù)設(shè)置對(duì)象的關(guān)聯(lián)對(duì)象值。
在單例實(shí)現(xiàn)中,我們可以使用這個(gè)關(guān)聯(lián)對(duì)象weak屬性來(lái)存儲(chǔ)單例對(duì)象,以防止循環(huán)引用。