該文章屬于劉小壯原創(chuàng)太防，轉(zhuǎn)載請注明：劉小壯

Category

有了之前Runtime的基礎(chǔ)拷窜，一些內(nèi)部實(shí)現(xiàn)就很好理解了。在OC中可以通過Category添加屬性氢架、方法、協(xié)議，在Runtime中Class和Category都是通過結(jié)構(gòu)體實(shí)現(xiàn)的系冗。

和Category語法很相似的還有Extension，二者的區(qū)別在于薪鹦，Extension在編譯期就直接和原類編譯在一起掌敬，而Category是在運(yùn)行時動態(tài)添加到原類中的。

基于之前的源碼分析距芬，我們來分析一下Category的實(shí)現(xiàn)原理涝开。

在_read_images函數(shù)中會執(zhí)行一個循環(huán)嵌套，外部循環(huán)遍歷所有類框仔，并取出當(dāng)前類對應(yīng)Category數(shù)組舀武。內(nèi)部循環(huán)會遍歷取出的Category數(shù)組，將每個category_t對象取出离斩，最終執(zhí)行addUnattachedCategoryForClass函數(shù)添加到Category哈希表中银舱。

// 將category_t添加到list中，并通過NXMapInsert函數(shù)跛梗，更新所屬類的Category列表
static void addUnattachedCategoryForClass(category_t *cat, Class cls, 
                                          header_info *catHeader)
{
    // 獲取到未添加的Category哈希表
    NXMapTable *cats = unattachedCategories();
    category_list *list;

    // 獲取到buckets中的value寻馏，并向value對應(yīng)的數(shù)組中添加category_t
    list = (category_list *)NXMapGet(cats, cls);
    if (!list) {
        list = (category_list *)
            calloc(sizeof(*list) + sizeof(list->list[0]), 1);
    } else {
        list = (category_list *)
            realloc(list, sizeof(*list) + sizeof(list->list[0]) * (list->count + 1));
    }
    // 替換之前的list字段
    list->list[list->count++] = (locstamped_category_t){cat, catHeader};
    NXMapInsert(cats, cls, list);
}

Category維護(hù)了一個名為category_map的哈希表，哈希表存儲所有category_t對象核偿。

// 獲取未添加到Class中的category哈希表
static NXMapTable *unattachedCategories(void)
{
    // 未添加到Class中的category哈希表
    static NXMapTable *category_map = nil;

    if (category_map) return category_map;

    // fixme initial map size
    category_map = NXCreateMapTable(NXPtrValueMapPrototype, 16);

    return category_map;
}

上面只是完成了向Category哈希表中添加的操作诚欠，這時候哈希表中存儲了所有category_t對象。然后需要調(diào)用remethodizeClass函數(shù)，向?qū)?yīng)的Class中添加Category的信息轰绵。

在remethodizeClass函數(shù)中會查找傳入的Class參數(shù)對應(yīng)的Category數(shù)組粉寞，然后將數(shù)組傳給attachCategories函數(shù)，執(zhí)行具體的添加操作左腔。

// 將Category的信息添加到Class唧垦，包含method、property液样、protocol
static void remethodizeClass(Class cls)
{
    category_list *cats;
    bool isMeta;
    isMeta = cls->isMetaClass();

    // 從Category哈希表中查找category_t對象振亮，并將已找到的對象從哈希表中刪除
    if ((cats = unattachedCategoriesForClass(cls, false/*not realizing*/))) {
        attachCategories(cls, cats, true /*flush caches*/);        
        free(cats);
    }
}

在attachCategories函數(shù)中，查找到Category的方法列表鞭莽、屬性列表坊秸、協(xié)議列表，然后通過對應(yīng)的attachLists函數(shù)撮抓，添加到Class對應(yīng)的class_rw_t結(jié)構(gòu)體中妇斤。

// 獲取到Category的Protocol list、Property list丹拯、Method list站超，然后通過attachLists函數(shù)添加到所屬的類中
static void attachCategories(Class cls, category_list *cats, bool flush_caches)
{
    if (!cats) return;
    if (PrintReplacedMethods) printReplacements(cls, cats);

    bool isMeta = cls->isMetaClass();

    // 按照Category個數(shù)，分配對應(yīng)的內(nèi)存空間
    method_list_t **mlists = (method_list_t **)
        malloc(cats->count * sizeof(*mlists));
    property_list_t **proplists = (property_list_t **)
        malloc(cats->count * sizeof(*proplists));
    protocol_list_t **protolists = (protocol_list_t **)
        malloc(cats->count * sizeof(*protolists));

    int mcount = 0;
    int propcount = 0;
    int protocount = 0;
    int i = cats->count;
    bool fromBundle = NO;
    
    // 循環(huán)查找出Protocol list乖酬、Property list死相、Method list
    while (i--) {
        auto& entry = cats->list[i];

        method_list_t *mlist = entry.cat->methodsForMeta(isMeta);
        if (mlist) {
            mlists[mcount++] = mlist;
            fromBundle |= entry.hi->isBundle();
        }

        property_list_t *proplist = 
            entry.cat->propertiesForMeta(isMeta, entry.hi);
        if (proplist) {
            proplists[propcount++] = proplist;
        }

        protocol_list_t *protolist = entry.cat->protocols;
        if (protolist) {
            protolists[protocount++] = protolist;
        }
    }

    auto rw = cls->data();

    // 執(zhí)行添加操作
    prepareMethodLists(cls, mlists, mcount, NO, fromBundle);
    rw->methods.attachLists(mlists, mcount);
    free(mlists);
    if (flush_caches  &&  mcount > 0) flushCaches(cls);

    rw->properties.attachLists(proplists, propcount);
    free(proplists);

    rw->protocols.attachLists(protolists, protocount);
    free(protolists);
}

這個過程就是將Category中的信息，添加到對應(yīng)的Class中咬像，一個類的Category可能不只有一個算撮，在這個過程中會將所有Category的信息都合并到Class中。

方法覆蓋

在有多個Category和原類的方法重復(fù)定義的時候县昂，原類和所有Category的方法都會存在肮柜，并不會被后面的覆蓋。假設(shè)有一個方法叫做method倒彰，Category和原類的方法都會被添加到方法列表中审洞，只是存在的順序不同。

排列順序

在進(jìn)行方法調(diào)用的時候待讳，會優(yōu)先遍歷Category的方法芒澜，并且后面被添加到項目里的Category，會被優(yōu)先調(diào)用创淡。上面的例子調(diào)用順序就是Category3 -> Category2 -> Category1 -> TestObject痴晦。如果從方法列表中找到方法后，就不會繼續(xù)向后查找琳彩，這就是類方法被Category”覆蓋”的原因誊酌。

問題

在有多個Category和原類方法重名的情況下部凑，怎樣在一個Category的方法被調(diào)用后，調(diào)用所有Category和原類的方法术辐？

可以在一個Category方法被調(diào)用后砚尽，遍歷方法列表并調(diào)用其他同名方法施无。但是需要注意一點(diǎn)是辉词，遍歷過程中不能再調(diào)用自己的方法，否則會導(dǎo)致遞歸調(diào)用猾骡。為了避免這個問題瑞躺，可以在調(diào)用前判斷被調(diào)動的方法IMP是否當(dāng)前方法的IMP。

那怎樣在任何一個Category的方法被調(diào)用后兴想，只調(diào)用原類方法呢幢哨？

根據(jù)上面對方法調(diào)用的分析，Runtime在調(diào)用方法時會優(yōu)先所有Category調(diào)用嫂便，所以可以倒敘遍歷方法列表捞镰，只遍歷第一個方法即可，這個方法就是原類的方法毙替。

Category Associate

在項目中經(jīng)常會用到Category岸售，有時候會遇到給Category添加屬性的需求，這時候就需要用到associated的Runtime API了厂画。例如下面的例子中凸丸，需要在屬性的set、get方法中動態(tài)添加實(shí)現(xiàn)袱院。

// 聲明文件
@interface TestObject (Category)
@property (nonatomic, strong) NSObject *object;
@end

// 實(shí)現(xiàn)文件
#import <objc/runtime.h>
#import <objc/message.h>
static void *const kAssociatedObjectKey = (void *)&kAssociatedObjectKey;

@implementation TestObject (Category)

- (NSObject *)object {
    return objc_getAssociatedObject(self, kAssociatedObjectKey);
}

- (void)setObject:(NSObject *)object {
    objc_setAssociatedObject(self, kAssociatedObjectKey, object, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

@end

在Category中添加屬性后屎慢，默認(rèn)是沒有實(shí)現(xiàn)方法的，如果調(diào)用屬性則會崩潰忽洛，而且還會提示下面兩個警告信息腻惠。

Property 'object' requires method 'object' to be defined - use @dynamic or provide a method implementation in this category

Property 'object' requires method 'setObject:' to be defined - use @dynamic or provide a method implementation in this category

下面讓我們看一下associated的源碼，看Runtime是怎么通過Runtime動態(tài)添加set欲虚、get的集灌。下面是objc_getAssociatedObject函數(shù)的實(shí)現(xiàn)代碼，objc_setAssociatedObject實(shí)現(xiàn)也是類似苍在，這里節(jié)省地方就不貼出來了绝页。

id _object_get_associative_reference(id object, void *key) {
    id value = nil;
    uintptr_t policy = OBJC_ASSOCIATION_ASSIGN;
    {
        AssociationsManager manager;
        AssociationsHashMap &associations(manager.associations());
        disguised_ptr_t disguised_object = DISGUISE(object);
        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()) {
                ObjcAssociation &entry = j->second;
                value = entry.value();
                policy = entry.policy();
                if (policy & OBJC_ASSOCIATION_GETTER_RETAIN) {
                    objc_retain(value);
                }
            }
        }
    }
    if (value && (policy & OBJC_ASSOCIATION_GETTER_AUTORELEASE)) {
        objc_autorelease(value);
    }
    return value;
}

從源碼可以看出，所有通過associated添加的屬性寂恬，都被存在一個單獨(dú)的哈希表AssociationsHashMap中续誉。objc_setAssociatedObject和objc_getAssociatedObject函數(shù)本質(zhì)上都是在操作這個哈希表，通過對哈希表進(jìn)行映射來存取對象初肉。

在associated的API中會設(shè)置一些內(nèi)存管理的關(guān)鍵字酷鸦，例如OBJC_ASSOCIATION_ASSIGN，這是用來指定對象的內(nèi)存管理的，這些關(guān)鍵字在Runtime源碼中也有對應(yīng)的處理臼隔。

簡書由于排版的問題嘹裂，閱讀體驗(yàn)并不好，布局摔握、圖片顯示寄狼、代碼等很多問題。所以建議到我Github上氨淌，下載Runtime PDF合集泊愧。把所有Runtime文章總計九篇，都寫在這個PDF中盛正，而且左側(cè)有目錄删咱，方便閱讀。

Runtime PDF

下載地址：Runtime PDF
麻煩各位大佬點(diǎn)個贊豪筝，謝謝痰滋！??

探秘Runtime - 深入剖析Category

探秘Runtime - 深入剖析Category

Category

方法覆蓋

問題

Category Associate