swift 中可以对协议进行扩展，OC中当然也有黑魔法做同样的事情，ProtocolKit 就是这样的一个第三方库，阅读源码是一件令人高兴的事情。
<pre>
ProtocolKit</pre>
<p>可以这样子使用：</p>
<pre>
@protocol Forkable <NSObject>
@optional
- (void)fork;
@required
- (NSString *)github;
@end
// Protocol Extension
@defs(Forkable)
(void)fork {
NSLog(@"Forkable protocol extension: I'm forking (%@).", self.github);
}(NSString *)github {
return @"This is a required method, concrete class must override me.";
}
@end
// Concrete Class
@interface Forkingdog : NSObject <Forkable>
@end
@implementation Forkingdog
- (NSString *)github {
return @"https://github.com/forkingdog";
}
@end
int main(int argc, const char * argv[]) {
@autoreleasepool {
[[Forkingdog new] fork];
}
return 0;
}
</pre>
<p>可以看出，在实现Forkable协议的具体类中不用实现fork 方法，就能在实例中使用fork方法。</p>
<p>用法，定义完协议之后，给协议添加默认的方法，</p>
<pre>
@defs(Forkable)
</pre>
<p>使用def关键字，Xcode 可以渲染成红色。保留关键字。</p>
<p>此处是一大堆关键的宏。宏，宏，不用怕，我们来展开一下。</p>
<pre>
// For a magic reserved keyword color, use @defs(your_protocol_name)
define defs _pk_extension
// Interface
define _pk_extension($protocol) _pk_extension_imp($protocol, _pk_get_container_class($protocol))
// Implementation
define _pk_extension_imp($protocol, $container_class) \
protocol $protocol; \
@interface $container_class : NSObject <$protocol> @end \
@implementation $container_class \
+ (void)load { \
_pk_extension_load(@protocol($protocol), $container_class.class); \
} \
// Get container class name by counter
define _pk_get_container_class($protocol) _pk_get_container_class_imp($protocol, COUNTER)
define _pk_get_container_class_imp($protocol, $counter) _pk_get_container_class_imp_concat(_PKContainer, $protocol, $counter)
define _pk_get_container_class_imp_concat($a, $b, $c) $a ## $b ## _ ## $c
void _pk_extension_load(Protocol *protocol, Class containerClass);
</code></pre>
<p>展开后如下：</p>
<pre>
@protocol Forkable;
@interface __PKContainer_Forkable_0 : NSObject <Forkable>
@end
@implementation __PKContainer_Forkable_0
- (void)load {
_pk_extension_load(@protocol(Forkable), __PKContainer_Forkable_0.class);
}
(void)fork {
NSLog(@"Forkable protocol extension: I'm forking (%@).", self.github);
}(NSString *)github {
return @"This is a required method, concrete class must override me.";
}
@end
</pre>
<p>新生成了一个类，并且新生成的类实现了Forkable 协议。然后在 load方法中执行_pk_extension_load</p>
<p>我们已经走完了宏干的事情。接来下看看函数做的事情，主要有两个函数：</p>
<pre>
_pk_extension_load //在load函数里把实现的协议方法签名放在全局的数组中
_pk_extension_inject_entry // 遍历所有class实现了对应的协议，并且没有实现方法，就把方法动态注入</code></pre>
<p>调用过程，如下图所示。</p>
<p>
</p>
<p> </p>
<p>技巧：使用attribute((constructor)) 可以在mian函数调用之前做事情</p>
<p>_pk_extension_inject_entry 就是利用 attribute 在main函数开始之前，实现class本身没有实现协议方法的注入。</p>
<pre>
attribute((constructor)) static void _pk_extension_inject_entry(void) {
//上锁，保证多线程安全
pthread_mutex_lock(&protocolsLoadingLock);
unsigned classCount = 0;
//获取所有的class
Class *allClasses = objc_copyClassList(&classCount);
@autoreleasepool {
//遍历全局保存的方法签名
for (unsigned protocolIndex = 0; protocolIndex < extendedProtcolCount; ++protocolIndex) {
PKExtendedProtocol extendedProtcol = allExtendedProtocols[protocolIndex];
for (unsigned classIndex = 0; classIndex < classCount; ++classIndex) {
Class class = allClasses[classIndex];
if (!class_conformsToProtocol(class, extendedProtcol.protocol)) {
continue;
}
//注入方法
_pk_extension_inject_class(class, extendedProtcol);
}
}
}
pthread_mutex_unlock(&protocolsLoadingLock);
free(allClasses);
free(allExtendedProtocols);
extendedProtcolCount = 0, extendedProtcolCapacity = 0;
}
</code></pre>
<p> </p>
