OC协议的默认实现

Swift 中,定义协议 protocol 时,可以使用 extension 给它的某些方法提供默认实现:

protocol Printable {
    func print()
    func print1()
}
extension Printable {
    func print() {
        print("default print")
    }
}

有了上面的代码,当创建一个遵从 Printable 协议的类或者是结构体时,就能获得 print() 方法。可以在需要的时候重新定义这个方法,如果不重新定义,就会使用这个默认方法。

遗憾的是 Objective-C 并没有提供这样的便利。当我们想为一个协议提供默认实现时,通常会提供一个基类,基类遵从协议并且提供默认实现。如果不想使用默认实现,需继承基类,在子类中重写方法。有没有办法可以像 Swift 那样,在协议声明的地方而不需要继承链,就能简便的提供默认实现呢?

为协议提供默认实现,可以理解为所有遵从协议的类都拥有提前定义的协议方法。Objective-C 强大的运行时可以获得程序中所有的类,可以获得类遵从的协议,还可以为每一个类添加方法。我们可以用一个临时类来实现协议的方法,在所有的类自己的方法全部加载完后,为遵从协议的类添加临时类的默认方法。之所以要在类自己的方法加载完成后,是为了防止类自己的实现被覆盖,当方法已经存在时,class_addMethod 会添加失败,这正是我们希望的。

__attribute__((constructor)) 特性修饰的函数,会在 runtime 加载完所有类之后,main 函数执行之前运行。通过这样一个函数,将满足条件的类加上提供的默认实现。

现在需要一个临时类,来为协议实现默认方法。假设有协议:

@protocol Printable
@optional
- (NSString *)desc;
@end

为协议提供临时类,来提供默认实现:

@interface PrintableTemporaryClass : NSObject <Printable>
@end 
@implementation PrintableTemporaryClass
- (NSString *)desc {
            return @"Default desc";
}
@end

上面代码中,print 方法前的部分对所有协议来说都是类似的的,可以定义个宏来简化代码:

#define extensionProtocol(NAME) \
interface NAME ## TemporaryClass : NSObject <NAME> \
@end \
@implementation NAME ## TemporaryClass \

此时,临时类的写法会变成下面这样:

@extensionProtocol(Printable)
- (NSString *)desc {
            return @"Default desc";
}
@end

然后通过遍历所有的类,进而遍历类所有的协议,并查找运行时中是否有协议对应的临时类(通过名字关联),如果有,遍历该临时类和元类的所有方法进行添加。关键的 __attribute__((constructor)) 函数如下:

__attribute__((constructor)) static void _append_default_implement_method_to_class() {
    unsigned classCount;
    Class *classes = objc_copyClassList(&classCount);
    //第一层遍历所有的类
    for (int i = 0; i < classCount; i ++) {
        Class class = classes[i];
        Class metaClass = object_getClass(class);
    
        unsigned protocolCount;
        Protocol * __unsafe_unretained *protocols = class_copyProtocolList(class, &protocolCount);
        //第二层遍历类中所有的协议
        for (int j = 0; j < protocolCount; j ++) {
            Protocol *protocol = protocols[j];
            NSString *tempClassName = [NSString stringWithFormat:@"%sTemporaryClass", protocol_getName(protocol)];
            Class tempClass = objc_getClass(tempClassName.UTF8String);
            if (!tempClass) continue;
        
            unsigned methodCount;
            Method *methods = class_copyMethodList(tempClass, &methodCount);
            //第三层遍历临时类的所有方法并添加
            for (int k = 0; k < methodCount; k ++) {
                Method method = methods[k];
                class_addMethod(class, method_getName(method), method_getImplementation(method), method_getTypeEncoding(method));
            }
            free(methods);
        
            Class metaTempClass = object_getClass(tempClass);
            unsigned metaMethodCount;
            Method *metaMethods = class_copyMethodList(metaTempClass, &metaMethodCount);
            //第三层遍历临时类元类的所有方法并添加
            for (int k = 0; k < metaMethodCount; k ++) {
                Method method = metaMethods[k];
                class_addMethod(metaClass, method_getName(method), method_getImplementation(method), method_getTypeEncoding(method));
            }
            free(metaMethods);
        }
        free(protocols);
    }
    free(classes);
}

这样,任何一个遵从 Printable 协议的类,即使没有实现 desc 方法,也具有了 desc 的默认实现。

以上包含了所有为 Objective-C 提供协议默认实现功能的完整代码,核心是 __attribute__((constructor)) 修饰的函数,利用运行时来为类和元类添加方法,还定义了一个简单的宏方便代码书写。如果有别的意见,欢迎与我交流。

Github 地址: https://github.com/Jeffery91/ExtensionProtocol

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 160,026评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,655评论 1 296
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,726评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,204评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,558评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,731评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,944评论 2 314
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,698评论 0 203
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,438评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,633评论 2 247
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,125评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,444评论 3 255
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,137评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,103评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,888评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,772评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,669评论 2 271

推荐阅读更多精彩内容

  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 170,573评论 25 707
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,103评论 18 139
  • importUIKit classViewController:UITabBarController{ enumD...
    明哥_Young阅读 3,676评论 1 10
  • *面试心声:其实这些题本人都没怎么背,但是在上海 两周半 面了大约10家 收到差不多3个offer,总结起来就是把...
    Dove_iOS阅读 27,036评论 29 470
  • 当伤心时只有我一个人,没有朋友安慰,这样的次数增多了以后,我开始羡慕那些可以轻易面对面看到朋友,可以当天倾诉的学生...
    飞鱼kiwi阅读 569评论 6 4