源码篇：Mantle

源码篇：Mantle

编辑： suiling 分类：iOS开发 来源：南峰子的技

 

Mantle是一个用于简化Cocoa或Cocoa Touch程序中model层的第三方库。通常我们的应该中都会定义大量的model来表示各种数据结构，而这些model的初始化和编码解码都需要写大量的代码。而Mantle的优点在于能够大大地简化这些代码。

Mantle源码中最主要的内容包括：

• MTLModel类：通常是作为我们的Model的基类，该类提供了一些默认的行为来处理对象的初始化和归档操作，同时可以获取到对象所有属性的键值集合。

• MTLJSONAdapter类：用于在MTLModel对象和JSON字典之间进行相互转换，相当于是一个适配器。

• MTLJSONSerializing协议：需要与JSON字典进行相互转换的MTLModel的子类都需要实现该协议，以方便MTLJSONApadter对象进行转换。

在此就以这三者作为我们的分析点。

基类MTLModel

MTLModel是一个抽象类，它主要提供了一些默认的行为来处理对象的初始化和归档操作。

初始化

MTLModel默认的初始化方法-init并没有做什么事情，只是调用了下[super init]。而同时，它提供了一个另一个初始化方法：

1	- (instancetype)initWithDictionary:(NSDictionary *)dictionaryValue error:(NSError **)error;

其中参数dictionaryValue是一个字典，它包含了用于初始化对象的key-value对。我们来看下它的具体实现：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

- (instancetype)initWithDictionary:(NSDictionary *)dictionary error:(NSError **)error {     ...     for (NSString *key in dictionary) {         // 1. 将value标记为__autoreleasing，这是因为在MTLValidateAndSetValue函数中，         //    可以会返回一个新的对象存在在该变量中         __autoreleasing id value = [dictionary objectForKey:key];         // 2. value如果为NSNull.null，会在使用前将其转换为nil         if ([value isEqual:NSNull.null]) value = nil;         // 3. MTLValidateAndSetValue函数利用KVC机制来验证value的值对于key是否有效，         //    如果无效，则使用使用默认值来设置key的值。         //    这里同样使用了对象的KVC特性来将value值赋值给model对应于key的属性。         //    有关MTLValidateAndSetValue的实现可参考源码，在此不做详细说明。         BOOL success = MTLValidateAndSetValue(self, key, value, YES, error);         if (!success) return nil;     }     ... }

子类可以重写该方法，以在设置完对象的属性后做进一步的处理或初始化工作，不过需要记住的是：应该通过super来调用父类的实现。

获取属性的键(key)、值(value)

MTLModel类提供了一个类方法+propertyKeys，该方法返回所有@property声明的属性所对应的名称字符串的一个集合，但不包括只读属性和MTLModel自身的属性。在这个类方法会去遍历model的所有属性，如果属性是非只读且其ivar值不为NULL，则获取到表示属性名的字符串，并将其放入到集合中，其实现如下：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24

+ (NSSet *)propertyKeys {     // 1. 如果对象中已有缓存的属性名的集合，则直接返回缓存。该缓存是放在一个关联对象中。     NSSet *cachedKeys = objc_getAssociatedObject(self, MTLModelCachedPropertyKeysKey);     if (cachedKeys != nil) return cachedKeys;     NSMutableSet *keys = [NSMutableSet set];     // 2. 遍历对象所有的属性     //    enumeratePropertiesUsingBlock方法会沿着superclass链一直向上遍历到MTLModel，     //    查找当前model所对应类的继承体系中所有的属性(不包括MTLModel)，并对该属性执行block中的操作。     //    有关enumeratePropertiesUsingBlock的实现可参考源码，在此不做详细说明。     [self enumeratePropertiesUsingBlock:^(objc_property_t property, BOOL *stop) {         mtl_propertyAttributes *attributes = mtl_copyPropertyAttributes(property);         @onExit {             free(attributes);         };         // 3. 过滤只读属性和ivar为NULL的属性         if (attributes->readonly && attributes->ivar == NULL) return;         // 4. 获取属性名字符串，并存储到集合中         NSString *key = @(property_getName(property));         [keys addObject:key];     }];     // 5. 将集合缓存到关联对象中。     objc_setAssociatedObject(self, MTLModelCachedPropertyKeysKey, keys, OBJC_ASSOCIATION_COPY);     return keys; }

有了上面这个类方法，要想获取到对象中所有属性及其对应的值就方法了。为此MTLModel提供了一个只读属性dictionaryValue来取一个包含当前model所有属性及其值的字典。如果属性值为nil，则会用NSNull来代替。另外该属性不会为nil。

1 2 3 4 5

@property (nonatomic, copy, readonly) NSDictionary *dictionaryValue; // 实现 - (NSDictionary *)dictionaryValue {     return [self dictionaryWithValuesForKeys:self.class.propertyKeys.allObjects]; }

合并对象

合并对象是指将两个MTLModel对象按照自定义的方法将其对应的属性值进行合并。为此，在MTLModel定义了以下方法：

1	- (void)mergeValueForKey:(NSString *)key fromModel:(MTLModel *)model;

该方法将当前对象指定的key属性的值与model参数对应的属性值按照指定的规则来进行合并，这种规则由我们自定义的-mergeFromModel:方法来确定。如果我们的子类中实现了-mergeFromModel:方法，则会调用它；如果没有找到，且model不为nil，则会用model的属性的值来替代当前对象的属性的值。具体实现如下：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

- (void)mergeValueForKey:(NSString *)key fromModel:(MTLModel *)model {     NSParameterAssert(key != nil);     // 1. 根据传入的key拼接"mergeFromModel:"字符串，并从该字符串中获取到对应的selector     //    如果当前对象没有实现-mergeFromModel:方法，且model不为nil，则用model的属性值     //    替代当前对象的属性值     //     //    MTLSelectorWithCapitalizedKeyPattern函数以C语言的方式来拼接方法字符串，具体实现请     //    参数源码，在此不详细说明     SEL selector = MTLSelectorWithCapitalizedKeyPattern("merge", key, "FromModel:");     if (![self respondsToSelector:selector]) {         if (model != nil) {             [self setValue:[model valueForKey:key] forKey:key];         }         return;     }     // 2. 通过NSInvocation方式来调用对应的-mergeFromModel:方法。     NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:[self methodSignatureForSelector:selector]];     invocation.target = self;     invocation.selector = selector;     [invocation setArgument:&model atIndex:2];     [invocation invoke]; }

此外，MTLModel还提供了另一个方法来合并两个对象所有的属性值，即：

1	- (void)mergeValuesForKeysFromModel:(MTLModel *)model;

需要注意的是model必须是当前对象所属类或其子类。

归档对象(Archive)

Mantle将对MTLModel的编码解码处理都放在了MTLModel的NSCoding分类中进行处理了，该分类及相关的定义都放在MTLModel+NSCoding文件中。

对于不同的属性，在编码解码过程中可能需要区别对待，为此Mentle定义了枚举MTLModelEncodingBehavior来确定一个MTLModel属性被编码到一个归档中的行为。其定义如下：

1 2 3 4 5

typedef enum : NSUInteger {     MTLModelEncodingBehaviorExcluded = 0,           // 属性绝不应该被编码     MTLModelEncodingBehaviorUnconditional,          // 属性总是应该被编码     MTLModelEncodingBehaviorConditional,            // 对象只有在其它地方被无条件编码时才应该被编码。这只适用于对象属性 } MTLModelEncodingBehavior;

具体每个属性的归档行为我们可以在+encodingBehaviorsByPropertyKey类方法中设置。MTLModel类为我们提供了一个默认实现，如下：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

+ (NSDictionary *)encodingBehaviorsByPropertyKey {     // 1. 获取所有属性键值     NSSet *propertyKeys = self.propertyKeys;     NSMutableDictionary *behaviors = [[NSMutableDictionary alloc] initWithCapacity:propertyKeys.count];     // 2. 对每一个属性进行处理     for (NSString *key in propertyKeys) {         objc_property_t property = class_getProperty(self, key.UTF8String);         NSAssert(property != NULL, @"Could not find property "%@" on %@", key, self);         mtl_propertyAttributes *attributes = mtl_copyPropertyAttributes(property);         @onExit {             free(attributes);         };         // 3. 当属性为weak时，默认设置为MTLModelEncodingBehaviorConditional，否则默认为MTLModelEncodingBehaviorUnconditional，设置完后，将其封装在NSNumber中并放入字典中。         MTLModelEncodingBehavior behavior = (attributes->weak ? MTLModelEncodingBehaviorConditional : MTLModelEncodingBehaviorUnconditional);         behaviors[key] = @(behavior);     }     return behaviors; }

任何不在该返回字典中的属性都不会被归档。子类可以根据自己的需要来指定各属性的归档行为。但在实际时应该通过super来调用父类的实现。

而为了从归档中解码指定的属性，Mantle提供了以下方法：

1	- (id)decodeValueForKey:(NSString *)key withCoder:(NSCoder *)coder modelVersion:(NSUInteger)modelVersion;

默认情况下，该方法会查找当前对象中类似于-decodeWithCoder:modelVersion:的方法，如果找到便会调用相应方法，并按照自定义的方式来处理属性的解码。如果我们没有实现自定义的方法或者coder不需要安全编码，则会对指定的key调用-[NSCoder decodeObjectForKey:]方法。其具体实现如下：

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

- (id)decodeValueForKey:(NSString *)key withCoder:(NSCoder *)coder modelVersion:(NSUInteger)modelVersion {     ...     SEL selector = MTLSelectorWithCapitalizedKeyPattern("decode", key, "WithCoder:modelVersion:");     // 1. 如果自定义了-decodeWithCoder:modelVersion:方法，则通过NSInvocation来调用方法     if ([self respondsToSelector:selector]) {         NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:[self methodSignatureForSelector:selector]];         invocation.target = self;         invocation.selector = selector;         [invocation setArgument:&coder atIndex:2];         [invocation setArgument:&modelVersion atIndex:3];         [invocation invoke];         __unsafe_unretained id result = nil;         [invocation getReturnValue:&result];         return result;     }     @try {         // 2. 如果没有找到自定义的-decodeWithCoder:modelVersion:方法，         //    则走以下流程。         //         // coderRequiresSecureCoding方法的具体实现请参数源码         if (coderRequiresSecureCoding(coder)) {             // 3. 如果coder要求安全编码，则会从需要安全编码的字典中取出属性所对象的类型，然后根据指定             //    类型来对属性进行解码操作。             //    为此，MTLModel提供了类方法allowedSecureCodingClassesByPropertyKey，来获取             //    类的对象包含的所有需要安全编码的属性及其对应的类的字典。该方法首先会查看是否已有             //    缓存的字典，如果没有则遍历类的所有属性。首先过滤掉那些不需要编码的属性，             //    然后遍历剩下的属性，如果是非对象类型或类类型，则其对应的类型设定为NSValue，             //    如果是这两者，则对应的类型即为相应类型。             //    该方法的具体实现请参考源代码。             NSArray *allowedClasses = self.class.allowedSecureCodingClassesByPropertyKey[key];             NSAssert(allowedClasses != nil, @"No allowed classes specified for securely decoding key "%@" on %@", key, self.class);             return [coder decodeObjectOfClasses:[NSSet setWithArray:allowedClasses] forKey:key];         } else {             // 4. 不需要安全编码             return [coder decodeObjectForKey:key];         }     } @catch (NSException *exception) {         ...     } }

当然，所有的编码解码工作还得需要我们实现-initWithCoder:和-encodeWithCoder:两个方法来完成。我们在定义MTLModel的子类时，可以根据自己的需要来对特定的属性进行处理，不过最好调用super的实现来执行父类的操作。MTLModel对这两个方法的实现请参考源码，在此不多作说明。

适配器MTLJSONApadter

为了便于在MTLModel对象和JSON字典之间进行相互转换，Mantle提供了类MTLJSONApadter，作为这两者之间的一个适配器。

MTLJSONSerializing协议

Mantle定义了一个协议MTLJSONSerializing，那些需要与JSON字典进行相互转换的MTLModel的子类都需要实现该协议，以方便MTLJSONApadter对象进行转换。这个协议中定义了三个方法，具体如下：

1 2 3 4 5 6 7

@protocol MTLJSONSerializing @required + (NSDictionary *)JSONKeyPathsByPropertyKey; @optional + (NSValueTransformer *)JSONTransformerForKey:(NSString *)key; + (Class)classForParsingJSONDictionary:(NSDictionary *)JSONDictionary; @end

这三个方法都是类方法。其中+JSONKeyPathsByPropertyKey是必须实现的，它返回的字典指定了如何将对象的属性映射到JSON中不同的key path(字符串值或NSNull)中。任何不在此字典中的属性被认为是与JSON中使用的key值相匹配。而映射到NSNull的属性在JSON序列化过程中将不进行处理。

+JSONTransformerForKey:方法指定了如何将一个JSON值转换为指定的属性值。反过来，转换器也用于将属性值转换成JSON值。如果转换器实现了+JSONTransformer方法，则MTLJSONAdapter会使用这个具体的方法，而不使用+JSONTransformerForKey:方法。另外，如果不需要执行自定义的转换，则返回nil。

重写+classForParsingJSONDictionary:方法可以将当前Model解析为一个不同的类对象。这对象类簇是非常有用的，其中抽象基类将被传递给-[MTLJSONAdapter initWithJSONDictionary:modelClass:]方法，而实例化的则是子类。

如果我们希望MTLModel的一个子类能使用MTLJSONApadter来进行转换，则需要实现这个协议，并实现相应的方法。

初始化

MTLJSONApadter对象有一个只读属性，该属性即为适配器需要处理的MTLModel对象，其声明如下：

1 2	@property (nonatomic, strong, readonly) MTLModel *model;

可见该对象必须是实现了MTLJSONSerializing协议的MTLModel对象。该属性是只读的，因此它只能通过初始化方法来初始化。

MTLJSONApadter对象不能通过-init来初始化，这个方法会直接断言。而是需要通过类提供的两个初始化方法来初始化，如下：

1 2	- (id)initWithJSONDictionary:(NSDictionary *)JSONDictionary modelClass:(Class)modelClass error:(NSError **)error; - (id)initWithModel:(MTLModel*)model;

其中-(id)initWithJSONDictionary:modelClass:error:是使用一个字典和需要转换的类来进行初始化。字典JSONDictionary表示一个JSON数据，这个字典需要符合NSJSONSerialization返回的格式。如果该参数为空，则方法返回nil，且返回带有MTLJSONAdapterErrorInvalidJSONDictionary码的error对象。该方法的具体实现如下：

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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68

- (id)initWithJSONDictionary:(NSDictionary *)JSONDictionary modelClass:(Class)modelClass error:(NSError **)error {     ...     if (JSONDictionary == nil || ![JSONDictionary isKindOfClass:NSDictionary.class]) {         ...         return nil;     }     if ([modelClass respondsToSelector:@selector(classForParsingJSONDictionary:)]) {         modelClass = [modelClass classForParsingJSONDictionary:JSONDictionary];         if (modelClass == nil) {             ...             return nil;         }         ...     }     ...     _modelClass = modelClass;     _JSONKeyPathsByPropertyKey = [[modelClass JSONKeyPathsByPropertyKey] copy];     NSMutableDictionary *dictionaryValue = [[NSMutableDictionary alloc] initWithCapacity:JSONDictionary.count];     NSSet *propertyKeys = [self.modelClass propertyKeys];     // 1. 检验model的+JSONKeyPathsByPropertyKey中字典key-value对的有效性     for (NSString *mappedPropertyKey in self.JSONKeyPathsByPropertyKey) {         // 2. 如果model对象的属性不包含+JSONKeyPathsByPropertyKey返回的字典中的某个属性键值         //    则返回nil。即+JSONKeyPathsByPropertyKey中指定的属性键值必须是model对象所包含         //    的属性。         if (![propertyKeys containsObject:mappedPropertyKey]) {             ...             return nil;         }         id value = self.JSONKeyPathsByPropertyKey[mappedPropertyKey];         // 3. 如果属性不是映射到一个JSON关键路径或者是NSNull，也返回nil。         if (![value isKindOfClass:NSString.class] && value != NSNull.null) {             ...             return nil;         }     }     for (NSString *propertyKey in propertyKeys) {         NSString *JSONKeyPath = [self JSONKeyPathForPropertyKey:propertyKey];         if (JSONKeyPath == nil) continue;         id value;         @try {             value = [JSONDictionary valueForKeyPath:JSONKeyPath];         } @catch (NSException *ex) {             ...             return nil;         }         if (value == nil) continue;         @try {             // 4. 获取一个转换器，             //    如上所述，+JSONTransformerForKey:会先去查看是否有+JSONTransformer方法，             //    如果有则会使用这个具体的方法，如果没有，则调用相应的+JSONTransformerForKey:方法             //    该方法具体实现请参考源码             NSValueTransformer *transformer = [self JSONTransformerForKey:propertyKey];             if (transformer != nil) {                 // 5. 获取转换器转换生的值                 if ([value isEqual:NSNull.null]) value = nil;                 value = [transformer transformedValue:value] ?: NSNull.null;             }             dictionaryValue[propertyKey] = value;         } @catch (NSException *ex) {             ...             return nil;         }     }     // 6. 初始化_model     _model = [self.modelClass modelWithDictionary:dictionaryValue error:error];     if (_model == nil) return nil;     return self; }

另外，MTLJSONApadter还提供了几个类方法来创建一个MTLJSONApadter对象，如下：

1 2 3

+ (id)modelOfClass:(Class)modelClass fromJSONDictionary:(NSDictionary *)JSONDictionary error:(NSError **)error; + (NSArray *)modelsOfClass:(Class)modelClass fromJSONArray:(NSArray *)JSONArray error:(NSError **)error; + (NSDictionary *)JSONDictionaryFromModel:(MTLModel*)model;

具体实现可参考源码。

从对象中获取JSON数据

从MTLModel对象中获取JSON数据是上述初始化过程中的一个逆过程。该过程由-JSONDictionary方法来实现，具体如下：

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

- (NSDictionary *)JSONDictionary {     NSDictionary *dictionaryValue = self.model.dictionaryValue;     NSMutableDictionary *JSONDictionary = [[NSMutableDictionary alloc] initWithCapacity:dictionaryValue.count];     [dictionaryValue enumerateKeysAndObjectsUsingBlock:^(NSString *propertyKey, id value, BOOL *stop) {         NSString *JSONKeyPath = [self JSONKeyPathForPropertyKey:propertyKey];         if (JSONKeyPath == nil) return;         // 1. 获取属性的值         NSValueTransformer *transformer = [self JSONTransformerForKey:propertyKey];         if ([transformer.class allowsReverseTransformation]) {             if ([value isEqual:NSNull.null]) value = nil;             value = [transformer reverseTransformedValue:value] ?: NSNull.null;         }         NSArray *keyPathComponents = [JSONKeyPath componentsSeparatedByString:@"."];         // 2. 对于嵌套属性值的设置，会先从keypath中获取每一层属性，         //    如果当前层级的obj中没有该属性，则为其设置一个空字典；然后再进入下一层级，依此类推         //    最后设置如下形式的字典: @{@"nested": @{@"name": @"foo"}}         id obj = JSONDictionary;         for (NSString *component in keyPathComponents) {             if ([obj valueForKey:component] == nil) {                 [obj setValue:[NSMutableDictionary dictionary] forKey:component];             }             obj = [obj valueForKey:component];         }         [JSONDictionary setValue:value forKeyPath:JSONKeyPath];     }];     return JSONDictionary; }

从上可以看出，该方法实际上最终获得的是一个字典。而获得字典后，再将其序列化为JSON串就容易了。

MTLJSONApadter也提供了一个简便的方法，来从一个model中获取一个JSON字典，其定义如下：

1 2	+ (NSDictionary *)JSONDictionaryFromModel:(MTLModel *)model;

MTLManagedObjectAdapter

为了适应Core Data，Mantle专门定义了MTLManagedObjectAdapter类。该类用作MTLModel对象与NSManagedObject对象之前的转换。具体的我们在此不详细描述。

技术点总结

Mantle的功能主要是进行对象间数据的转换：即如何在一个MTLModel和一个JSON字典中进行数据的转换。因此，所使用的技术大都是Cocoa Foundation提供的功能。除了对于Core Data的处理之外，主要用到的技术的有如下几条：

KVC的应用：这主要体现在对MTLModel子类的属性赋值中，通过KVC机制来验证值的有效性并为属性赋值。

NSValueTransform：这主要用于对JSON值转换为属性值的处理，我们可以自定义转换器来满足我们自己的转换需求。

NSInvocation：这主要用于统一处理针对特定key值的一些方法的调用。比如-mergeFromModel:这一类方法。

Run time函数的使用：这主要用于对从一个字符串中获取到方法对应的字符串，然后通过sel_registerName函数来注册一个selector。

当然在Mantle中还会涉及到其它的一些技术点，在此不多做叙述。

参考

Mantle工程