扩展的作用
OC 和Swift中都有扩展,通过扩展可以:
- 为已有的类型添加新的功能,对不可以访问的源码提供可扩展的能力;
- 在不用subclass的情况下可以修复一些bug;
- 让类的接口和实现分离到多个文件中,可以模块化较大的工程,为维护大型代码库的减轻负担。
OC中的扩展####
OC中扩展分为Category(分类)和Extension(扩展)两种。
Category######
Category 通常用于添加 instance methods(实例方法)和class methods(类方法),通常不适用于添加properties(属性)。
但当我们想要给我们不能获取到的源码中类添加属性时,我们可以通过分类来实现,分类不适用于添加属性,其实是不会对添加的属性自动生成getter和setter方法,所以只要我们在分类中实现getter和setter方法就可以了。
实现这两个方法我们需要利用rumtime的关联值的特性。通过objc_setAssociatedObject
和objc_getAssociatedObject
来完成。
//Ship+Property.h
#import "Ship.h"
@interface Ship(Property)
@property (nonatomic, copy) NSString *name;
@end
//Ship+Property.m
#import "Ship+Property.h"
#import <objc/runtime.h>
@implementation Ship(Property)
-(void)setName: (NSString *)name {
objc_setAssociatedObject(self, @selector(name), name, OBJC_ASSOCIATION_COPY);
}
-(NSString *)name{
return objc_getAssociatedObject(self, @selector(name));
}
@end
⚠️ 分类中定义的方法名不要与原类或者其他分类中方法名相同,这样程序在运行时将不知道去访问哪一个方法。为了避免这种命名冲突,可以在分类方法名前为其添加前缀。
Extension######
- Extension是一种特殊的分类,它没有名字,并且只能在类的实现文件(.m)中实现。通常用于添加private methods (私有方法)和properties。
Swift中的扩展####
Swift中Extension 和OC中的分类相似,不同的是extension是没有名字的,并且可以对class、struct、enum、protocol进行扩展 。
Swift的Extension中可以:
- 定义实例方法和类型方法
- 提供初始化方法 // OC 中也可以
- 定义subscripts方法
- 定义和使用嵌套类型
- 可以让已有类型去实现某个协议 // OC 中也可以
- 可以添加计算属性和静态计算属性
⚠️扩展可以添加新的计算属性,但是不可以添加存储属性,也不可以向已有属性添加属性观测器。
特别是通过Extension还可以为protocol中定义的方法添加默认实现。
protocol PersonProtocol {
func say(something: String)
}
extension PersonProtocol {
func say(something: String) {
print("default say: \(something)")
}
}
class Person: PersonProtocol {}
let tom = Person()
tom.say("hello")
//default say: hello
//after we extension Person to implement method say
extension Person {
func say(something: String) {
print("override say: \(something)")
}
}
let tom = Person()
tom.say("hello")
//override say: hello