背景
在客户端编程中,字典转模型是一个极为常见的问题,苹果提供了KVC来实现NSDictionary到Model的注入,但是KVC只能进行单层浅注入,且无法处理类型转换、key与属性名不对应、深度注入等问题,笔者从Masonry得到启发,开发了一个通过链式配置注入器实现深度注入、类型转换、key-属性名映射等功能的轻量级注入框架SuperKVC。目前已经开源到GitHub,点击这里前往,欢迎Star和Fork,欢迎和我一起完善这个框架!
本文将从应用和原理两个角度介绍SuperKVC
用法
该框架与Masonry类似,要求用户在相关方法参数的block中通过链式编程的方法配置注入器参数,例如配置要创建和注入的class、类型转换器、名称映射等,然后方法会返回注入后的结果。如果JSON的根元素为字典,则返回一个模型实例;如果JSON的根元素为数组,则返回一个模型数组。
浅注入示例
假设我们有如下的JSON,要注入到UserModel中。
{
"id": 100075,
"name": "Greedy",
"birthday": "1993-03-06",
"isVip": true,
"partners": [100236, 100244, 100083]
}
@interface UserModel : NSObject
@property (nonatomic, assign) int64_t userId;
@property (nonatomic, copy) NSString *name;
@property (nonatomic, strong) NSDate *birthday;
@property (nonatomic, assign) BOOL isVip;
@property (nonatomic, strong) NSArray *partners;
@end
注意到这里的几个“坑点”,响应的id
对应的是模型的userId
,响应的birthday
类型为NSString
,而模型的birthday
类型为NSDate
,使用SuperKVC进行注入的代码如下。
// responseObject is a JSONObject(NSDictionary).
UserModel *userModel = [responseObject sk_injectWithInjector:^(SuperKVCInjector *injector) {
// 配置要生成和注入的模型实例类型
injector.bind([UserModel class]);
// 配置key到属性名的映射
injector.mapping(@"id").to(@"userId");
// 配置类型转换
injector.format(@"birthday").with.converter(^NSDate* (NSString *birthdayString) {
NSDateFormatter *fmt = [NSDateFormatter new];
fmt.dateFormat = @"yyyy-MM-dd";
return [fmt dateFromString:birthdayString];
});
}];
可以看到,SuperKVC的使用方式与Masonry十分类似,通过链式编程的方式配置注入器,具有较高的可读性,且使用方便,不需要手动创建模型和从字典中取值和对模型赋值。
数组注入示例
SuperKVC还能够自动处理字典数组到模型数组的转换,对于下面的JSON。
[{
"id": 100075,
"name": "Greedy",
"birthday": "1993-03-06",
"isVip": true,
"partners": [100236, 100244, 100083]
},
{
"id": 100724,
"name": "Charlie",
"birthday": "1996-08-12",
"isVip": false,
"partners": [100710, 100715]
},{},]
按照上面的配置调用注入器,即可得到UserModel的模型数组。
// responseObject is a JSONObject(NSArray).
NSArray<UserModel *> *userModels = [responseObject sk_injectWithInjector:^(SuperKVCInjector *injector) {
injector.bind([UserModel class]);
injector.mapping(@"id").to(@"userId");
injector.format(@"birthday").with.converter(^NSDate* (NSString *birthdayString) {
NSDateFormatter *fmt = [NSDateFormatter new];
fmt.dateFormat = @"yyyy-MM-dd";
return [fmt dateFromString:birthdayString];
});
}];
深度注入示例
对于模型中包含模型的情况,通过converter来嵌套调用注入器,例如下面的JSON。
[{
"id": 100075,
"name": "Greedy",
"birthday": "1993-03-06",
"isVip": false,
"cards": [
{
"id": 400820666,
"name": "King Card of Unity",
"expire": "2026-03-27"
},
{
"id": 622800333,
"name": "Silver Card of Glory",
"expire": "2029-02-21"
},
{
"id": 623400765,
"name": "King Card of Floyt",
"expire": "2024-08-15"
}
]
},{},]
@interface UserModel : NSObject
@property (nonatomic, assign) int64_t userId;
@property (nonatomic, copy) NSString *name;
@property (nonatomic, strong) NSDate *birthday;
@property (nonatomic, assign) BOOL isVip;
@property (nonatomic, strong) NSArray<CardModel *> *cards;
@end
@interface CardModel : NSObject
@property (nonatomic, assign) int64_t cardId;
@property (nonatomic, copy) NSString *name;
@property (nonatomic, strong) NSDate *expireDate;
@end
这里的关键问题是如何配置UserModel的cards属性使得注入的Card字典数组被自动转换为CardModel数组,我们需要借助于converter,对cards对应的字典数组进行格式转换,代码如下。
NSArray *userArray = [responseObject sk_injectWithInjector:^(SuperKVCInjector *injector) {
injector.bind([UserModel class]);
injector.mapping(@"id").to(@"userId");
injector.format(@"birthday").with.converter(^NSDate* (NSString *birthdayString) {
return [fmt dateFromString:birthdayString];
});
injector.format(@"cards").with.converter(^CardModel* (NSDictionary *cardDictArray) {
return [cardDictArray sk_dequeInjectorForClass:[CardModel class] emptyHandler:^(SuperKVCInjector *injector) {
injector.bind([CardModel class]);
injector.mapping(@"id").to(@"cardId");
injector.mapping(@"expire").to(@"expireDate");
injector.format(@"expireDate").with.converter(^NSDate* (NSString *birthdayString) {
return [fmt dateFromString:birthdayString];
});
}];
});
}];
上面几行都是常规配置,注意cards处理方式,借助converter,嵌套的调用injector来实现内层的注入,这里同样涉及到名称映射和日期格式转换,注意到内层的injector使用了deque开头的方法,这是为了避免重复创建相同的注入器而采用的复用机制。
原理
SuperKVC的注入基于反射和KVC实现的,反射是为了获取属性列表和填充实例变量,KVC是为了处理基本类型在运行时对实例变量的填充,链式配置通过block实现,注入器和方法缓存通过LRU(NSCache)实现,整个框架的结构如下。
其中直接暴露给用户的只有NSObject的Category、SuperKVCInjector和SKVManager,Category中包含了执行注入的两个方法。
@interface NSObject (SuperKVC)
- (id)sk_injectWithInjector:(void(^)(SuperKVCInjector *injector))block;
- (id)sk_dequeInjectorForClass:(Class)clazz emptyHandler:(void(^)(SuperKVCInjector *injector))block;
@end
第一个方法用于常规注入,每次调用都会创建一个全新的注入器;第二个方法包含了注入器复用的逻辑,通常用于嵌套调用时提高执行效率。由于第二个方法只是对注入器的缓存逻辑,不涉及核心算法,因此这里不再赘述,我们来看第一个方法的内部实现。
- (id)sk_injectWithInjector:(void (^)(SuperKVCInjector *))block {
SuperKVCInjector *injector = [SuperKVCInjector new];
block(injector);
return [self parseAttributesForInjector:injector];
}
方法内部先是创建了一个注入器,然后调用外部block要求用户配置注入器,最后使用注入器来处理注入逻辑,这里的关键是注入器的配置。根据前面的描述,我们知道要配置的内容主要包括下列内容。
名称 | 功能 | 使用 |
---|---|---|
bind | 配置要生成和注入的模型类 | injector.bind(Class ); |
mapping | 配置字典key到模型属性名的映射 | injector.mapping(responseKey ).to(propertyName ); |
format | 自主格式化一个属性的值 | injector.format(propertyName ).with.converter(^id (id oldVar) { /* your format code */ return newVar ; }); |
ignore | 忽略一些模型属性的注入 | injector.ignore(propName1 ).and(propName2 ) ... |
synthesize | 如果通过@synthesize配置了新的存取规则,需要手动指定属性对应的实例变量名称 | injector.synthesize(propertyName ).to(ivarName ); |
这几类属性的实现比较类似,下面讲解bind和mapping的实现。
bind的用法为传入一个class,并且要求使用小括号调用,为了实现这种效果,要把bind的调用转为函数调用而不能是方法调用,并且利用OC中返回值非空的无参方法可以用点语法调用,因此让bind方法无参,但是返回一个有参block,这个block接收bind的参数即可,由于bind之后不需要连接其他操作,因此block的返回值为空,bind方法的实现如下。
- (void(^)(Class clazz))bind {
return ^(Class clazz) {
SKVBindAttribute *attr = [[SKVBindAttribute alloc] initWithBindClass:clazz];
[self.attributes addObject:attr];
};
}
可以看到,调用bind方法的实质是返回了一个block,这个block接收一个Class参数,并且没有返回值,block内创建了一个SKVBindAttribute,并且存储Class信息,最后将这个属性存储到注入器的属性列表中。注入器的属性列表用来存储所有配置信息,每个信息都是SKVAttribute的一个子类,每个子类对应一行链式操作,用来存储一行的完整上下文。对于bind操作,由于只接收一个Class参数,因此处理较为简单。
下面我们看一下mapping这个接收两个参数的操作,与bind类似,mapping也是一个无参方法,但是由于有后续操作,mapping方法返回的block的返回值不能为空,而是一个mapping时创建的SVKMappingAttribute,这个attribute已经存储了mapping传入的参数,并沿着调用链继续向后传递,具体代码如下。
- (SKVMappingAttribute *(^)(NSString *))mapping {
return ^SKVMappingAttribute* (NSString *responseKey) {
SKVMappingAttribute *attr = [SKVMappingAttribute new];
attr.responseKey = responseKey;
[self.attributes addObject:attr];
return attr;
};
}
经过mapping操作之后,返回的是已经存储了响应key的SKVMappingAttribute实例,通过调用实例的to方法继续处理要映射到的属性key的名称,SKVMappingAttribute的结构如下。
@interface SKVMappingAttribute : SKVAttribute
@property (nonatomic, strong) NSString *responseKey;
@property (nonatomic, strong) NSString *modelKey;
- (void(^)(NSString *modelKey))to;
@end
在前面的调用中,已经设置了responseKey,并且返回了attribute实例,这时候可以继续调用to方法,设置modelKey,从而完成mapping上下文的存储。
- (void (^)(NSString *))to {
return ^ (NSString *to) {
self.modelKey = to;
};
}
其他操作都是类似的,这里不再赘述,如有兴趣可去看源码。经过所有配置之后,接下来要进行的是属性处理,首先将属性排序,排序的优先级通过priority确定,其中bind的优先级最高,经过排序后,将这些配置依次写入注入器,为反射之前做准备,注入器存储这些配置的数据结构如下。
@interface SuperKVCInjector ()
@property (nonatomic, assign) Class bindClass;
@property (nonatomic, strong) NSMutableDictionary *mappingDict;
@property (nonatomic, strong) NSMutableDictionary *synthesizeDict;
@property (nonatomic, strong) NSMutableSet *ignoreSet;
@property (nonatomic, strong) NSMutableDictionary<NSString *, SKVFormatAttribute *> *formatDict;
@end
在遍历的属性中会构建这些集合结构,用于反射时查询和处理,在经过这些操作之后,就可以生成模型开始注入了,这时还要区分要处理的JSONObject是数组还是字典,如果是数组,需要遍历每一个元素分别处理模型注入,最后生成一个全新的模型数组;如果是字典,则直接处理接下来的注入。
- (id)buildAndInjectWithClass:(Class)clazz forInjector:(SuperKVCInjector *)injector {
id ret = nil;
if ([self isKindOfClass:[NSArray class]]) {
ret = @[].mutableCopy;
NSArray *dicts = (NSArray *)self;
for (NSDictionary *dict in dicts) {
id model = [self buildAndInjectModelWithClass:clazz dict:dict forInjector:injector];
[ret addObject:model];
}
} else if ([self isKindOfClass:[NSDictionary class]]) {
ret = [self buildAndInjectModelWithClass:clazz dict:(NSDictionary *)self forInjector:injector];
}
return ret;
}
处理单个模型注入的方法较为复杂,这里截取核心部分,主要是对不同配置的查询和处理,下面的代码是其中使用的一个处理单个属性注入的代码块。
BOOL (^handlePropNamed)(id model, NSString *propName) = ^BOOL (id model, NSString *propName) {
// 一般情况下,字典的key与属性名相同
NSString *responseName = propName;
// 查询映射表,如果不为空,则按照映射表获取属性名对应的响应名
if (injector.mappingDict[propName] != nil) {
responseName = injector.mappingDict[propName];
}
// 根据属性名获取实例变量名,一般情况下为_属性名
NSString *ivarName = [NSString stringWithFormat:@"_%@",propName];
// 如果使用了@synthesize,则需要查表获得实例变量名
if (injector.hasSynthesize && injector.synthesizeDict[propName]) {
ivarName = injector.synthesizeDict[propName];
}
Ivar ivar = class_getInstanceVariable(clazz, ivarName.UTF8String);
if (ivar == NULL) {
return NO;
}
id value = dict[responseName];
// 对于NSNull,不处理注入,否则对模型的操作容易引起崩溃
if ([value isKindOfClass:[NSNull class]]) {
return YES;
}
// 查询自定义格式转换表,如果有自定义格式转换,则调用格式转换的block来转换格式
if (injector.formatDict[propName] != nil) {
SKVFormatAttribute *formatAttr = injector.formatDict[propName];
value = formatAttr.converterBlock(value);
}
// 对于基本类型,需要借助KVC来注入
if ([value isKindOfClass:[NSValue class]]) {
[model setValue:value forKey:propName];
} else {
// 对于对象类型,通过runtime直接注入
object_setIvar(model, ivar, value);
}
return YES;
};
以上就是框架的主要逻辑,具体处理细节请参考源码,源码地址在文章开头。