tripleCC的技术博客

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

Objective-C使用范型实现map提示

在当前项目的一些内容加工逻辑较多的界面,我会使用ViewModel来对Model进行一层包装,这样可以保持Model的纯净,也可以减少Controller中弱逻辑代码的堆叠。当然,部分公用内容也是可以通过给Model添加分类来实现的,ViewModel更多是提供特定页面的定制化内容。

由于项目并没有采用MVVM模式,也就没有引入ReactiveCocoa,所以项目里面比较多这样的代码:

1
2
3
4
5
6
7
NSArray <TBVEmployee *> *employees = @[[TBVEmployee new]];

NSMutableArray *items = [NSMutableArray array];
for (TBVEmployee *employee in employees) {
    TBVEmployeeItemViewModel *item = [TBVEmployeeItemViewModel itemWithEmployee:employee];
    [items addObject:item];
}

这段代码主要是为了将Model 转化成ViewModel

这里可以给NSArray增加一个tbv_map方法,让这段代码阅读起来更加函数式:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
@interface NSArray (SwiftOperation)
- (instancetype)tbv_map:(id (^)(id value))block;
@end

@implementation NSArray (SwiftOperation)
- (instancetype)tbv_map:(id(^)(id))block {
    NSCParameterAssert(block != NULL);
    NSMutableArray *temp = [NSMutableArray array];
    for (id element in self) {
        [temp addObject:block(element)];
    }
    return temp;
}
@end

添加分类后,上面那段代码可以这样写:

1
2
3
4
NSArray *employees = @[[TBVEmployee new]];
NSMutableArray *items = [employees tbv_map:^id(id value) {
    return [TBVEmployeeItemViewModel itemWithEmployee:value];
}];

嗯!看起来清爽了不少。但是写多了之后会有一个小瑕疵:为了能mapNSArray可容纳的所有类型,block的传参使用了id类型,当需要使用形参的个别属性时,我需要手动更改id为具体的类名:

1
2
3
4
NSMutableArray *items = [employees tbv_map:^id(TBVEmployee *value) {
    NSLog(@"employee's name: %@", value.name);
    return [TBVEmployeeItemViewModel itemWithEmployee:value];
}];

有没有什么法子,能让Xcode的智能提示帮我直接预测到想要map的元素类型呢?

答案是Objective-C的范型。虽然Objective-C对于范型的支持还是比较弱的,但是处理当前的这个需求还是可以的。

先给NSArray分类添加范型:

1
2
3
@interface NSArray <T> (SwiftOperation)
- (instancetype)tbv_map:(id (^)(T value))block;
@end

然后在使用时,指定需要map数组的元素类型:

1
NSArray <TBVEmployee *> *employees = @[[TBVEmployee new]];

然后Xcode就会根据数组元素的类型,做出智能提示啦:

以此还可以增加其它的操作:

1
2
3
4
5
6
7
8
9
10
@interface NSArray <T> (SwiftOperation)
/** -> swift map */
- (instancetype)tbv_map:(id (^)(T value))block;
/** -> swift filter */
- (instancetype)tbv_filter:(BOOL (^)(T value))block;
/** -> swift reduce */
- (id)tbv_foldLeftWithStart:(T)start reduce:(T (^)(T result, T next))reduce;
/** -> swift forEach */
- (void)tbv_forEach:(void (^)(T value))block;
@end

这里只添加了NSArray类型的操作,NSDictionaryNSSet这类集合类型也可以以此类推来实现。不过上面的方法只是做了一层简易的包装,并没有延迟计算啥的,只是让我写起来能更加开心、舒畅点吧。

更改

2017-07-06 :

需要注意的是 instancetype 会包含范型检查,为了避免使用者过多地进行强制类型转换, tbv_map 可以修改成如下形式:

1
- (NSArray *)tbv_map:(id (^)(T value))block;

Comments