一般来说,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