如何在 Category 如何調用本類方法
實際上厉膀，如果一個類的分類重寫了這個類的方法后，那么這個類的這個方法將失效粟焊，起作用的將會是分類的那個重寫方法付魔，在分類重寫的時候 Xcode 也會給出相應警告。
Category is implementing a method which will also be implemented by its primary class
ViewController
@interface ViewController : UIViewController
- (void)test;
@end
@implementation ViewController
- (void)test {
NSLog(@"ViewController");
}
ViewController + GetSelf
@interface ViewController (GetSelf)
- (void)test;
@end
@implementation ViewController (GetSelf)
- (void)test {
NSLog(@"ViewController category (GetSelf)");
}
@end
輸出結果：
2019-02-17 22:37:12.972196+0800 CategoryDemo[10142:78261] ViewController category (GetSelf)
實際上肋坚，Category 并沒有覆蓋主類的同名方法乡括，只是 Category 的方法排在方法列表前面丘侠，而主類的方法被移到了方法列表的后面握牧。
于是，我們可以在 Category 方法里伸眶，利用 Runtime 提供的 API铣鹏，從方法列表里拿回原方法敷扫，從而調用。示例：
@interface ViewController (GetSelf)
- (void)test;
@end
@implementation ViewController (GetSelf)
- (void)test {
NSLog(@"ViewController category (GetSelf)");
[[CategoryManager shared] invokeOriginalMethod:self selector:_cmd];
}
@end
@interface CategoryManager : NSObject
+ (instancetype)shared;
- (void)invokeOriginalMethod:(id)target selector:(SEL)selector;
@end
@implementation CategoryManager
+ (instancetype)shared {
static CategoryManager *shareManager = nil;
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
shareManager = [[CategoryManager alloc] init];
});
return shareManager;
}
- (void)invokeOriginalMethod:(id)target selector:(SEL)selector {
// Get the class method list
uint count;
Method *methodList = class_copyMethodList([target class], &count);
// Print to console
for (int i = 0; i < count; i++) {
Method method = methodList[i];
NSLog(@"Category catch selector : %d %@", i, NSStringFromSelector(method_getName(method)));
}
// Call original method . Note here take the last same name method as the original method
for (int i = count - 1 ; i >= 0; i--) {
Method method = methodList[i];
SEL name = method_getName(method);
IMP implementation = method_getImplementation(method);
if (name == selector) {
// id (*IMP)(id, SEL, ...)
((void (*)(id, SEL))implementation)(target, name);
break;
}
}
free(methodList);
}
@end
遍歷 ViewController 類的方法列表诚卸，列表里最后一個同名的方法葵第，便是原方法。
原理【load 和 Initialize 加載過程】
Category 中不能動態(tài)添加成員變量合溺？
解答：
很多人在面試的時候都會被問到 Category卒密，既然允許用 Category 給類增加方法和屬性，那為什么不允許增加成員變量棠赛？
打開 objc 源代碼哮奇，在objc-runtime-new.h中我們可以發(fā)現(xiàn)：
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;
method_list_t *methodsForMeta(bool isMeta) {
if (isMeta) return classMethods;
else return instanceMethods;
}
property_list_t *propertiesForMeta(bool isMeta) {
if (isMeta) return nil; // classProperties;
else return instanceProperties;
}
};
注意：
-
name：是指class_name而不是category_name。 -
cls：要擴展的類對象睛约，編譯期間是不會定義的鼎俘，而是在運行時通過 *name對應到對應的類對象。 -
instanceMethods：category 中所有給類添加的實例方法的列表辩涝。 -
classMethods：category 中所有添加的類方法的列表而芥。 -
protocols：category 實現(xiàn)的所有協(xié)議的列表。 -
instanceProperties：category 中添加的所有屬性膀值。
從 category 的定義也可以看出 category 可以添加實例方法棍丐，類方法误辑，甚至可以實現(xiàn)協(xié)議，添加屬性歌逢，無法添加實例變量巾钉。
另外在 Objective-C 提供的 runtime 函數(shù)中，確實有一個 class_addIvar() 函數(shù)用于給類添加成員變量秘案，但是閱讀過蘋果的官方文檔的人應該會看到：
This function may only be called after objc_allocateClassPair and before objc_registerClassPair. Adding an instance variable to an existing class is not supported.
大概的意思說砰苍，這個函數(shù)只能在“構建一個類的過程中”調用。當編譯類的時候阱高，編譯器生成了一個實例變量內存布局 ivar layout赚导，來告訴運行時去那里訪問類的實例變量們，一旦完成類定義赤惊，就不能再添加成員變量了吼旧。經(jīng)過編譯的類在程序啟動后就被 runtime 加載，沒有機會調用 addIvar未舟。程序在運行時動態(tài)構建的類需要在調用 objc_registerClassPair 之后才可以被使用圈暗，同樣沒有機會再添加成員變量。
從運行結果中看出裕膀，你不能為一個類動態(tài)的添加成員變量员串，可以給類動態(tài)增加方法和屬性。
因為方法和屬性并不“屬于”類實例昼扛，而成員變量“屬于”類實例寸齐。我們所說的“類實例”概念，指的是一塊內存區(qū)域抄谐，包含了 isa 指針和所有的成員變量渺鹦。所以假如允許動態(tài)修改類成員變量布局，已經(jīng)創(chuàng)建出的類實例就不符合類定義了斯稳，變成了無效對象。但方法定義是在 objc_class 中管理的迹恐，不管如何增刪類方法挣惰，都不影響類實例的內存布局，已經(jīng)創(chuàng)建出的類實例仍然可正常使用殴边。
同理：
某一個類的分類是在 runTime 時憎茂，被動態(tài)的添加到類的結構中。
想了解分類是如何加載的請看 iOS RunTime之六：Category
Category 中動態(tài)添加的屬性
我們知道在 Category 里面是無法為 Category 添加實例變量的锤岸。但是我們很多時候需要在 Category 中添加和對象關聯(lián)的值竖幔，這個時候可以求助關聯(lián)對象來實現(xiàn)。
@interface ViewController (Attribute)
@property (nonatomic, copy) NSString *name;
@end
static char *attributeNameKey;
@implementation ViewController (Attribute)
- (void)setName:(NSString *)name {
objc_setAssociatedObject(self, &attributeNameKey, name, OBJC_ASSOCIATION_COPY);
}
- (NSString *)name {
return objc_getAssociatedObject(self, &attributeNameKey);
}
@end
可以借助關聯(lián)對象來為一個類動態(tài)添加屬性是偷，但是關聯(lián)對象又是存在什么地方呢拳氢？ 如何存儲募逞？ 對象銷毀時候如何處理關聯(lián)對象呢？
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);
object->setHasAssociatedObjects();
}
} 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);
}
可以看到所有的關聯(lián)對象都由 AssociationsManager 管理馋评，而 AssociationsManager 定義如下：
class AssociationsManager {
// associative references: object pointer -> PtrPtrHashMap.
static AssociationsHashMap *_map;
public:
AssociationsManager() { AssociationsManagerLock.lock(); }
~AssociationsManager() { AssociationsManagerLock.unlock(); }
AssociationsHashMap &associations() {
if (_map == NULL)
_map = new AssociationsHashMap();
return *_map;
}
};
AssociationsManager 里面是由一個靜態(tài) AssociationsHashMap 來存儲所有的關聯(lián)對象的放接。這相當于把所有對象的關聯(lián)對象都存在一個全局 map 里面。而 map 的 key 是這個對象的指針地址（任意兩個不同對象的指針地址一定是不同的）留特，而這個 map 的 value 又是另外一個 AssociationsHashMap纠脾，里面保存了關聯(lián)對象的 kv 對。
void *objc_destructInstance(id obj)
{
if (obj) {
// Read all of the flags at once for performance.
bool cxx = obj->hasCxxDtor();
bool assoc = obj->hasAssociatedObjects();
// This order is important.
if (cxx) object_cxxDestruct(obj);
if (assoc) _object_remove_assocations(obj);
obj->clearDeallocating();
}
return obj;
}
runtime 的銷毀對象函數(shù) objc_destructInstance 里面會判斷這個對象有沒有關聯(lián)對象蜕青，如果有苟蹈，會調用 _object_remove_assocations 做關聯(lián)對象的清理工作。
Category 和 Extension 的區(qū)別
-
Extension在編譯期決議右核，它就是類的一部分慧脱，在編譯期和頭文件里的@interface以及實現(xiàn)文件里的@implement一起形成一個完整的類，它伴隨類的產(chǎn)生而產(chǎn)生蒙兰，亦隨之一起消亡磷瘤。Extension一般用來隱藏類的私有信息，你必須有一個類才能為這個類添加Extension搜变，所以你無法為系統(tǒng)的類比如NSString添加Extension采缚。 -
Category則完全不一樣，它是在運行期決議的挠他。 -
Extension可以添加成員變量扳抽，而Category一般不可以。
總之殖侵，就 Category 和 Extension 的區(qū)別來看贸呢，Extension 可以添加成員變量，而 Category 是無法添加成員變量的拢军。因為 Category 在運行期楞陷，對象的內存布局已經(jīng)確定，如果添加實例變量就會破壞類的內部布局茉唉。
如果有覺得上述我講的不對的地方歡迎指出固蛾，大家多多交流溝通。
