Protocol协议分发器

Protocol协议代理在开发中应用频繁，开发者经常会遇到一个问题——事件的连续传递。比如，为了隔离封装，开发者可能经常会把tableview的delegate或者datesource抽离出独立的对象，而其它对象（比如VC）需要获取某些delegate事件时，只能通过事件的二次传递。有没有更简单的方法了？协议分发器正好可以派上用场

话不多说，先上干货：HJProtocolDispatcher是一个协议实现分发器，通过该工具能够轻易实现将协议事件分发给多个实现者。比如最常见的tableview的delegate协议，通过HJProtocolDispatcher，能够非常容易的分发给多个对象，具体可参考Demo

self.tableView.delegate = AOProtocolDispatcher(UITableViewDelegate, self, self.delegateSource);

原理解析

原理并不复杂， 协议分发器Dispatcher并不实现Protocol协议，其只需将对应的Protocol事件分发给不同的实现者Implemertor。如何实现分发？

熟悉类Class响应链的童鞋都知道，NSObject对象主要通过以下函数响应未实现的Selector函数调用

- (id)forwardingTargetForSelector:(SEL)aSelector OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0);

- (void)forwardInvocation:(NSInvocation *)anInvocation OBJC_SWIFT_UNAVAILABLE("");

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector

因此，协议分发器Dispatcher可以在该函数中将Protocol中Selector的调用传递给实现者Implemertor，由实现者Implemertor实现具体的Selector函数即可

- (void)forwardInvocation:(NSInvocation *)anInvocation {

    SEL aSelector = anInvocation.selector;

    if (!ProtocolContainSel(self.prococol, aSelector)) {

        [super forwardInvocation:anInvocation];

        return;

    }

    for (ImplemertorContext *implemertorContext in self.implemertors) {

        if ([implemertorContext.implemertor respondsToSelector:aSelector]) {

            [anInvocation invokeWithTarget:implemertorContext.implemertor];

        }

    }

}

设计关键

如何做到只对Protocol中Selector函数的调用做分发是设计的关键，系统提供有函数

objc_method_description protocol_getMethodDescription(Protocol *p, SEL aSel, BOOL isRequiredMethod, BOOL isInstanceMethod)

通过以下方法即可判断Selector是否属于某一Protocol

struct objc_method_description MethodDescriptionForSELInProtocol(Protocol *protocol, SEL sel) {

    struct objc_method_description description = protocol_getMethodDescription(protocol, sel, YES, YES);

    if (description.types) {

        return description;

    }

    description = protocol_getMethodDescription(protocol, sel, NO, YES);

    if (description.types) {

        return description;

    }

    return (struct objc_method_description){NULL, NULL};

}

BOOL ProtocolContainSel(Protocol *protocol, SEL sel) {

    return MethodDescriptionForSELInProtocol(protocol, sel).types ? YES: NO;

}

注意事项

协议分发器使用需要了解如何处理带有返回值的函数 ，比如

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section

我们知道，iOS中，函数执行返回的结果存在于寄存器R0中，后执行的会覆盖先执行的结果。因此，当遇到有返回结果的函数时，返回结果以后执行的函数返回结果为最终值，以Demo为例

self.tableView.delegate = AOProtocolDispatcher(UITableViewDelegate, self, self.delegateSource);

TableView的DataSource以后面的self.delegateSource中实现函数返回的结果为准

备注

开发完本项目后发现网上已有朋友实现了协议分发器AOMultiproxier，因此，技术版权属于原作者，本文只做宣传，特此说明！

注：二次传递，在demo里面就是viewController中会响应这个代理方法，在delegateSource中也会响应这个代理方法。也就是两个地方都实现代理方法，会依次响应。

转载：http://www.olinone.com/