tripleCC的技术博客

ʕ•̫͡•ʔ-̫͡-ʕ•͓͡•ʔ-̫͡-ʕ•̫͡•ʔ-̫͡-ʕ•͓͡•ʔ-̫͡-ʔ

利用runtime兼容老代码小记

通常来说,在项目的初期,因为各种原因,要么人力不够,要么项目周期过紧,会产生难以维护、阅读性较差的代码。而这种代码,我习惯称之为“老代码”。就比如现在手上的项目,初期是由一个被迫转到 iOS 的后端 java 哥们写的,所以工程里面到处都可以看见 java 的一些编码风格,比如模型以 Vo 结尾、接口以 I 开头等,甚至转场动画都是简单粗暴地通过 view 叠加再辅以动画实现的。“老代码”是项目特定时间段的产物,因此,也不能把锅全部推给写这些代码的人。不过前人埋坑,后人总得填啊。

好了,吐槽完毕,开始正题。

当前项目的业务里面有各种可供筛选的条件,比如分类、通道、类别等。“老代码”是这样实现相关功能的:

  • 每种筛选条件都创建一个模型,它们的父类并没有统一
  • 分类存在父子分类,所以有一个 ItreeNode 协议来规范分类模型
  • 通道和类别等其他条件是没有父子分类的,不过“老代码”为了能让视图统一根据 ItreeNode 来渲染界面,依然让它们遵守 ItreeNode ,这里暂且不论这样做是否科学
  • 后台返回的数据中没“全部”条件,需要前端自己添加,“老代码”以 -1 作为“全部”条件的 id ,但是没有统一入口,外部都是手动创建模型,赋值 id 属性。
  • 筛选 UI 控件使用 id 保证条件的唯一性,而获取 id 是通过 ItreeNode 的 obtainItemId 方法

做为一个比较懒的人,做新页面的时候,我肯定不想每次都手动创建一次“全部”条件,至少不能每次都创建不一样的模型,只为了给当前筛选控件添加“全部”条件。更进一步讲,能不能创建一个如下的 NSArray 分类:

1
- (NSArray <id<ITreeNode>> *)tdf_prefixAllTypeWithName:(NSString *)name;

这样针对条件数组,我只需要这样调用就可以了:

1
2
// name可以是@“全部分类"、@“全部通道”或者其他筛选条件
[categories tdf_prefixAllTypeWithName:name]

那么问题来了,要怎么做才能让不同筛选的条件都能使用同一个“全部”接口呢?

首先,我不想用 if-else 或者 switch-if 的,虽然简单粗暴,但是以后如果要添加新的条件类型的话,还需要修改这个判断分支,不够优雅。 既然筛选的UI控件使用的是 ItreeNode 方法获取,那我直接创建一个 AllType 好了:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
@interface TDFTreeNodeAllType : Base <ITreeNode>
/** 类型名(默认‘全部分类’) */
@property (assign, nonatomic) NSString *name;

+ (instancetype)node;
/** 通过这个方法来比较是否是全部分类 */
- (BOOL)isEqual:(id <ITreeNode>)object;
@end

@implementation TDFTreeNodeAllType
+ (instancetype)node {
    TDFTreeNodeAllType *node = [[TDFTreeNodeAllType alloc] init];
    node.id = TDFSilverBulletId;
    return node;
}
- (BOOL)isEqual:(id <ITreeNode>)object {
    return [[self obtainItemId] isEqualToString:[object obtainItemId]];
}
- (NSString *)obtainItemId {
    return self.id;
}
- (NSString *)obtainItemName {
    return self.name
}
...
@end


@implementation NSArray (AllType)
- (NSArray <id<ITreeNode>> *)tdf_prefixAllTypeWithName:(NSString *)name {
    NSAssert([self.firstObject conformsToProtocol:@protocol(ITreeNode)], @"items should conform to ITreeNode");

    NSMutableArray *result = self.mutableCopy;
    if (![[self.firstObject obtainItemId] isEqualToString:TDFSilverBulletId]) {
        TDFTreeNodeAllType *all = [TDFTreeNodeAllType node];
        all.name = name;
        [result insertObject:all atIndex:0];
    }
    return result;
}
@end

搞定收工!

但是稍微跑下后,我发现了一个蛋疼的问题:外部要使用这些条件类型特定属性的时候,程序会崩溃,因为 TDFTreeNodeAllType 找不到对应的方法。 这时,我第一个想到的就是动态创建一个“全部”类型对象:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
- (NSArray <id<ITreeNode>> *)tdf_prefixAllTypeWithName:(NSString *)name {
    NSAssert([self.firstObject conformsToProtocol:@protocol(ITreeNode)], @"items should conform to ITreeNode");

    NSMutableArray *result = self.mutableCopy;
    if (![[self.firstObject obtainItemId] isEqualToString:TDFSilverBulletId]) {
        Class class = [self.firstObject class];
        id <ITreeNode> node = [[class alloc] init];

          ???????????????????????????????????????????
      
        [result insertObject:node atIndex:0];
    }
    return result;
}

但是这里我怎么让 node 成为全部类型对象呢?实际上我并不能直接让 nodeobtainItemId 方法返回 -1 ,也不能通过 method_setImplementation 或者 method_exchangeImplementations 修改 obtainItemId 的实现,因为会影响到类原本对于这个方法的实现。

绕了一圈,还是回到了单独创建一个模型 TDFTreeNodeAllType 表示“全部”条件的方案。现在明确要解决的问题是:如何让外部调用 TDFTreeNodeAllType 没有,筛选条件模型有的方法,还可以不崩溃?嗯,答案在消息转发相关的三步骤。

首先要记录筛选条件的 Class

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
@interface TDFTreeNodeAllType : Base <ITreeNode>
/** 当前类型对应的类,保证同样的操作不会崩溃 */
@property (assign, nonatomic) Class targetClass;

/** 类型名(默认‘全部分类’) */
@property (assign, nonatomic) NSString *name;

+ (instancetype)node;
/** 通过这个方法来比较是否是全部分类 */
- (BOOL)isEqual:(id <ITreeNode>)object;
@end

@implementation NSArray (AllType)
- (NSArray <id<ITreeNode>> *)tdf_prefixAllTypeWithName:(NSString *)name {
    NSAssert([self.firstObject conformsToProtocol:@protocol(ITreeNode)], @"items should conform to ITreeNode");

    NSMutableArray *result = self.mutableCopy;
    if (![[self.firstObject obtainItemId] isEqualToString:TDFSilverBulletId]) {
        TDFTreeNodeAllType *all = [TDFTreeNodeAllType node];
        all.targetClass = [self.firstObject class];
        all.name = name;
        [result insertObject:all atIndex:0];
    }
    return result;
}
@end

在消息转发前,可以通过重载 forwardingTargetForSelector 知悉当前要调用的方法。所以只要在这个方法里面判断 TDFTreeNodeAllType 是否存在此 SEL ,没有的话返回上面保存 Class 创建的对象就可以了:

1
2
3
4
5
6
7
- (id)forwardingTargetForSelector:(SEL)aSelector {
    if (![self respondsToSelector:aSelector]) {
        NSAssert(self.targetClass, @"target class should't be nil in order to avoiding forwarding Invocation");
        return [[self.targetClass alloc] init];
    }
    return [super forwardingTargetForSelector:aSelector];
}

再给NSObject添加一个判断是否为“全部”条件的分类:

1
2
3
4
5
6
@implementation NSObject (AllType)
- (BOOL)tdf_isAllType {
    NSAssert([self conformsToProtocol:@protocol(ITreeNode)], @"object should confirm ITreeNode");
    return [[(id <ITreeNode>)self obtainItemId] isEqualToString:TDFSilverBulletId];
}
@end

这样一来,添加和使用“全部”条件就非常方便了。

虽然很多论调都说 runtime 工作上用的场景非常之少,面试问这些的没啥用,但是个人感觉还是要知道一些的。因为假如真的遇到一些难以通过平常手段搞定的需求的话, runtime 能给开发者带来一些别样的灵感。

Comments