社区首页 >专栏 >iOS中OC给Category添加属性

iOS中OC给Category添加属性

Cloudox

发布于 2021-11-23 05:22:03

1.6K00

代码可运行

文章被收录于专栏：月亮与二进制月亮与二进制

运行总次数：0

代码可运行

引：

很多人知道可以用Category给已有的类添加一些新方法，但是不同于swift中的extension，Objective-C中的Category（类别）是不支持直接添加属性的，那如果就是需要添加新的属性怎么办呢？有一个办法，使用Runtime中的关联对象技术就可以实现。关于Runtime的说明可以查看这篇博客：传送门：OC中Runtime浅析

Category（类别）：

对于不熟悉Category的人，这里也先说明一下Category是什么以及怎么创建Category。

由于一些特殊的需要，我们可能要给现有的类添加一些新的方法，这个需求用继承也可以做到，但是会显得比较重，这时候就可以用Category来达到目的，创建一个已有类的Category，可以给这个类添加你需要的方法，在使用的时候，只需要import你创建的Category，在使用的时候还是使用原来的类，但是你会惊奇的发现他支持你自己在Category中添加的方法。

我们看到的一些名为类似“UINavigationController+Cloudox.h”的文件就是类别了。

那么怎么创建类别呢？

在工程中按住command+N来添加新文件，选择Objectiv-C File：

在出来的界面中的File Type选择Category，就是创建类别文件了，File填写我们要加在类别尾巴上的名字，Class选择你要添加类别的已有类，这里我们为UINavigationController添加类别：

类别文件这样就创建成功了。

添加属性

类别可以为已有的类添加方法，但是却不能直接添加属性，因为即使你添加了@property，它既不会生成实例变量，也不会生成setter、getter方法，即使你添加了也无法使用。

所以我们首先需要自己去添加setter、getter方法，这个好办，直接在.m文件里加就可以了，但是要真正添加可以使用的属性，还需要利用Runtime来关联对象，关于关联对象的技术可以看传送门：OC中Runtime浅析，这里只讲怎么用来添加属性，我们在setter方法里关联一个对象，在getter方法里获取对应key关联的对象，就可以啦，代码如下，很简单：

//UINavigationController+Cloudox.h文件
#import <UIKit/UIKit.h>

@interface UINavigationController (Cloudox)

@property (copy, nonatomic) NSString *cloudox;

@end

//UINavigationController+Cloudox.m文件
#import "UINavigationController+Cloudox.h"
#import <objc/runtime.h>

@implementation UINavigationController (Cloudox)

//定义常量 必须是C语言字符串
static char *CloudoxKey = "CloudoxKey";

-(void)setCloudox:(NSString *)cloudox{
    /*
     objc_AssociationPolicy参数使用的策略：
     OBJC_ASSOCIATION_ASSIGN;            //assign策略
     OBJC_ASSOCIATION_COPY_NONATOMIC;    //copy策略
     OBJC_ASSOCIATION_RETAIN_NONATOMIC;  // retain策略
     
     OBJC_ASSOCIATION_RETAIN;
     OBJC_ASSOCIATION_COPY;
     */
    /*
     关联方法：
     objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy);
     
     参数：
     * id object 给哪个对象的属性赋值
     const void *key 属性对应的key
     id value  设置属性值为value
     objc_AssociationPolicy policy  使用的策略，是一个枚举值，和copy，retain，assign是一样的，手机开发一般都选择NONATOMIC
     */
    
    objc_setAssociatedObject(self, CloudoxKey, cloudox, OBJC_ASSOCIATION_COPY_NONATOMIC);
}

-(NSString *)cloudox{
    return objc_getAssociatedObject(self, CloudoxKey);
}


@end

注意要使用Runtime需要 import runtime 的框架，如代码所示。

现在我们试试效果，我们按照平常的方式用UINavigationController包装一个控制器作为根视图：

// AppDelegate
- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions {
    ViewController *vc = [[ViewController alloc] init];
    UINavigationController *nav = [[UINavigationController alloc] initWithRootViewController:vc];
    self.window.rootViewController = nav;
    return YES;
}

这个视图控制器中我们需要导入我们的类别文件，然后我们就会发现我们能够通过self.navigationController获取到我们新添加的名为“cloudox”的属性了！我们可以给这个NSString类型的属性赋值，然后获取它进行显示：

#import "UINavigationController+Cloudox.h"

- (void)viewDidLoad {
    [super viewDidLoad];
    self.title = @"Demo";
    self.view.backgroundColor = [UIColor lightGrayColor];
    
    // 给UINavigationController新加的属性赋值
    self.navigationController.cloudox = @"Hey,this is category's new property!";
    
    UILabel *label = [[UILabel alloc] initWithFrame:CGRectMake(([UIScreen mainScreen].bounds.size.width-300)/2, 100, 300, 50)];
    label.textColor = [UIColor whiteColor];
    label.textAlignment = NSTextAlignmentCenter;
    label.text = self.navigationController.cloudox;
    [self.view addSubview:label];
}