利用Runtime实现对Json数据的ORM框架

代码：

//
//  PBFORMTools.m
//  PBFBaseToolsDemo
//
//  Created by BY-MAC01 on 16/5/12.
//  Copyright © 2016年 BY-MAC01. All rights reserved.
//

#import "PBFORMTools.h"
#import <objc/runtime.h>

@implementation PBFORMTools

//ORM框架，利用Runtime构建Model
+ (void)assembleDictionaryData:(NSObject*)objDes objRes:(NSDictionary*)objRes{
    Class classType = [objDes class];
    //NSArray类型对照表
    NSDictionary *dicArrClassNames = nil;
    NSString *strDicArrNames = @"_dicArrayClassName";
    Ivar ivarDicArrayClassNames = class_getInstanceVariable(classType, [strDicArrNames UTF8String]);
    if(ivarDicArrayClassNames != NULL){
        id idDicArray = object_getIvar(objDes, ivarDicArrayClassNames);
        if ([idDicArray isKindOfClass:[NSDictionary class]]) {
            dicArrClassNames = idDicArray;
        }
    }
    if (objRes) {
        for (NSString * keyNameRes in [objRes allKeys]) {
            if (!keyNameRes || [keyNameRes length]<=0) {
                continue;
            }
            //1、获取Class对应属性，判断属性类型
            const char * attNameRes = [[NSString stringWithFormat:@"_%@",keyNameRes] UTF8String];
            Ivar varTmpDes = class_getInstanceVariable(classType,attNameRes);
            //2、判断Json数据中，对应数据是否存在
            //2.1、如不存在，continue
            if (varTmpDes == NULL) {
                continue;
            }
            //2.2、判断json数据和属性的类型是否一致
            NSObject *objValueRes = [objRes objectForKey:keyNameRes];
            const char *ivarType = ivar_getTypeEncoding(varTmpDes);
            NSString *strIvarTypeDes = [NSString stringWithUTF8String:ivarType];
            strIvarTypeDes = [strIvarTypeDes substringWithRange:NSMakeRange(2, [strIvarTypeDes length]-3)];
            //2.2.1、NSString类型
            if([objValueRes isKindOfClass:[NSString class]] && [strIvarTypeDes containsString:@"NSString"]){
                //设置值
                object_setIvar(objDes, varTmpDes, objValueRes);
            }
            //2.2.2、NSNumber类型
            else if([objValueRes isKindOfClass:[NSNumber class]] && [strIvarTypeDes containsString:@"NSNumber"]){
                //设置值
                object_setIvar(objDes, varTmpDes, objValueRes);
            }
            //2.2.3、NSDictionary类型
            //如属于NSDictionary、且Class该属性类型不属于上述类型，递归调用 assembleDictionaryData
            else if ([objValueRes isKindOfClass:[NSDictionary class]]) {
                //设置值
                Class classDes = objc_getClass([strIvarTypeDes UTF8String]);
                id objDesDic = [[classDes alloc] init];
                [PBFORMTools assembleDictionaryData:objDesDic objRes:(NSDictionary*)objValueRes];
                object_setIvar(objDes, varTmpDes, objDesDic);
            }
            //2.2.4、NSArray类型
            //如类型属于NSArray、循环递归 assembleArrayData
            else if ([objValueRes isKindOfClass:[NSArray class]]) {
                //设置值
                if (dicArrClassNames == NULL) {
                    continue;
                }
                id classType = [dicArrClassNames objectForKey:keyNameRes];
                if (classType == NULL) {
                    continue;
                }
                NSObject *objNSArray = [PBFORMTools assembleArrayData:classType objRes:(NSArray*)objValueRes];
                object_setIvar(objDes, varTmpDes, objNSArray);
            }
        }
    }
    return ;
}

//ORM框架，利用Runtime构建Model
+ (NSArray*)assembleArrayData:(Class)classType objRes:(NSArray*)objRes{
    NSMutableArray *arrRtn = [[NSMutableArray alloc] init];
    for(NSObject* objTmp in objRes){
        //1、NSString类型
        if([objTmp isKindOfClass:[NSString class]] && classType == [NSString class]){
            //设置值
            [arrRtn addObject:objTmp];
        }
        //2、NSNumber类型
        else if([objTmp isKindOfClass:[NSNumber class]] && classType == [NSNumber class]){
            //设置值
            [arrRtn addObject:objTmp];
        }
        //3、属于NSArray，递归调用本方法 assembleArrayData
        else if ([objTmp isKindOfClass:[NSArray class]]) {
            NSArray *arrDes = [PBFORMTools assembleArrayData:classType objRes:(NSArray*)objTmp];
            [arrRtn addObject:arrDes];
        }
        //4、不属于NSArray，属于NSDictionary，直接调用 assembleDictionaryData
        else if ([objTmp isKindOfClass:[NSDictionary class]]) {
            id objDes = [[classType alloc] init];
            [PBFORMTools assembleDictionaryData:objDes objRes:(NSDictionary*)objTmp];
            [arrRtn addObject:objDes];
        }
    }
    return [arrRtn copy];
}



@end

单元测试：

//
//  PBFORMToolsTests.m
//  PBFBaseToolsDemo
//
//  Created by BY-MAC01 on 16/5/12.
//  Copyright © 2016年 BY-MAC01. All rights reserved.
//

#import <XCTest/XCTest.h>
#import "PBFORMTools.h"

//跟随者类
@interface PBFORMToolsTestModelFollower : NSObject
@property (nonatomic,strong)NSString                *name;
@property (nonatomic,strong)NSNumber                *code;
@end
@implementation PBFORMToolsTestModelFollower
@end

//主类
@interface PBFORMToolsTestModelUser : NSObject
@property (nonatomic,strong)NSMutableDictionary                     *KPBFORMToolsDicArrayName;
@property (nonatomic,strong)NSString                                *name;
@property (nonatomic,strong)NSNumber                                *code;
@property (nonatomic,strong)NSMutableArray                          *followers;
@property (nonatomic,strong)NSArray                                 *date;
@property (nonatomic,strong)PBFORMToolsTestModelFollower            *mainFollower;
@end
@implementation PBFORMToolsTestModelUser
- (instancetype)init{
    self = [super init];
    self.KPBFORMToolsDicArrayName = [[NSMutableDictionary alloc] init];
    [self.KPBFORMToolsDicArrayName setValue:[PBFORMToolsTestModelFollower class] forKey:@"followers"];
    [self.KPBFORMToolsDicArrayName setValue:[NSString class] forKey:@"date"];
    return self;
}
@end

@interface PBFORMToolsTests : XCTestCase

@end

@implementation PBFORMToolsTests

- (void)setUp {
    [super setUp];
    // Put setup code here. This method is called before the invocation of each test method in the class.
}

- (void)tearDown {
    // Put teardown code here. This method is called after the invocation of each test method in the class.
    [super tearDown];
}

- (void)testAssembleByDictionary{
    //构造
    NSMutableDictionary *dicAttUser = [[NSMutableDictionary alloc] init];
    [dicAttUser setValue:@"朱林" forKey:@"name"];
    [dicAttUser setValue:[NSNumber numberWithInt:123456] forKey:@"code"];
    //followers
    NSMutableDictionary *dicAttFollower1 = [[NSMutableDictionary alloc] init];
    [dicAttFollower1 setValue:@"张三" forKey:@"name"];
    [dicAttFollower1 setValue:[NSNumber numberWithInt:456] forKey:@"code"];
    NSMutableDictionary *dicAttFollower2 = [[NSMutableDictionary alloc] init];
    [dicAttFollower2 setValue:@"李四" forKey:@"name"];
    [dicAttFollower2 setValue:[NSNumber numberWithInt:789] forKey:@"code"];
    NSArray *arrFollowers = [NSArray arrayWithObjects:dicAttFollower1,dicAttFollower2, nil];
    [dicAttUser setValue:arrFollowers forKey:@"followers"];
    //date
    NSArray *arrDates = [NSArray arrayWithObjects:@"hello",@"world", nil];
    [dicAttUser setValue:arrDates forKey:@"date"];
    //mainFollower
    NSMutableDictionary *dicAttFollowerMain = [[NSMutableDictionary alloc] init];
    [dicAttFollowerMain setValue:@"王五" forKey:@"name"];
    [dicAttFollowerMain setValue:[NSNumber numberWithDouble:789.23] forKey:@"code"];
    [dicAttUser setValue:dicAttFollowerMain forKey:@"mainFollower"];
    
    //work
    PBFORMToolsTestModelUser *userInfo = [[PBFORMToolsTestModelUser alloc] init];
    [PBFORMTools assembleDictionaryData:userInfo objRes:dicAttUser];
    
    //断言
    XCTAssertEqual(userInfo.name, @"朱林",@"name error");
    XCTAssertEqual([userInfo.code integerValue], 123456,@"code error");
    XCTAssertEqual([userInfo.followers count],2 ,@"followers error");
    XCTAssertEqual([userInfo.date count],2 ,@"date error");
    XCTAssertEqual(userInfo.mainFollower.name, @"王五",@"mainFollower.name error");
    XCTAssertEqual([userInfo.mainFollower.code doubleValue], 789.23,@"mainFollower.code error");
}

- (void)testExample {
    // This is an example of a functional test case.
    // Use XCTAssert and related functions to verify your tests produce the correct results.
}

- (void)testPerformanceExample {
    // This is an example of a performance test case.
    [self measureBlock:^{
        // Put the code you want to measure the time of here.
    }];
}

@end