不管一个类设计的多么完美，在将来的需求演进中，都有可能会碰到一些没法预测的状况。那怎么扩展已有的类呢？通常而言，继承和组合是不错的选择。可是在Objective-C 2.0中，又提供了category这个语言特性，能够动态地为已有类添加新行为。现在category已经遍及于Objective-C代码的各个角落，从Apple官方的framework到各个开源框架，从功能繁复的大型APP到简单的应用，catagory无处不在。本文对category作了比较全面的整理，但愿对读者有所裨益。html
简介
本文系学习Objective-C的runtime源码时整理所成，主要剖析了category在runtime层的实现原理以及和category相关的方方面面，内容包括：java
- 初入宝地-category简介
- 连类比事-category和extension
- 挑灯细览-category真面目
- 追本溯源-category如何加载
- 旁枝末叶-category和+load方法
- 举一反三-category和方法覆盖
- 更上一层-category和关联对象
一、初入宝地-category简介
category是Objective-C 2.0以后添加的语言特性，category的主要做用是为已经存在的类添加方法。除此以外，apple还推荐了category的另外两个使用场景1ios
- 能够把类的实现分开在几个不一样的文件里面。这样作有几个显而易见的好处，a)能够减小单个文件的体积 b)能够把不一样的功能组织到不一样的category里 c)能够由多个开发者共同完成一个类 d)能够按需加载想要的category 等等。
- 声明私有方法
不过除了apple推荐的使用场景，广大开发者脑洞大开，还衍生出了category的其余几个使用场景：数组
- 模拟多继承
- 把framework的私有方法公开
Objective-C的这个语言特性对于纯动态语言来讲可能不算什么，好比javascript，你能够随时为一个“类”或者对象添加任意方法和实例变量。可是对于不是那么“动态”的语言而言，这确实是一个了不得的特性。app
二、连类比事-category和extension
extension看起来很像一个匿名的category，可是extension和有名字的category几乎彻底是两个东西。 extension在编译期决议，它就是类的一部分，在编译期和头文件里的@interface以及实现文件里的@implement一块儿造成一个完整的类，它伴随类的产生而产生，亦随之一块儿消亡。extension通常用来隐藏类的私有信息，你必须有一个类的源码才能为一个类添加extension，因此你没法为系统的类好比NSString添加extension。（详见2）框架
可是category则彻底不同，它是在运行期决议的。
就category和extension的区别来看，咱们能够推导出一个明显的事实，extension能够添加实例变量，而category是没法添加实例变量的（由于在运行期，对象的内存布局已经肯定，若是添加实例变量就会破坏类的内部布局，这对编译型语言来讲是灾难性的）。ide
三、挑灯细览-category真面目
咱们知道，全部的OC类和对象，在runtime层都是用struct表示的，category也不例外，在runtime层，category用结构体category_t（在objc-runtime-new.h中能够找到此定义），它包含了
1)、类的名字（name）
2)、类（cls）
3)、category中全部给类添加的实例方法的列表（instanceMethods）
4)、category中全部添加的类方法的列表（classMethods）
5)、category实现的全部协议的列表（protocols）
6)、category中添加的全部属性（instanceProperties）函数
typedef struct category_t { const char *name; classref_t cls; struct method_list_t *instanceMethods; struct method_list_t *classMethods; struct protocol_list_t *protocols; struct property_list_t *instanceProperties; } category_t;
从category的定义也能够看出category的可为（能够添加实例方法，类方法，甚至能够实现协议，添加属性）和不可为（没法添加实例变量）。
ok，咱们先去写一个category看一下category到底为什么物：布局
MyClass.h：
#import <Foundation/Foundation.h> @interface MyClass : NSObject - (void)printName; @end @interface MyClass(MyAddition) @property(nonatomic, copy) NSString *name; - (void)printName; @end
MyClass.m：
#import "MyClass.h" @implementation MyClass - (void)printName { NSLog(@"%@",@"MyClass"); } @end @implementation MyClass(MyAddition) - (void)printName { NSLog(@"%@",@"MyAddition"); } @end
咱们使用clang的命令去看看category到底会变成什么：
clang -rewrite-objc MyClass.m
好吧，咱们获得了一个3M大小，10w多行的.cpp文件（这绝对是Apple值得吐槽的一点），咱们忽略掉全部和咱们无关的东西，在文件的最后，咱们找到了以下代码片断：
static struct /*_method_list_t*/ { unsigned int entsize; // sizeof(struct _objc_method) unsigned int method_count; struct _objc_method method_list[1]; } _OBJC_$_CATEGORY_INSTANCE_METHODS_MyClass_$_MyAddition __attribute__ ((used, section ("__DATA,__objc_const"))) = { sizeof(_objc_method), 1, {{(struct objc_selector *)"printName", "v16@0:8", (void *)_I_MyClass_MyAddition_printName}} }; static struct /*_prop_list_t*/ { unsigned int entsize; // sizeof(struct _prop_t) unsigned int count_of_properties; struct _prop_t prop_list[1]; } _OBJC_$_PROP_LIST_MyClass_$_MyAddition __attribute__ ((used, section ("__DATA,__objc_const"))) = { sizeof(_prop_t), 1, {{"name","T@\"NSString\",C,N"}} }; extern "C" __declspec(dllexport) struct _class_t OBJC_CLASS_$_MyClass; static struct _category_t _OBJC_$_CATEGORY_MyClass_$_MyAddition __attribute__ ((used, section ("__DATA,__objc_const"))) = { "MyClass", 0, // &OBJC_CLASS_$_MyClass, (const struct _method_list_t *)&_OBJC_$_CATEGORY_INSTANCE_METHODS_MyClass_$_MyAddition, 0, 0, (const struct _prop_list_t *)&_OBJC_$_PROP_LIST_MyClass_$_MyAddition, }; static void OBJC_CATEGORY_SETUP_$_MyClass_$_MyAddition(void ) { _OBJC_$_CATEGORY_MyClass_$_MyAddition.cls = &OBJC_CLASS_$_MyClass; } #pragma section(".objc_inithooks$B", long, read, write) __declspec(allocate(".objc_inithooks$B")) static void *OBJC_CATEGORY_SETUP[] = { (void *)&OBJC_CATEGORY_SETUP_$_MyClass_$_MyAddition, }; static struct _class_t *L_OBJC_LABEL_CLASS_$ [1] __attribute__((used, section ("__DATA, __objc_classlist,regular,no_dead_strip")))= { &OBJC_CLASS_$_MyClass, }; static struct _class_t *_OBJC_LABEL_NONLAZY_CLASS_$[] = { &OBJC_CLASS_$_MyClass, }; static struct _category_t *L_OBJC_LABEL_CATEGORY_$ [1] __attribute__((used, section ("__DATA, __objc_catlist,regular,no_dead_strip")))= { &_OBJC_$_CATEGORY_MyClass_$_MyAddition, };
咱们能够看到，
1)、首先编译器生成了实例方法列表OBJC$_CATEGORY_INSTANCE_METHODSMyClass$_MyAddition和属性列表OBJC$_PROP_LISTMyClass$_MyAddition，二者的命名都遵循了公共前缀+类名+category名字的命名方式，并且实例方法列表里面填充的正是咱们在MyAddition这个category里面写的方法printName，而属性列表里面填充的也正是咱们在MyAddition里添加的name属性。还有一个须要注意到的事实就是category的名字用来给各类列表以及后面的category结构体自己命名，并且有static来修饰，因此在同一个编译单元里咱们的category名不能重复，不然会出现编译错误。
2)、其次，编译器生成了category自己OBJC$_CATEGORYMyClass$_MyAddition，并用前面生成的列表来初始化category自己。
3)、最后，编译器在DATA段下的objc_catlist section里保存了一个大小为1的category_t的数组L_OBJC_LABELCATEGORY$（固然，若是有多个category，会生成对应长度的数组^_^），用于运行期category的加载。
到这里，编译器的工做就接近尾声了，对于category在运行期怎么加载，咱们下节揭晓。
四、追本溯源-category如何加载
咱们知道，Objective-C的运行是依赖OC的runtime的，而OC的runtime和其余系统库同样，是OS X和iOS经过dyld动态加载的。
想了解更多dyld地同窗能够移步这里（3）。
对于OC运行时，入口方法以下（在objc-os.mm文件中）：
void _objc_init(void) { static bool initialized = false; if (initialized) return; initialized = true; // fixme defer initialization until an objc-using image is found? environ_init(); tls_init(); lock_init(); exception_init(); // Register for unmap first, in case some +load unmaps something _dyld_register_func_for_remove_image(&unmap_image); dyld_register_image_state_change_handler(dyld_image_state_bound, 1/*batch*/, &map_images); dyld_register_image_state_change_handler(dyld_image_state_dependents_initialized, 0/*not batch*/, &load_images); }
category被附加到类上面是在map_images的时候发生的，在new-ABI的标准下，_objc_init里面的调用的map_images最终会调用objc-runtime-new.mm里面的_read_images方法，而在_read_images方法的结尾，有如下的代码片断：
// Discover categories. for (EACH_HEADER) { category_t **catlist = _getObjc2CategoryList(hi, &count); for (i = 0; i < count; i++) { category_t *cat = catlist[i]; class_t *cls = remapClass(cat->cls); if (!cls) { // Category's target class is missing (probably weak-linked). // Disavow any knowledge of this category. catlist[i] = NULL; if (PrintConnecting) { _objc_inform("CLASS: IGNORING category \?\?\?(%s) %p with " "missing weak-linked target class", cat->name, cat); } continue; } // Process this category. // First, register the category with its target class. // Then, rebuild the class's method lists (etc) if // the class is realized. BOOL classExists = NO; if (cat->instanceMethods || cat->protocols || cat->instanceProperties) { addUnattachedCategoryForClass(cat, cls, hi); if (isRealized(cls)) { remethodizeClass(cls); classExists = YES; } if (PrintConnecting) { _objc_inform("CLASS: found category -%s(%s) %s", getName(cls), cat->name, classExists ? "on existing class" : ""); } } if (cat->classMethods || cat->protocols /* || cat->classProperties */) { addUnattachedCategoryForClass(cat, cls->isa, hi); if (isRealized(cls->isa)) { remethodizeClass(cls->isa); } if (PrintConnecting) { _objc_inform("CLASS: found category +%s(%s)", getName(cls), cat->name); } } } }
首先，咱们拿到的catlist就是上节中讲到的编译器为咱们准备的category_t数组，关因而如何加载catlist自己的，咱们暂且不表，这和category自己的关系也不大，有兴趣的同窗能够去研究如下Apple的二进制格式和load机制。
略去PrintConnecting这个用于log的东西，这段代码很容易理解：
1)、把category的实例方法、协议以及属性添加到类上
2)、把category的类方法和协议添加到类的metaclass上
值得注意的是，在代码中有一小段注释 / || cat->classProperties /，看来苹果有过给类添加属性的计划啊。
ok，咱们接着往里看，category的各类列表是怎么最终添加到类上的，就拿实例方法列表来讲吧：
在上述的代码片断里，addUnattachedCategoryForClass只是把类和category作一个关联映射，而remethodizeClass才是真正去处理添加事宜的功臣。
static void remethodizeClass(class_t *cls) { category_list *cats; BOOL isMeta; rwlock_assert_writing(&runtimeLock); isMeta = isMetaClass(cls); // Re-methodizing: check for more categories if ((cats = unattachedCategoriesForClass(cls))) { chained_property_list *newproperties; const protocol_list_t **newprotos; if (PrintConnecting) { _objc_inform("CLASS: attaching categories to class '%s' %s", getName(cls), isMeta ? "(meta)" : ""); } // Update methods, properties, protocols BOOL vtableAffected = NO; attachCategoryMethods(cls, cats, &vtableAffected); newproperties = buildPropertyList(NULL, cats, isMeta); if (newproperties) { newproperties->next = cls->data()->properties; cls->data()->properties = newproperties; } newprotos = buildProtocolList(cats, NULL, cls->data()->protocols); if (cls->data()->protocols && cls->data()->protocols != newprotos) { _free_internal(cls->data()->protocols); } cls->data()->protocols = newprotos; _free_internal(cats); // Update method caches and vtables flushCaches(cls); if (vtableAffected) flushVtables(cls); } }
而对于添加类的实例方法而言，又会去调用attachCategoryMethods这个方法，咱们去看下attachCategoryMethods：
static void attachCategoryMethods(class_t *cls, category_list *cats, BOOL *inoutVtablesAffected) { if (!cats) return; if (PrintReplacedMethods) printReplacements(cls, cats); BOOL isMeta = isMetaClass(cls); method_list_t **mlists = (method_list_t **) _malloc_internal(cats->count * sizeof(*mlists)); // Count backwards through cats to get newest categories first int mcount = 0; int i = cats->count; BOOL fromBundle = NO; while (i--) { method_list_t *mlist = cat_method_list(cats->list[i].cat, isMeta); if (mlist) { mlists[mcount++] = mlist; fromBundle |= cats->list[i].fromBundle; } } attachMethodLists(cls, mlists, mcount, NO, fromBundle, inoutVtablesAffected); _free_internal(mlists); }
attachCategoryMethods作的工做相对比较简单，它只是把全部category的实例方法列表拼成了一个大的实例方法列表，而后转交给了attachMethodLists方法（我发誓，这是本节咱们看的最后一段代码了^_^），这个方法有点长，咱们只看一小段：
for (uint32_t m = 0; (scanForCustomRR || scanForCustomAWZ) && m < mlist->count; m++) { SEL sel = method_list_nth(mlist, m)->name; if (scanForCustomRR && isRRSelector(sel)) { cls->setHasCustomRR(); scanForCustomRR = false; } else if (scanForCustomAWZ && isAWZSelector(sel)) { cls->setHasCustomAWZ(); scanForCustomAWZ = false; } } // Fill method list array newLists[newCount++] = mlist; . . . // Copy old methods to the method list array for (i = 0; i < oldCount; i++) { newLists[newCount++] = oldLists[i]; }
须要注意的有两点：
1)、category的方法没有“彻底替换掉”原来类已经有的方法，也就是说若是category和原来类都有methodA，那么category附加完成以后，类的方法列表里会有两个methodA
2)、category的方法被放到了新方法列表的前面，而原来类的方法被放到了新方法列表的后面，这也就是咱们日常所说的category的方法会“覆盖”掉原来类的同名方法，这是由于运行时在查找方法的时候是顺着方法列表的顺序查找的，它只要一找到对应名字的方法，就会罢休^_^，却不知后面可能还有同样名字的方法。
五、旁枝末叶-category和+load方法
咱们知道，在类和category中均可以有+load方法，那么有两个问题：
1)、在类的+load方法调用的时候，咱们能够调用category中声明的方法么？
2)、这么些个+load方法，调用顺序是咋样的呢？
鉴于上述几节咱们看的代码太多了，对于这两个问题咱们先来看一点直观的：
咱们的代码里有MyClass和MyClass的两个category （Category1和Category2），MyClass和两个category都添加了+load方法，而且Category1和Category2都写了MyClass的printName方法。
在Xcode中点击Edit Scheme，添加以下两个环境变量（能够在执行load方法以及加载category的时候打印log信息，更多的环境变量选项可参见objc-private.h）:
运行项目，咱们会看到控制台打印不少东西出来，咱们只找到咱们想要的信息，顺序以下：
objc[1187]: REPLACED: -[MyClass printName] by category Category1
objc[1187]: REPLACED: -[MyClass printName] by category Category2
.
.
.
objc[1187]: LOAD: class 'MyClass' scheduled for +load
objc[1187]: LOAD: category 'MyClass(Category1)' scheduled for +load
objc[1187]: LOAD: category 'MyClass(Category2)' scheduled for +load
objc[1187]: LOAD: +[MyClass load]
.
.
.
objc[1187]: LOAD: +[MyClass(Category1) load]
.
.
.
objc[1187]: LOAD: +[MyClass(Category2) load]
因此，对于上面两个问题，答案是很明显的：
1)、能够调用，由于附加category到类的工做会先于+load方法的执行
2)、+load的执行顺序是先类，后category，而category的+load执行顺序是根据编译顺序决定的。
目前的编译顺序是这样的：
咱们调整一个Category1和Category2的编译顺序，run。ok，咱们能够看到控制台的输出顺序变了：
objc[1187]: REPLACED: -[MyClass printName] by category Category2
objc[1187]: REPLACED: -[MyClass printName] by category Category1
.
.
.
objc[1187]: LOAD: class 'MyClass' scheduled for +load
objc[1187]: LOAD: category 'MyClass(Category2)' scheduled for +load
objc[1187]: LOAD: category 'MyClass(Category1)' scheduled for +load
objc[1187]: LOAD: +[MyClass load]
.
.
.
objc[1187]: LOAD: +[MyClass(Category2) load]
.
.
.
objc[1187]: LOAD: +[MyClass(Category1) load]
虽然对于+load的执行顺序是这样，可是对于“覆盖”掉的方法，则会先找到最后一个编译的category里的对应方法。
这一节咱们只是用很直观的方式获得了问题的答案，有兴趣的同窗能够继续去研究一下OC的运行时代码。
六、举一反三-category和方法覆盖
鉴于上面几节咱们已经把原理都讲了，这一节只有一个问题:
怎么调用到原来类中被category覆盖掉的方法？
对于这个问题，咱们已经知道category其实并非彻底替换掉原来类的同名方法，只是category在方法列表的前面而已，因此咱们只要顺着方法列表找到最后一个对应名字的方法，就能够调用原来类的方法：
Class currentClass = [MyClass class]; MyClass *my = [[MyClass alloc] init]; if (currentClass) { unsigned int methodCount; Method *methodList = class_copyMethodList(currentClass, &methodCount); IMP lastImp = NULL; SEL lastSel = NULL; for (NSInteger i = 0; i < methodCount; i++) { Method method = methodList[i]; NSString *methodName = [NSString stringWithCString:sel_getName(method_getName(method)) encoding:NSUTF8StringEncoding]; if ([@"printName" isEqualToString:methodName]) { lastImp = method_getImplementation(method); lastSel = method_getName(method); } } typedef void (*fn)(id,SEL); if (lastImp != NULL) { fn f = (fn)lastImp; f(my,lastSel); } free(methodList); }
七、更上一层-category和关联对象
如上所见，咱们知道在category里面是没法为category添加实例变量的。可是咱们不少时候须要在category中添加和对象关联的值，这个时候能够求助关联对象来实现。
MyClass+Category1.h:
#import "MyClass.h" @interface MyClass (Category1) @property(nonatomic,copy) NSString *name; @end
MyClass+Category1.m:
#import "MyClass+Category1.h" #import <objc/runtime.h> @implementation MyClass (Category1) + (void)load { NSLog(@"%@",@"load in Category1"); } - (void)setName:(NSString *)name { objc_setAssociatedObject(self, "name", name, OBJC_ASSOCIATION_COPY); } - (NSString*)name { NSString *nameObject = objc_getAssociatedObject(self, "name"); return nameObject; } @end
可是关联对象又是存在什么地方呢？ 如何存储？ 对象销毁时候如何处理关联对象呢？
咱们去翻一下runtime的源码，在objc-references.mm文件中有个方法_object_set_associative_reference：
void _object_set_associative_reference(id object, void *key, id value, uintptr_t policy) { // retain the new value (if any) outside the lock. ObjcAssociation old_association(0, nil); id new_value = value ? acquireValue(value, policy) : nil; { AssociationsManager manager; AssociationsHashMap &associations(manager.associations()); disguised_ptr_t disguised_object = DISGUISE(object); if (new_value) { // break any existing association. AssociationsHashMap::iterator i = associations.find(disguised_object); if (i != associations.end()) { // secondary table exists ObjectAssociationMap *refs = i->second; ObjectAssociationMap::iterator j = refs->find(key); if (j != refs->end()) { old_association = j->second; j->second = ObjcAssociation(policy, new_value); } else { (*refs)[key] = ObjcAssociation(policy, new_value); } } else { // create the new association (first time). ObjectAssociationMap *refs = new ObjectAssociationMap; associations[disguised_object] = refs; (*refs)[key] = ObjcAssociation(policy, new_value); _class_setInstancesHaveAssociatedObjects(_object_getClass(object)); } } else { // setting the association to nil breaks the association. AssociationsHashMap::iterator i = associations.find(disguised_object); if (i != associations.end()) { ObjectAssociationMap *refs = i->second; ObjectAssociationMap::iterator j = refs->find(key); if (j != refs->end()) { old_association = j->second; refs->erase(j); } } } } // release the old value (outside of the lock). if (old_association.hasValue()) ReleaseValue()(old_association); }
咱们能够看到全部的关联对象都由AssociationsManager管理，而AssociationsManager定义以下：
class AssociationsManager { static OSSpinLock _lock; static AssociationsHashMap *_map; // associative references: object pointer -> PtrPtrHashMap. public: AssociationsManager() { OSSpinLockLock(&_lock); } ~AssociationsManager() { OSSpinLockUnlock(&_lock); } AssociationsHashMap &associations() { if (_map == NULL) _map = new AssociationsHashMap(); return *_map; } };
AssociationsManager里面是由一个静态AssociationsHashMap来存储全部的关联对象的。这至关于把全部对象的关联对象都存在一个全局map里面。而map的的key是这个对象的指针地址（任意两个不一样对象的指针地址必定是不一样的），而这个map的value又是另一个AssociationsHashMap，里面保存了关联对象的kv对。
而在对象的销毁逻辑里面，见objc-runtime-new.mm:
void *objc_destructInstance(id obj) { if (obj) { Class isa_gen = _object_getClass(obj); class_t *isa = newcls(isa_gen); // Read all of the flags at once for performance. bool cxx = hasCxxStructors(isa); bool assoc = !UseGC && _class_instancesHaveAssociatedObjects(isa_gen); // This order is important. if (cxx) object_cxxDestruct(obj); if (assoc) _object_remove_assocations(obj); if (!UseGC) objc_clear_deallocating(obj); } return obj; }
嗯，runtime的销毁对象函数objc_destructInstance里面会判断这个对象有没有关联对象，若是有，会调用_object_remove_assocations作关联对象的清理工做。