巧用 Class Extension 隐藏属性

一般来说,Extension用来给Class增加私有属性和方法,写在 Class 的.m文件。但是Extension不是必须要写在.m文件,你可以写在任何地方,只要在 @implementation 前定义就可以。所以我们可以利用这个特性实现属性隐藏。

Case: 模块内部需要访问某属性,模块外需隐藏。这种情况经常会遇到。例如对于某一公司来讲,每个员工都需要员工ID来唯一标识,但是员工作为自然人,在其他地方,别人不需要知道这个ID。所以对于该员工的ID,在公司内部需要访问,在外部需要隐藏。

定义一个company对象,一个person对象。

@interface Person : NSObject

@property (nonatomic, strong) NSString *name;

@end
@interface Company : NSObject

@property (nonatomic, strong) NSString *name;

- (void)addStaff:(Person *)person;

- (void)printAllStaffDisplayName;

@end

公司可以增加一个自然人当自己的员工。可以打印全部员工ID,

@interface Company ()

@property (nonatomic, strong) NSMutableArray<Person *> *staffs;

@end

@implementation Company

- (void)addStaff:(Person *)person {
    person.companyID = [NSString stringWithFormat:@"大汉%ld",self.staffs.count];
    [self.staffs addObject:person];
}

- (void)printAllStaffDisplayName {
    for (Person *staff in self.staffs) {
        NSLog(@"name:%@ displayName:%@ \n",staff.name,staff.companyID);
    }
}

- (NSMutableArray<Person *> *)staffs {
    if (!_staffs) {
        _staffs = [NSMutableArray arrayWithCapacity:0];
    }
    return _staffs;
}

@end

可以看到在 Company 中需要访问 Person 的 companyID,所以我们给 Person 增加一个 Extention。写在 Company.h。

@interface Company : NSObject

@property (nonatomic, strong) NSString *name;

- (void)addStaff:(Person *)person;

- (void)printAllStaffCompanyID;

@end

@interface Person ()

@property (nonatomic, strong) NSString *companyID;

@end

这样我们就可以在模块内部增加了一个public属性,而模块外部不知道有这个属性。

    Company *cisco = [Company new];
    cisco.name = @"cisco";
    
    Person *tao = [Person new];
    tao.name = @"涛";
    
    [cisco addStaff:tao];
    [cisco printAllStaffDisplayName];

关于这个特性,可以参考sunny的文章:http://blog.sunnyxx.com/2016/04/22/objc-class-extension-tips/。但是,当你运行的时候会发现,crash了!log显示找不到setCompanyID方法,原来我们把extention写在非.m文件的时候,oc不会帮我们自动生成set,get方法。我想手动写一个set方法来解决这个问题,发现如果想在 @implenmatation 写,就必须知道他的实例变量 _companyID,我们现在拿不到。联想到readOnly属性可以使用 extention 在.m中修改为readWrite:

.h

@interface Person : NSObject

@property (readonly, nonatomic, strong) NSString *gender;

@end

.m

@interface Person ()

@property (nonatomic, strong) NSString *gender;

@end

@implementation Person


@end

那我们也可以在 Person.m 中再次申明一个companyID的私有属性。这样,oc会帮我们自动生成set,get方法。试验一下,完美!

@interface Person ()

@property (nonatomic, strong) NSString *companyID;

@end

@implementation Person


@end

Demo下载

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券