【疯狂造轮子-iOS】JSON转Model系列之一

本文转载请注明出处 —— polobymulberry-博客园

1. 前言

之前一直看别人的源码，虽然对自己提升比较大，但毕竟不是自己写的，很容易遗忘。这段时间准备自己造一些轮子，主要目的还是为了提升自身实力，总不能一遇到问题就Google【疯狂造轮子-iOS】JSON转Model系列之一。

之前写i博客园客户端的时候，经常会遇到JSON数据转Model的功能。一般遇到这种问题我都是自己在对应Model类中定义一个+ (instance)initWithAttributes:(NSDictionary *)attributes函数来将NSDictionary*数据转化为对应Model。

下面是i博客园中ICUser的部分代码，其中就使用了initWithAttributes。

// ICUser.h<br/>
#import <Foundation/Foundation.h>

extern NSString *const kUserId;<br/>
extern NSString *const kUserBlogId;<br/>
extern NSString *const kUserDisplayName;<br/>
extern NSString *const kUserAvatarURL;<br/>
@interface ICUser : NSObject

@property (nonatomic, copy)     NSString    *userId;<br/>
@property (nonatomic, assign)   NSInteger   blogId;<br/>
@property (nonatomic, copy)     NSString    *displayName;<br/>
@property (nonatomic, strong)   NSURL       *avatarURL;

+ (instancetype)initWithAttributes:(NSDictionary *

<strong>)attributes;</strong>

@end

// ICUser.m<br/>
#import "ICUser.h"

NSString *const kUserId                 = @"UserId";<br/>
NSString *const kUserBlogId             = @"BlogId";<br/>
NSString *const kUserDisplayName        = @"DisplayName";<br/>
NSString *const kUserAvatarURL          = @"Avatar";<br/>
@implementation ICUser

+ (instancetype)initWithAttributes:(NSDictionary *

<strong>)attributes</strong><br/>
{<br/>
    ICUser *user = [[ICUser alloc] init];

    user.userId             = attributes[kUserId];<br/>
    user.blogId             = [attributes[kUserBlogId] integerValue];<br/>
    user.displayName        = attributes[kUserDisplayName];<br/>
    user.avatarURL          = [NSURL URLWithString:attributes[kUserAvatarURL]];<br/>
return user;<br/>
}

@end

如果我们需要处理的情况符合下面两个要求：

  1. Model类的个数比较少
  2. 每个Model的成员不是很复杂

这种情况下使用上述方法还可以接受。但是一旦Model这一层急剧膨胀，这时候就会让人苦不堪言：

  1. initWithAttributes函数容易写错，而且出错后不方便排查。
  2. 机械性的代码会比较多，不利于提高效率。

考虑到手动转JSON为Model的种种不便，我决定自己写一个JSON转Model的库。虽然网上已经有很多这方面的第三方库，但是我还是想自己造轮子，目的是为了更深入地学习iOS。

2. 设计思路

1.首先要考虑到输入输出是什么？

输入：NSDictionary类型的数据

这里我们先从简，一般我们使用到解析JSON的场合是在网络请求。服务器端返回JSON格式的数据，我们需要转化成本地的Model（此处不讨论直接使用NSDictionary好还是转化为Model好）。并且本篇文章只假设我们网络请求获取到的JSON数据已经在客户端处理成了NSDictionary类型的数据（比较常见）。

输出：Model类型的数据

Model类型的数据。

举例：

目前我实现的一个简单的例子：

#pragma mark - PJXUser<br/>
@interface PJXUser : NSObject<br/>
@property (nonatomic, copy) NSString* username; // 用户名<br/>
@property (nonatomic, copy) NSString* password; // 密码<br/>
@property (nonatomic, copy) NSString* avatarImageURL; // 头像的URL地址<br/>
@end

- (void)runSimpleSample<br/>
{<br/>
    NSDictionary *userDict = @{@"username"      :@"shuaige",<br/>
                               @"password"      :@"",<br/>
                               @"avatarImageURL":@"http://www.example.com/shuaige.png"};

    PJXUser *user = [[PJXUser alloc] <strong>initWithAttributes</strong>:userDict];;

    NSLog(@"username:%@\n",user.username);<br/>
    NSLog(@"password:%@\n",user.password);<br/>
    NSLog(@"avatarImageURL:%@\n",user.avatarImageURL);<br/>
}

这个例子的输入就是userDict这个NSDictionary数据，输出则是一个PJXUser类的对象user。不知道大家有没有注意到，attributes中的key必须和Model中的property的名称一致，比如上例中PJXUser的username、password等属性（当然，你可以使用一个映射表解决这个问题，不过我们先暂时不想那么多）。

2. 核心算法怎么做（输入转输出）？

核心算法部分其实就是调用initWithAttributes:这个函数。那这个函数该如何设计呢？

既然我们需要所有的Model类都可以调用这个initWithAttributes:来完成JSON转Model的工作。那么我们首先想到的就是将这个函数添加到NSObject的category中，并要求所有Model类都继承自NSObject。

所以我首先新建了一个NSObject+Extension的category。并在其中添加了- (instancetype)initWithAttributes:(NSDictionary *)attributes方法。下面我简单阐述下该函数的实现。

其实我的实现思路基本参照的YYKit（传送门）中的YYModel（传送门）部分。其最核心的部分就是调用Model中每个属性的setter方法，并且将传入的attributes中每个元素的value作为setter的参数。

好的，到此为止最核心的部分已经讲完了。可能大家会有很多疑问，比如说如何获取到属性的setter方法，获取后又如何调用setter方法，毕竟此时的操作是在NSObject这个父类中进行的，并没有具体子类的信息。这里我简单提一下，既然编译期我们无法解决上述问题，那么我们就需要借助于OC的runtime机制了。当然，下面会具体讲解如何实现。

3. 具体实现

根据上面的核心思路，我觉得实现起来还存在一些问题：

如何获取每个属性的setter方法？如果现在获取到了每个属性的setter方法（注意是SEL类型），怎么给每个属性调用此方法？

现在是在NSObject中操作，所以不指望使用obj.username = attributes[@”username”]。所以需要使用runtime中的objc_msgSend，使用方法举例如下：

((void (*)(id, SEL, id))(void *) objc_msgSend)((id)self, NSSelectorFromString(@"setUsername:"), @"shuaige");

可以看到我们只需要把其中的@”setUsername”和@”shuaige”替换成我们自己的变量就行。具体怎么替换呢？这时候我们就需要创建一些数据结构来处理和保存相关的属性信息。当然，这些数据结构也是我在实现过程中不断修正的结果。至于中间如何修正，就不细说了，直接上结果。

数据结构的构建其实也很符合我们的思考习惯。既然我们需要对某个类进行处理，不可避免的，我们需要新建一个类来存储Class信息(PJXClassInfo)，而每个Class是由property、ivar和method组成的，所以针对不同组成，我们需要定义三个类来存储property、ivar、method。但是此处我们只需要property信息，所以只建立了property相关的类(PJXPropertyInfo)。

我首先创建了一个PJXClassInfo的类。这个类目前只存放了一个NSMutableDictionary类型的propertyInfos属性，该属性是用来存储这个Class的property信息。而propertyInfos中每个元素其实就是一个个PJXPropertyInfo对象。而每个PJXPropertyInfo保存的就是property的name，setter方法等等，当然，后期会根据需求为PJXPropertyInfo添加新的属性。

这两个类的关系如下：

【疯狂造轮子-iOS】JSON转Model系列之一

下面我们看看具体代码如何实现，如下：

PJXPropertyInfo代码

/**<br/>
 * @brief 存储Model中每个property的信息<br/>
 * @param property 是一个objc_property_t类型变量<br/>
 * @param name 表示该property的名称<br/>
 * @param setter 是一个SEL类型变量，表示该property的setter方法<br/>
 */<br/>
@interface PJXPropertyInfo : NSObject<br/>
@property (nonatomic, assign) objc_property_t property;<br/>
@property (nonatomic, strong) NSString *name;<br/>
@property (nonatomic, assign) SEL setter;<br/>
@end

@implementation PJXPropertyInfo

- (instancetype)initWithPropertyInfo:(objc_property_t)property<br/>
{<br/>
    self = [self init];

    if (self) {<br/>
        // 以备不时之需<br/>
        _property = property;

        // 使用property_getName获取到该property的名称<br/>
        const char *name = property_getName(property);<br/>
        if (name) {<br/>
            _name = [NSString stringWithUTF8String:name];<br/>
        }

        // 目前不考虑自定义setter方法，只考虑系统默认生成setter方法<br/>
         // 也就是说属性username的setter方法为setUsername:<br/>
        NSString *setter = [NSString stringWithFormat:@"%@%@", [_name substringToIndex:].uppercaseString, [_name substringFromIndex:]];<br/>
        _setter = NSSelectorFromString([NSString stringWithFormat:@"set%@:", setter]);<br/>
    }

    return self;<br/>
}

@end

PJXClassInfo代码

/**<br/>
 * @brief 存储Model的Class信息，不过目前只存储Class的property信息<br/>
 * @param propertyInfos 是一个NSMutableDictionary类型的变量，key存储property的名称，value存储对应的PJXPropertyInfo对象<br/>
 */<br/>
@interface PJXClassInfo : NSObject<br/>
@property (nonatomic, strong) NSMutableDictionary *propertyInfos;<br/>
@end

@implementation PJXClassInfo

- (instancetype)initWithClassInfo:(Class)cls<br/>
{<br/>
    self = [self init];

    // 使用class_copyPropertyList获取到Class的所有property（objc_property_t类型）<br/>
     unsigned int propertyCount = ;<br/>
    objc_property_t *properties = class_copyPropertyList(cls, &propertyCount);

    _propertyInfos = [NSMutableDictionary dictionary];

    // 遍历properties数组<br/>
     // 根据对应的objc_property_t信息构建出PJXPropertyInfo对象，并给propertyInfos赋值<br/>
    if (properties) {<br/>
        for (unsigned int i = ; i < propertyCount; i++) {<br/>
            PJXPropertyInfo *propertyInfo = [[PJXPropertyInfo alloc] initWithPropertyInfo:properties[i]];<br/>
            _propertyInfos[propertyInfo.name] = propertyInfo;<br/>
        }<br/>
        // 注意释放空间<br/>
        free(properties);<br/>
    }

    return self;<br/>
}

@end

现在我们回到之前的问题，即如何获取setter并应用？可以看到有了这两个数据结构，我们就已经解决了如何获取到每个property的setter的问题（使用PJXClassInfo的propertyInfos的属性）。剩下的事情就简单了，调用setter方法进行赋值。这里参考YYModel中的方式，使用了一个Core Foundation函数CFDictionaryApplyFunction。

void CFDictionaryApplyFunction(CFDictionaryRef theDict, CFDictionaryApplierFunction applier, void *context);

该函数的作用是对于theDict每个key-value元素都应用applier函数。

所以我们来看看这个applier函数应该怎么设计。

注意这种C语言的applier回调函数不能设计为成员函数，因为成员函数隐藏了一个self参数。此处我们将该回调函数设计成static，并且命名为PropertyWithDictionaryFunction。

// 注意我传入的dictionary就是用户提供的JSON数据<br/>
// 比如此处传入的key==@"username",value==@"shuaige"<br/>
static void PropertyWithDictionaryFunction(const void *key, const void *value, void *context)<br/>
{<br/>
    // 先将key和value转化到Cocoa框架下<br/>
    NSString *keyStr    = (__bridge NSString *)(key);<br/>
    id setValue         = (__bridge id)(value);

    // modelSelf其实就是self，不过我这里用的是static函数，所以没有默认参数self<br/>
    // 此时我们需要借助context参数来获取到这个self<br/>
    // 所以我设计了一个PJXModelContext，用来存储self信息<br/>
    // 另外，此函数的参数中也没有保存每个property信息，也得靠context这个参数来传递<br/>
    // 所以PJXModelContext还需要存储PJXClassInfo对象信息<br/>
    PJXModelContext *modelContext = context;

    id modelSelf = (__bridge id)(modelContext->modelSelf);

    PJXClassInfo *classInfo = (__bridge PJXClassInfo *)(modelContext->modelClassInfo);<br/>
    PJXPropertyInfo *info = classInfo.propertyInfos[keyStr];

    ((void (*)(id, SEL, id))(void *) objc_msgSend)(modelSelf, info.setter, setValue);<br/>
}

最后一步就是在我们的initWithAttributes:函数中构建PJXModelContext并应用到上述函数。

typedef struct {<br/>
    void *modelSelf;<br/>
    void *modelClassInfo;<br/>
}PJXModelContext;

- (instancetype)initWithAttributes:(NSDictionary *)attributes<br/>
{<br/>
    self = [self init];

    if (self) {<br/>
        // 初始化PJXClassInfo对象，并给modelContext赋值<br/>
        PJXModelContext modelContext = {};<br/>
        modelContext.modelSelf = (__bridge void *)(self);<br/>
        PJXClassInfo *classInfo = [[PJXClassInfo alloc] initWithClassInfo:[self class]];<br/>
        modelContext.modelClassInfo = (__bridge void *)classInfo;

        // 应用该函数，将得到JSON->Model后的Model数据<br/>
        CFDictionaryApplyFunction((CFDictionaryRef)attributes, PropertyWithDictionaryFunction, &modelContext);<br/>
    }

    return self;<br/>
}

4. 测试结果

在2.设计思路这一部分，我们举了一个案例。现在我们运行下，看看NSLog的结果：

【疯狂造轮子-iOS】JSON转Model系列之一

成功了！

5. 存在问题

目前的函数整体才100来行，还是存在很多问题没有考虑到。

比如:

  1. 没有考虑用户传入的JSON数据的key值和property的名称不一致
  2. 没有考虑用户传入的JSON数据有嵌套
  3. 没有考虑JSON数据的value值不一定是NSString类型
  4. 没有考虑JSON数据并不一定是NSDictionary类型
  5. 没有考虑用户自定义了Model属性的setter方法
  6. ……