runtime 为 category 添加成员变量的应用

在 Objective-C 中,我们可以为一个 第三方类,或者 Cocoa 原生类添加 Category的方式,来扩展 自己的行为。

比如,我们可以给NSString 添加 自己的 方法,来实现自己的行为。

比如 我们可以加上 NSString(Md5)的 CateGory , 来实现NSString 对 md5的 扩展。

但是,我们经常会有需要添加一些 成员变量的情况,比如我想给NSString 都加上一个资源 ID , 用来标记唯一的字符串资源呢?

很遗憾,Category 并不支持对 成员变量的扩展。

这个时候,我们就可以使用 runtime 机制来 扩展成员变量了。

具体代码如下:

#import <Foundation/Foundation.h>

@interface NSString (Addtional)

@property(nonatomic, retain) NSNumber * resId;

@end
@implementation NSString (Addtional)

-(NSNumber *) resId {
    return objc_getAssociatedObject(self, "resId");
}

-(void) setResId:(NSNumber *)resId {
    objc_setAssociatedObject(self, "resId", resId, OBJC_ASSOCIATION_RETAIN);
}

@end

中间设计到 两个 runtime 函数, 一个是 objc_setAssociatedObject ,  用来给 一个 class 添加 附属的数据, 另外一个 是 objc_getAssociatedObject , 用来获取附加的数据,从而实现对原生类的扩展。

我们看看使用例子:

    NSString * strName = @"Dog XiaoHua";
    strName.resId = @3;
    
    NSLog(@"Dog's resId = %@", strName.resId);

 输出结果为 3, 就是我们设置的值。

是不是很神奇,这其中的原理是什么呢,这就涉及到了 oc 的runtime 机制了, 且期待下篇文章分解。

详见 oc 消息机制探索

猜你喜欢

转载自blog.csdn.net/tangaowen/article/details/88046109