给定的答案效果很好,我的建议只是对它的扩展,以避免编写过多的样板代码。
为了避免为类别属性重复编写 getter 和 setter 方法,此答案引入了宏。此外,这些宏简化了原始类型属性的使用,例如 int 或 BOOL。
没有宏的传统方法
传统上,您定义一个类别属性,如
@interface MyClass (Category)
@property (strong, nonatomic) NSString *text;
@end
那么你需要实现一个getter和setter方法,使用一个关联对象和get选择器作为键(see original answer):
#import <objc/runtime.h>
@implementation MyClass (Category)
- (NSString *)text{
return objc_getAssociatedObject(self, @selector(text));
}
- (void)setText:(NSString *)text{
objc_setAssociatedObject(self, @selector(text), text, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
@end
我的建议方法
现在,您将改用宏来编写:
@implementation MyClass (Category)
CATEGORY_PROPERTY_GET_SET(NSString*, text, setText:)
@end
宏定义如下:
#import <objc/runtime.h>
#define CATEGORY_PROPERTY_GET(type, property) - (type) property { return objc_getAssociatedObject(self, @selector(property)); }
#define CATEGORY_PROPERTY_SET(type, property, setter) - (void) setter (type) property { objc_setAssociatedObject(self, @selector(property), property, OBJC_ASSOCIATION_RETAIN_NONATOMIC); }
#define CATEGORY_PROPERTY_GET_SET(type, property, setter) CATEGORY_PROPERTY_GET(type, property) CATEGORY_PROPERTY_SET(type, property, setter)
#define CATEGORY_PROPERTY_GET_NSNUMBER_PRIMITIVE(type, property, valueSelector) - (type) property { return [objc_getAssociatedObject(self, @selector(property)) valueSelector]; }
#define CATEGORY_PROPERTY_SET_NSNUMBER_PRIMITIVE(type, property, setter, numberSelector) - (void) setter (type) property { objc_setAssociatedObject(self, @selector(property), [NSNumber numberSelector: property], OBJC_ASSOCIATION_RETAIN_NONATOMIC); }
#define CATEGORY_PROPERTY_GET_UINT(property) CATEGORY_PROPERTY_GET_NSNUMBER_PRIMITIVE(unsigned int, property, unsignedIntValue)
#define CATEGORY_PROPERTY_SET_UINT(property, setter) CATEGORY_PROPERTY_SET_NSNUMBER_PRIMITIVE(unsigned int, property, setter, numberWithUnsignedInt)
#define CATEGORY_PROPERTY_GET_SET_UINT(property, setter) CATEGORY_PROPERTY_GET_UINT(property) CATEGORY_PROPERTY_SET_UINT(property, setter)
宏CATEGORY_PROPERTY_GET_SET 为给定属性添加了getter 和setter。只读或只写属性将分别使用CATEGORY_PROPERTY_GET 和CATEGORY_PROPERTY_SET 宏。
原始类型需要更多关注
由于原始类型不是对象,上述宏包含一个使用unsigned int 作为属性类型的示例。它通过将整数值包装到 NSNumber 对象中来实现。所以它的用法类似于前面的例子:
@interface ...
@property unsigned int value;
@end
@implementation ...
CATEGORY_PROPERTY_GET_SET_UINT(value, setValue:)
@end
按照这种模式,您可以简单地添加更多宏来支持signed int、BOOL等...
限制
所有宏默认使用OBJC_ASSOCIATION_RETAIN_NONATOMIC。
像 App Code 这样的 IDE 目前在重构属性名称时无法识别 setter 的名称。您需要自己重命名。