OC vs Swift -- Extension

扩展的作用

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