IOS 数据库升级数据迁移的实例详解
程序员文章站
2023-11-14 20:58:40
ios 数据库升级数据迁移的实例详解
概要：
很久以前就遇到过数据库版本升级的引用场景，当时的做法是简单的删除旧的数据库文件，重建数据库和表结构，这种暴力升级的方式会导...
ios 数据库升级数据迁移的实例详解
概要：
很久以前就遇到过数据库版本升级的引用场景，当时的做法是简单的删除旧的数据库文件，重建数据库和表结构，这种暴力升级的方式会导致旧的数据的丢失，现在看来这并不不是一个优雅的解决方案，现在一个新的项目中又使用到了数据库，我不得不重新考虑这个问题，我希望用一种比较优雅的方式去解决这个问题，以后我们还会遇到类似的场景，我们都想做的更好不是吗？
理想的情况是：数据库升级，表结构、主键和约束有变化，新的表结构建立之后会自动的从旧的表检索数据，相同的字段进行映射迁移数据，而绝大多数的业务场景下的数据库版本升级是只涉及到字段的增减、修改主键约束，所以下面要实现的方案也是从最基本的、最常用的业务场景去做一个实现，至于更加复杂的场景，可以在此基础上进行扩展，达到符合自己的预期的。
选型定型
网上搜索了下，并没有数据库升级数据迁移简单完整的解决方案，找到了一些思路
1.清除旧的数据，重建表
优点：简单
缺点：数据丢失
2.在已有表的基础上对表结构进行修改
优点：能够保留数据
缺点：规则比较繁琐，要建立一个数据库的字段配置文件，然后读取配置文件，执行sql修改表结构、约束和主键等等，涉及到跨多个版本的数据库升级就变得繁琐并且麻烦了
3.创建临时表，把旧的数据拷贝到临时表，然后删除旧的数据表并且把临时表设置为数据表。
优点：能够保留数据，支持表结构的修改，约束、主键的变更，实现起来比较简单
缺点：实现的步骤比较多
综合考虑，第三种方法是一个比较靠谱的方案。
主要步骤
根据这个思路，分析了一下数据库升级了主要步骤大概如下：
- 获取数据库中旧的表
- 修改表名,添加后缀“_bak”，把旧的表当做备份表
- 创建新的表
- 获取新创建的表
- 遍历旧的表和新表，对比取出需要迁移的表的字段
- 数据迁移处理
- 删除备份表
使用到的sql语句分析
这些操作都是和数据库操作有关系的，所以问题的关键是对应步骤的sql语句了，下面分析下用到的主要的sql语句：
获取数据库中旧的表
select * from sqlite_master where type='table'
结果如下，可以看到有type | name | tbl_name | rootpage | sql 这些数据库字段，我们只要用到name也就是数据库名称这个字段就行了
sqlite> select * from sqlite_master where type='table' ...> ; +-------+---------------+---------------+----------+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+ | type | name | tbl_name | rootpage | sql | +-------+---------------+---------------+----------+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+ | table | t_message_bak | t_message_bak | 2 | create table "t_message_bak" (messageid text, messagetype integer, messagejsoncontent text, retrivetimestring integer, posttimestring integer, readstate integer, primary key(messageid)) | | table | t_message | t_message | 4 | create table t_message ( messageid text, messagetype integer, messagejsoncontent text, retrivetimestring integer, posttimestring integer, readstate integer, addcolumn integer, primary key(messageid) ) | +-------+---------------+---------------+----------+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+ 2 行于数据集 (0.03 秒)
修改表名,添加后缀“_bak”，把旧的表当做备份表
-- 把t_message表修改为t_message_bak表 alter table t_message rename to t_message_bak
获取表字段信息
-- 获取t_message_bak表的字段信息
pragma table_info('t_message_bak')
获取到的表字段信息如下，可以看到有| cid | name | type | notnull | dflt_value | pk | 这些数据库字段，我们只要用到name也就是字段名称这个字段就行了
sqlite> pragma table_info('t_message_bak');
+------+--------------------+---------+---------+------------+------+
| cid | name | type | notnull | dflt_value | pk |
+------+--------------------+---------+---------+------------+------+
| 0 | messageid | text | 0 | null | 1 |
| 1 | messagetype | integer | 0 | null | 0 |
| 2 | messagejsoncontent | text | 0 | null | 0 |
| 3 | retrivetimestring | integer | 0 | null | 0 |
| 4 | posttimestring | integer | 0 | null | 0 |
| 5 | readstate | integer | 0 | null | 0 |
+------+--------------------+---------+---------+------------+------+
6 行于数据集 (0.01 秒)
使用子查询进行数据迁移处理
insert into t_message(messageid, messagetype, messagejsoncontent, retrivetimestring, posttimestring, readstate) select messageid, messagetype, messagejsoncontent, retrivetimestring, posttimestring, readstate from t_message_bak
把t_message_bak表中的messageid, messagetype, messagejsoncontent, retrivetimestring, posttimestring, readstate这些字段的值复制到t_message表中
代码实现
接下来就到了代码的实现步骤了
// 创建新的临时表，把数据导入临时表，然后用临时表替换原表
- (void)basedbversioncontrol {
nsstring * version_old = valueorempty(mmuserdefault.dbversion);
nsstring * version_new = [nsstring stringwithformat:@"%@", db_version];
nslog(@"dbversioncontrol before: %@ after: %@",version_old,version_new);
// 数据库版本升级
if (version_old != nil && ![version_new isequaltostring:version_old]) {
// 获取数据库中旧的表
nsarray* existstables = [self sqliteexiststables];
nsmutablearray* tmpexiststables = [nsmutablearray array];
// 修改表名,添加后缀“_bak”，把旧的表当做备份表
for (nsstring* tablename in existstables) {
[tmpexiststables addobject:[nsstring stringwithformat:@"%@_bak", tablename]];
[self.databasequeue indatabase:^(fmdatabase *db) {
nsstring* sql = [nsstring stringwithformat:@"alter table %@ rename to %@_bak", tablename, tablename];
[db executeupdate:sql];
}];
}
existstables = tmpexiststables;
// 创建新的表
[self inittables];
// 获取新创建的表
nsarray* newaddedtables = [self sqlitenewaddedtables];
// 遍历旧的表和新表，对比取出需要迁移的表的字段
nsdictionary* migrationinfos = [self generatemigrationinfoswitholdtables:existstables newtables:newaddedtables];
// 数据迁移处理
[migrationinfos enumeratekeysandobjectsusingblock:^(nsstring* newtablename, nsarray* publiccolumns, bool * _nonnull stop) {
nsmutablestring* colunmsstring = [nsmutablestring new];
for (int i = 0; i<publiccolumns.count; i++) {
[colunmsstring appendstring:publiccolumns[i]];
if (i != publiccolumns.count-1) {
[colunmsstring appendstring:@", "];
}
}
nsmutablestring* sql = [nsmutablestring new];
[sql appendstring:@"insert into "];
[sql appendstring:newtablename];
[sql appendstring:@"("];
[sql appendstring:colunmsstring];
[sql appendstring:@")"];
[sql appendstring:@" select "];
[sql appendstring:colunmsstring];
[sql appendstring:@" from "];
[sql appendformat:@"%@_bak", newtablename];
[self.databasequeue indatabase:^(fmdatabase *db) {
[db executeupdate:sql];
}];
}];
// 删除备份表
[self.databasequeue indatabase:^(fmdatabase *db) {
[db begintransaction];
for (nsstring* oldtablename in existstables) {
nsstring* sql = [nsstring stringwithformat:@"drop table if exists %@", oldtablename];
[db executeupdate:sql];
}
[db commit];
}];
mmuserdefault.dbversion = version_new;
} else {
mmuserdefault.dbversion = version_new;
}
}
- (nsdictionary*)generatemigrationinfoswitholdtables:(nsarray*)oldtables newtables:(nsarray*)newtables {
nsmutabledictionary<nsstring*, nsarray* >* migrationinfos = [nsmutabledictionary dictionary];
for (nsstring* newtablename in newtables) {
nsstring* oldtablename = [nsstring stringwithformat:@"%@_bak", newtablename];
if ([oldtables containsobject:oldtablename]) {
// 获取表数据库字段信息
nsarray* oldtablecolumns = [self sqlitetablecolumnswithtablename:oldtablename];
nsarray* newtablecolumns = [self sqlitetablecolumnswithtablename:newtablename];
nsarray* publiccolumns = [self publiccolumnswitholdtablecolumns:oldtablecolumns newtablecolumns:newtablecolumns];
if (publiccolumns.count > 0) {
[migrationinfos setobject:publiccolumns forkey:newtablename];
}
}
}
return migrationinfos;
}
- (nsarray*)publiccolumnswitholdtablecolumns:(nsarray*)oldtablecolumns newtablecolumns:(nsarray*)newtablecolumns {
nsmutablearray* publiccolumns = [nsmutablearray array];
for (nsstring* oldtablecolumn in oldtablecolumns) {
if ([newtablecolumns containsobject:oldtablecolumn]) {
[publiccolumns addobject:oldtablecolumn];
}
}
return publiccolumns;
}
- (nsarray*)sqlitetablecolumnswithtablename:(nsstring*)tablename {
__block nsmutablearray<nsstring*>* tablecolumes = [nsmutablearray array];
[self.databasequeue indatabase:^(fmdatabase *db) {
nsstring* sql = [nsstring stringwithformat:@"pragma table_info('%@')", tablename];
fmresultset *rs = [db executequery:sql];
while ([rs next]) {
nsstring* columnname = [rs stringforcolumn:@"name"];
[tablecolumes addobject:columnname];
}
}];
return tablecolumes;
}
- (nsarray*)sqliteexiststables {
__block nsmutablearray<nsstring*>* existstables = [nsmutablearray array];
[self.databasequeue indatabase:^(fmdatabase *db) {
nsstring* sql = @"select * from sqlite_master where type='table'";
fmresultset *rs = [db executequery:sql];
while ([rs next]) {
nsstring* tablename = [rs stringforcolumn:@"name"];
[existstables addobject:tablename];
}
}];
return existstables;
}
- (nsarray*)sqlitenewaddedtables {
__block nsmutablearray<nsstring*>* newaddedtables = [nsmutablearray array];
[self.databasequeue indatabase:^(fmdatabase *db) {
nsstring* sql = @"select * from sqlite_master where type='table' and name not like '%_bak'";
fmresultset *rs = [db executequery:sql];
while ([rs next]) {
nsstring* tablename = [rs stringforcolumn:@"name"];
[newaddedtables addobject:tablename];
}
}];
return newaddedtables;
}
问题
sqlite 删除表文件的大小不变的问题
如有疑问请留言或者到本站社区交流讨论，感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！
推荐阅读