利用类别(Category)为对象添加”实例变量”

最新推荐文章于 2021-01-07 23:09:56 发布

转载最新推荐文章于 2021-01-07 23:09:56 发布 · 360 阅读

iOS初级专栏收录该内容

30 篇文章

订阅专栏

本文介绍如何使用Objective-C的CAPI函数进行属性和方法的扩展，通过引入runtime、定义类别并实现getter和setter方法，实现对象的自定义属性管理。通过实例演示了在UIView类上扩展属性viewName，并在viewDidLoad方法中测试该属性的使用。

属性扩展用objc_setAssociatedObject，objc_getAssociatedObject，

方法扩展用category

主要利用ObjectiveC的C API中函数:

void objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
id objc_getAssociatedObject(id object, const void *key)

原理详细参见官方的ObjectiveC Runtime Reference.

好了,现在直接写怎么做了.

1.在定义类别之前, 引入runtime.

#import <objc/runtime.h>

2.定义&实现类别(以UIView为例)
定义:

@interface UIView(AddVariables)

@property (nonatomic, retain) NSString *viewName;

@end

实现:

// 定义存取的Key
static const char *viewNameKey = "viewNameKey";

@implementation UIView(AddVariables)

// get方法
- (NSString *)viewName {
    return (NSString *)objc_getAssociatedObject(self, viewNameKey);
}
// set方法
- (void)setViewName:(NSString *)newViewNameKey {
    objc_setAssociatedObject(self, viewNameKey, newViewNameKey, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

@end

3.调用测试:

- (void)viewDidLoad {

    UIView *testView = [[UIView alloc] initWithFrame:self.view.bounds];
    [self.view addSubview:testView];
    [testView release];

    [testView setViewName:@"Nori's View"];
    NSLog(@"[testView's newPropery]:%@", testView.viewName);

    [super viewDidLoad];
}

输出结果如下: