巧用 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下载