給分類(Category)添加屬性


遇到一個問題,寫了一個分類,但原先類的屬性不夠用。添加一個屬性,調用的時候崩潰了,說是找不到getter、setter方法。查了下文檔發現,OC的分類允許給分類添加屬性,但不會自動生成getter、setter方法。有沒有解決方案呢?有,通過運行時建立關聯引用。接下來以添加一個這樣的屬性為例:

@property (nonatomic, copy) NSString *str;

1、引入運行時頭文件。

#import <objc/runtime.h>

2、在匿名分類或者頭文件中添加屬性。區別是:匿名分類中添加的是私有屬性,只在本類中可以使用,類的實例中不可以使用。頭文件中添加的在類的實例中也可以使用。

//分類的頭文件
@interface ClassName (CategoryName)
//我要添加一個實例也可以訪問的變量所以就寫在這里了
@property (nonatomic, strong) NSString *str;
@end

//匿名分類
@interface ClassName ()

@end

3、在實現里面寫要添加屬性的getter、setter方法。

@implementation ClassName (CategoryName) 

-(void)setStr:(NSString *)str  
{  
    objc_setAssociatedObject(self, &strKey, str, OBJC_ASSOCIATION_COPY);  
}  

-(NSString *)str  
{  
    return objc_getAssociatedObject(self, &strKey);  
}
@end

setStr:方法中使用了一個objc_setAssociatedObject的方法,這個方法有四個參數,分別是:源對象,關聯時的用來標記是哪一個屬性的key(因為你可能要添加很多屬性),關聯的對象和一個關聯策略。

用來標記是哪一個屬性的key常見有三種寫法,但代碼效果是一樣的,如下:

//利用靜態變量地址唯一不變的特性
1、static void *strKey = &strKey;

2、static NSString *strKey = @"strKey"; 

3、static char strKey;

關聯策略是個枚舉值,解釋如下:

enum {
    OBJC_ASSOCIATION_ASSIGN = 0, //關聯對象的屬性是弱引用 

    OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, //關聯對象的屬性是強引用並且關聯對象不使用原子性

    OBJC_ASSOCIATION_COPY_NONATOMIC = 3, //關聯對象的屬性是copy並且關聯對象不使用原子性

    OBJC_ASSOCIATION_RETAIN = 01401, //關聯對象的屬性是copy並且關聯對象使用原子性

    OBJC_ASSOCIATION_COPY = 01403 //關聯對象的屬性是copy並且關聯對象使用原子性
};

4、完成后的整體代碼如下:

.h文件

//分類的頭文件
@interface ClassName (CategoryName)
@property (nonatomic, strong) NSString *str;
@end

.m文件

//實現文件
#import "ClassName + CategoryName.h"
#import <objc/runtime.h>

static void *strKey = &strKey;

@implementation ClassName (CategoryName) 
-(void)setStr:(NSString *)str  
{  
    objc_setAssociatedObject(self, & strKey, str, OBJC_ASSOCIATION_COPY);  
}  

-(NSString *)str  
{  
    return objc_getAssociatedObject(self, &strKey);  
}
@end

  


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM