从0开始弄一个面向OC数据库(三)--数据库升级,数据迁移,删除数据

1,372 阅读10分钟

前言

首先,在上一篇文章从0开始弄一个面向OC数据库(二),讲解了如何向数据库保存或更新一个模型、如何查询数据库里面的数据。其次,本篇要说的内容有:

  • 数据库更新、数据迁移。
  • 删除数据

使用场景: 随着项目的迭代,数据库的内容会越来越多,假如有一天,保存数据库的数据字段增加或者减少怎么办?比如第一个版本,我们保存了学生的姓名,学号,年龄,成绩。到了第10个版本,我们要多保存一项学生的身高,甚至还要再保存学生的体重、性别等等。。怎么办?难道要把之前的数据库表删了,重新建一个数据库表,然后重新插入数据吗?如果我录入了1万个学生的数据,重新开始工作量非常大,之前的数据也会丢失。所以!我们必须要实现数据库更新,以及数据迁移。要增字段就增,要减就减,更新一下就好了。。删除数据的场景咱就不多说了,有个学生转学了,得把他的资料移除吧~

功能实现

数据库更新、数据迁移

当用户对model进行insertOrUpdate的时候,如果这个model里新增了成员变量或者删除了成员变量,这时候我们去进行保存数据是会失败的,因为保存的模型的字段和数据库表结构的字段对应不上。这时候我们就需要进行数据更新。要实现数据库更新,得先缕一缕我们的思路:

首先判断是否需要更新
-- 获取数据库对应的表格创建时的sql语句 从中拿到所有的字段        得到A数组
-- 获取模型中的所有成员变量                                  得到B数组
-- 比较AB数组 如果相等 则不需要更新表 不相等则更新表,并且迁移数据

然后进行迁移数据步骤
-- 根据model的字段,创建一个新的临时表格。
create table if not exists cwstu_tmp(stuNum integer, name text, age integer, address text, primary key(stuNum));
-- 从原来的表格里面,将主键存在的数据从原来的表格插入至新的临时表格
--insert into cwstu_tmp(stuNum) select stuNum from CWStu;
-- 通过主键将老表对应字段的值更新到新表内。
--update cwstu_tmp set name = (select name from cwstu where cwstu_tmp.stuNum = cwstu.stuNum);
-- update cwstu_tmp set age = (select age from cwstu where cwstu_tmp.stuNum = cwstu.stuNum);
-- 删除原有的表格
-- drop table if exists cwstu;
-- 更改临时表格的名字,用户并不知道其实我们偷天换日了
-- alter table cwstu_tmp rename to cwstu;

以上的语句要全部执行成功,数据迁移才算完成,如果执行到一半失败,那么数据库里面可能就会无缘无故多了一个临时表,和一些半完成的数据,显然我们要避免这个问题,于是我们使用到数据库事务

简单介绍一下数据库事务:

一般我们常用的方法有3个 BEGIN TRANSACTION(开始事务) COMMIT TRANSACTION(提交事务)ROLLBACK TRANSACTION(回滚) 然后事务有4个基本属性ACID这些我们就不详细说了。

如何使用事务

在开始执行sql语句之前,我们开启事务,然后逐条执行sql语句,如果某一条sql语句执行失败,则进行回滚,当执行回滚时,之前执行的操作会被取消,数据库会回到开始事务的阶段,当所有sql语句都执行成功之后提交事务即可。

探究数据库是如何进行数据回滚的呢?sqlitie数据库回滚是通过回滚日志实现的,所有事务进行的修改都会先记录到这个回滚日志中,然后在对数据库中的对应行进行写入,进行回滚时,会根据回滚日志滚回之前的状态,打个比方:SVN、git每次提交都会有log,当有一天你想要回退到某个版本,只需要选在对应的log记录revert就可以了,sqlite的回滚类似这样。。还有一个注意点,事务操作一定要是同一个数据库,以及同一个数据库操作句柄。

理论补充完了,现在我们开始上代码,用代码一一实以上的思路

首先获取数据库表格的所有字段,在CWSqliteTableTool封装一个方法

// 获取表的所有字段名,排序后返回
+ (NSArray *)allTableColumnNames:(NSString *)tableName uid:(NSString *)uid {
    
    NSString *queryCreateSqlStr = [NSString stringWithFormat:@"select sql from sqlite_master where type = 'table' and name = '%@'",tableName];
    NSArray *dictArr = [CWDatabase querySql:queryCreateSqlStr uid:uid];
    NSMutableDictionary *dict = dictArr.firstObject;
//    NSLog(@"---------------%@",dict);
    NSString *createSql = dict[@"sql"];
    if (createSql.length == 0) {
        return nil;
    }
    // sql = "CREATE TABLE Student(age integer,stuId integer,score real,height integer,name text, primary key(stuId))";
    createSql = [createSql stringByReplacingOccurrencesOfString:@"\"" withString:@""];
    createSql = [createSql stringByReplacingOccurrencesOfString:@"\n" withString:@""];
    createSql = [createSql stringByReplacingOccurrencesOfString:@"\t" withString:@""];
    
    NSString *nameTypeStr = [createSql componentsSeparatedByString:@"("][1];
    NSArray *nameTypeArray = [nameTypeStr componentsSeparatedByString:@","];
    
    NSMutableArray *names = [NSMutableArray array];
    
    for (NSString *nameType in nameTypeArray) {
        // 去掉主键
        if ([nameType containsString:@"primary"]) {
            continue;
        }
        // 压缩掉字符串里面的 @“ ”  只压缩两端的
        NSString *nameType2 = [nameType stringByTrimmingCharactersInSet:[NSCharacterSet characterSetWithCharactersInString:@" "]];
        
        // age integer
        NSString *name = [nameType2 componentsSeparatedByString:@" "].firstObject;
        [names addObject:name];
    }
    
    [names sortUsingComparator:^NSComparisonResult(NSString *obj1, NSString *obj2) {
        return [obj1 compare:obj2];
    }];
    
    return names;
}

然后再获取模型中的所有成员变量,在CWModelTool内

+ (NSArray *)allIvarNames:(Class)cls {
    NSDictionary *dict = [self classIvarNameAndTypeDic:cls];
    NSArray *names = dict.allKeys;
    // 排序
    names = [names sortedArrayUsingComparator:^NSComparisonResult(id  _Nonnull obj1, id  _Nonnull obj2) {
        return [obj1 compare:obj2];
    }];
    return names;
}

比较两个数组是够相等,相等则不需要更新,否则进行数据库表更新

// 数据库表是否需要更新
+ (BOOL)isTableNeedUpdate:(Class)cls uid:(NSString *)uid targetId:(NSString *)targetId {
    
    NSArray *modelNames = [CWModelTool allIvarNames:cls];
    
    NSString *tableName = [CWModelTool tableName:cls targetId:targetId];
    NSArray *tableNames = [self allTableColumnNames:tableName uid:uid];
    
    return ![modelNames isEqualToArray:tableNames];
}

判断数据库属否需要更新做完了,我们接下来要实现一个方法用事务控制并一次执行多个sql语句,在CWDatabase内:

#pragma mark - 事务
+ (void)beginTransaction:(NSString *)uid {
    [self execSQL:@"BEGIN TRANSACTION" uid:uid];
}

+ (void)commitTransaction:(NSString *)uid {
     [self execSQL:@"COMMIT TRANSACTION" uid:uid];
}

+ (void)rollBackTransaction:(NSString *)uid {
     [self execSQL:@"ROLLBACK TRANSACTION" uid:uid];
}

// 执行多个sql语句
+ (BOOL)execSqls:(NSArray <NSString *>*)sqls uid:(NSString *)uid {
    // 事务控制所有语句必须返回成功,才算执行成功
    [self beginTransaction:uid];
    
    for (NSString *sql in sqls) {
        BOOL result = [self execSQL:sql uid:uid];
        if (result == NO) {
            [self rollBackTransaction:uid];
            return NO;
        }
    }
    [self commitTransaction:uid];
    return YES;
}

做完以上步骤,接下来我们主要来完成数据迁移的多个sql语句的拼接,然后执行。

#pragma mark - 更新数据库表结构、字段改名、数据迁移
// 更新表并迁移数据
+ (BOOL)updateTable:(Class)cls uid:(NSString *)uid targetId:(NSString *)targetId{
    
    // 1.创建一个拥有正确结构的临时表
    // 1.1 获取表格名称
    NSString *tmpTableName = [CWModelTool tmpTableName:cls targetId:targetId];
    NSString *tableName = [CWModelTool tableName:cls targetId:targetId];
    
    if (![cls respondsToSelector:@selector(primaryKey)]) {
        NSLog(@"如果想要操作这个模型,必须要实现+ (NSString *)primaryKey;这个方法,来告诉我主键信息");
        return NO;
    }
    
    // 保存所有需要执行的sql语句
    NSMutableArray *execSqls = [NSMutableArray array];
    
    NSString *primaryKey = [cls primaryKey];
    // 1.2 获取一个模型里面所有的字段,以及类型
    NSString *createTableSql = [NSString stringWithFormat:@"create table if not exists %@(%@, primary key(%@))",tmpTableName,[CWModelTool sqlColumnNamesAndTypesStr:cls],primaryKey];
    
    [execSqls addObject:createTableSql];
    
    // 2.根据主键插入数据
    //--insert into cwstu_tmp(stuNum) select stuNum from CWStu;
    NSString *inserPrimaryKeyData = [NSString stringWithFormat:@"insert into %@(%@) select %@ from %@",tmpTableName,primaryKey,primaryKey,tableName];
    
    [execSqls addObject:inserPrimaryKeyData];
    
    // 3.根据主键,把所有的数据插入到怕新表里面去
    NSArray *oldNames = [CWSqliteTableTool allTableColumnNames:tableName uid:uid];
    NSArray *newNames = [CWModelTool allIvarNames:cls];
    
    // 4.获取更名字典
    NSDictionary *newNameToOldNameDic = @{};
    if ([cls respondsToSelector:@selector(newNameToOldNameDic)]) {
        newNameToOldNameDic = [cls newNameToOldNameDic];
    }
    
    for (NSString *columnName in newNames) {
        NSString *oldName = columnName;
        // 找映射的旧的字段名称
        if ([newNameToOldNameDic[columnName] length] != 0) {
            if ([oldNames containsObject:newNameToOldNameDic[columnName]]) {
                oldName = newNameToOldNameDic[columnName];
            }
        }
        // 如果老表包含了新的列名,应该从老表更新到临时表格里面
        if ((![oldNames containsObject:columnName] && [columnName isEqualToString:oldName]) ) {
            continue;
        }
        // --update cwstu_tmp set name = (select name from cwstu where cwstu_tmp.stuNum = cwstu.stuNum);
        // 5.更新数据
        NSString *updateSql = [NSString stringWithFormat:@"update %@ set %@ = (select %@ from %@ where %@.%@ = %@.%@)",tmpTableName,columnName,oldName,tableName,tmpTableName,primaryKey,tableName,primaryKey];
        
        [execSqls addObject:updateSql];
        
    }
    // 6、删除原来的表格
    NSString *deleteOldTable = [NSString stringWithFormat:@"drop table if exists %@",tableName];
    [execSqls addObject:deleteOldTable];
    // 7、修改临时表格的名字
    NSString *renameTableName = [NSString stringWithFormat:@"alter table %@ rename to %@",tmpTableName,tableName];
    [execSqls addObject:renameTableName];
    
    BOOL result = [CWDatabase execSqls:execSqls uid:uid];
    
    [CWDatabase closeDB];
    
    return result;
}

测试代码就不贴了,最终测试是没问题的,当然我们还有一部分工作没有完成,为了使用我们框架的人更方便,我们必须把这个方法整合到插入或者更新数据那个方法里面,也就是说,当用户保存一条数据时,我们先给他判断是否需要更新数据库表结构,如果需要,我们进行乾坤大挪移默默的帮他把数据库迁移了,然后再进行数据插入或更新。。就像每一个成功的男人背后都有一个默默付出的女人,我们就给用户来当这个女人吧~😁我们在之前封装的insertOrUpdateModel:方法内增加一段代码

#pragma mark 插入或者更新数据
+ (BOOL)insertOrUpdateModel:(id)model uid:(NSString *)uid targetId:(NSString *)targetId {
    // 获取表名
    Class cls = [model class];
    NSString *tableName = [CWModelTool tableName:cls targetId:targetId];
    
    // 判断数据库是否存在对应的表,不存在则创建
    if (![CWSqliteTableTool isTableExists:tableName uid:uid]) {
        [self createSQLTable:cls uid:uid targetId:targetId];
    }else { // 如果表格存在,则检测表格是否需要更新
        if ([CWSqliteTableTool isTableNeedUpdate:cls uid:uid targetId:targetId] ) {
            BOOL result = [self updateTable:cls uid:uid targetId:targetId];
            if (!result) {
                NSLog(@"更新数据库表结构失败!插入或更新数据失败!");
                return NO;
            }
        }
    }
    // 这里是以前的逻辑......
}

数据删除

我们把复杂的流程实现之后,数据删除相对我们来说,简直是小菜一碟。。不多BB,直接上代码

// 根据模型的主键来删除
+ (BOOL)deleteModel:(id)model uid:(NSString *)uid targetId:(NSString *)targetId {
    Class cls = [model class];
    NSString *tableName = [CWModelTool tableName:cls targetId:targetId];
    if (![cls respondsToSelector:@selector(primaryKey)]) {
        NSLog(@"如果想要操作这个模型,必须要实现+ (NSString *)primaryKey;这个方法,来告诉我主键信息");
        return NO;
    }
    NSString *primaryKey = [cls primaryKey];
    id primaryValue = [model valueForKeyPath:primaryKey];
    NSString *deleteSql = [NSString stringWithFormat:@"delete from %@ where %@ = '%@'",tableName,primaryKey,primaryValue];
    
    // 执行数据库
    BOOL result = [CWDatabase execSQL:deleteSql uid:uid];
    // 关闭数据库
    [CWDatabase closeDB];
    
    return result;
}

上面就是进行删除的一个场景,为了方便用户,我们当然要封装更多的场景,这个也非常简单,无非就是拼接一下sql语句delete from %@ where %@ = '%@'还可以加and,or 这种多条件的,反正思路都是一样的,就是多干点苦力活罢了~

4.本篇结束

在此,我们将数据库更新、数据迁移操作合并到了插入数据的方法内,成为了用户背后默默付出的女人,然后数据删除这种对目前的我们来说小意思的东西也实现了。下一篇文章,我们要实现复杂数据类型和对象的存储,比如NSArray,NSDictionary,NSObject,CGRect,UIImage等....以及数组内嵌套模型,嵌套字典等等。。。然后最后的文章我们会对多线程安全进行处理,欢迎围观。

github地址 本次的代码,tag为1.2.0,你可以在release下找到对应的tag下载下来

最后觉得有用的同学,希望能给本文点个喜欢,给github点个star以资鼓励,谢谢大家。

PS: 因为我也是一边封装,一边写文章。效率可能比较低,问题也会有,欢迎大家向我抛issue,有更好的思路也欢迎大家留言!

最后再为大家提供上两篇文章的地址。

从0开始弄一个面向OC数据库(一)

从0开始弄一个面向OC数据库(二)

以及一个0耦合的仿QQ侧滑框架: 一行代码集成超低耦合的侧滑功能

啦啦啦啦。。生命不止。。推广不断😁