项目中我们非常多的使用类扩展,即 Category 类别,我们可以在类别中扩展很多需要的方法。 那如果需要扩展属性呢? 如果按照正常的给类添加 @property 属性,然后实现 set/get 方法,则会出现报错。 这里直接上正确的添加方式。
首先创建一个 RuntimeTest 的 Demo, 然后在 Demo 中创建一个 NSObject 的扩展类 NSObject+Extensions ,我们将在这个类别中动态添加属性。 然后 将这个类添加到 .pch 中,这样一来所有继承 NSObject 的类,都拥有动态添加的属性。 具体类目中的代码:
#import <Foundation/Foundation.h>
@interface NSObject (Extensions)
@property (nonatomic, copy) NSString *stringProperty; // 扩展字符串属性
@property (nonatomic, assign) NSInteger integerProperty; // 扩展整形属性
@property (nonatomic, assign) BOOL boolProperty; // 扩展BOOL属性
@end
#import "NSObject+Extensions.h"
#import <objc/runtime.h>
// 扩展属性 对应的地址值, 保证 set/get 方法内使用的 地址完全一样。
static const void *stringPropertyKey = &stringPropertyKey;
static const void *integerPropertyKey = &integerPropertyKey;
static const void *boolPropertyKey = &boolPropertyKey;
@implementation NSObject (Extensions)
- (void)setStringProperty:(NSString *)stringProperty {
/**
* 根据某个对象,还有key,还有对应的策略(copy,strong等) 动态的将值设置到这个对象的key上
*
* @param object 某个对象
* @param key 属性名,根据key去获取关联的对象, 是一个字符串常亮,是一个地址(这里注意,地址必须是不变的,地址不同但是内容相同的也不算同一个key)
* @param value 要设置的值
* @param policy 策略(copy,strong,assign等)
*/
objc_setAssociatedObject(self, stringPropertyKey, stringProperty, OBJC_ASSOCIATION_COPY_NONATOMIC);
}
- (NSString *)stringProperty {
return objc_getAssociatedObject(self, stringPropertyKey);
}
- (void)setIntegerProperty:(NSInteger)integerProperty {
// 写法一
NSNumber *number = [[NSNumber alloc] initWithInteger:integerProperty];
objc_setAssociatedObject(self, integerPropertyKey, number, OBJC_ASSOCIATION_ASSIGN);
// 写法二 (NSNumber 的创建简写)
// objc_setAssociatedObject(self, integerPropertyKey, @(integerProperty), OBJC_ASSOCIATION_ASSIGN);
}
- (NSInteger)integerProperty {
return [objc_getAssociatedObject(self, integerPropertyKey) integerValue];
}
- (void)setBoolProperty:(BOOL)boolProperty {
objc_setAssociatedObject(self, boolPropertyKey, @(boolProperty), OBJC_ASSOCIATION_ASSIGN);
}
- (BOOL)boolProperty {
return [objc_getAssociatedObject(self, boolPropertyKey) boolValue];
}
@end
主要的原理是,在 Set 方法中使用 objc_setAssociatedObject() 函数 , Get 方法中使用 objc_getAssociatedObject() 函数。 需要注意的是 objc_setAssociatedObject(object ,key,value,policy) 函数中,第二个参数 key 和 objc_getAssociatedObject(object ,key) 函数中 第二个参数 key 地址一定要相同才行。 具体代码中也有相应的注释。 类目实现完成之后,可以做一下测试。
在 ViewController 中添加 跳转到测试页面 PageViewController, 在创建 PageViewController 时,使用动态属性,并给予赋值。
#import "ViewController.h"
#import "PageViewController.h"
@interface ViewController ()
@end
@implementation ViewController
- (void)viewDidLoad {
[super viewDidLoad];
self.navigationItem.title = @"Home";
}
- (IBAction)gotoPageVC:(UIButton *)sender {
PageViewController *vc = [[PageViewController alloc] init];
vc.stringProperty = @"PageViewController";
vc.integerProperty = 333333;
vc.boolProperty = NO;
[self.navigationController pushViewController:vc animated:YES];
}
@end
然后再在 PageViewController.m 中, viewDidLoad 实现输出这几个动态的参数值,并且 额外做一组测试,创建 NSArray 数组 array, 也使用动态属性赋值,并实时输出进行测试。
#import "PageViewController.h"
@interface PageViewController ()
@end
@implementation PageViewController
- (void)viewDidLoad {
[super viewDidLoad];
self.navigationItem.title = @"PageViewController";
self.view.backgroundColor = [UIColor whiteColor];
// UIViewController 使用的扩展属性
NSLog(@"stringProperty: %@", self.stringProperty);
NSLog(@"integerProperty: %ld", self.integerProperty);
NSLog(@"boolProperty: %d", self.boolProperty);
// NSArray 使用扩展属性
NSArray *array = [NSArray new];
array.stringProperty = @"array";
array.integerProperty = 11111;
array.boolProperty = YES;
NSLog(@"array-stringProperty: %@", array.stringProperty);
NSLog(@"array-integerProperty: %ld", array.integerProperty);
NSLog(@"array-boolProperty: %d", array.boolProperty);
}
@end
最后输出的结果:
测试Demo下载