一、场景
在工作中，由于新需求总是不断，我们经常会遇到项目中的一些表的结构要改变，比如最常见的就是 新增字段 了。这里我总结一下我遇到这种情况时的处理方法，SQLite 也有一些坑，希望能帮到有同样需求的朋友们。下面我是用的 fmdb 进行的数据库操作，用原生或者其他工具的升级思想也是一样的。
二、升级操作
用户升级 app 时，当我们判断到旧项目的版本号（或者数据库的单独版本号）小于某个版本号时，就进行我们的升级操作。
1、新增普通字段
如果你要新增的是个普通字段，并不是个 主键（后面会介绍什么情况下会新增主键），那就好办了，操作很简单，直接执行新增字段语句即可。
先复习一下 SQL 的 新增字段 语句：
alter table mydownload add column 'IsFree' varchar(100) default '1'其中的 column 可以省略，同时也可以不给 default 默认值。
下面是判断升级并进行新增操作的具体代码：
if (![dbPointer columnExists:@"LoginUserId" inTableWithName:@"mydownload"]) {
// 如果不存在 LoginUserId 字段则执行添加 LoginUserId 语句，默认值是当前登录的用户id
NSString *addStr = [NSString stringWithFormat:@"alter table mydownload add column 'LoginUserId' varchar(100) default '%@'", [Global sharedGlobal].loginInfo.userId];
if ([dbPointer executeUpdate:addStr]){
DebugLog(@"添加 LoginUserId 字段成功！");
} else {
DebugLog(@"添加 LoginUserId 字段失败！");
}
}
复制代码
⚠️ 注意： 这里要说一下，我们的 SQLite 是个阉割的数据库，有很多数据库操作语句都是不支持的，比如不支持 批量增加字段，所以如果你要新增多个字段，那也只能一个一个的加了。
2、新增、修改主键或删除某字段
如果你新增的是个 主键。比如你想在以前的文章 id 为主键的的基础上，再新增一个用户 id，把文章 id 和用户 id 作为 联合主键，那此时就不能执行新增字段方法了，也是因为 SQLite 的限制，
⚠️ 注意： SQLite 限制了 ALTER TABLE 的部分功能，只能将列添加到表的末尾或更改表的名称。 如果要在表的结构中进行更复杂的更改，则必须 重新创建表。 您可以将现有数据保存到临时表，删除旧表，创建新表，然后从临时表中复制数据。
例如，假设您有一个名为“person”的表，其列名为“id”，“name”和“age”，并且您要从此表中删除列“age”。 以下 SQL 语句步骤说明了如何完成此操作：
BEGIN TRANSACTION; CREATE TEMPORARY TABLE person_backup(id,name); INSERT INTO person_backup SELECT id,name FROM person; DROP TABLE person; CREATE TABLE person(id,name); INSERT INTO person SELECT id,name FROM person_backup; DROP TABLE person_backup; COMMIT; 复制代码
具体例子
下面再把我项目中具体的一个简略例子展示出来供大家参考： 原表状态： 原下载表 mydownload 只有两个字段 EpisodeId 和 DownloaderFilePath，主键是 EpisodeId 。下载的东西不跟随用户。 新需求： 现在要求下载的东西跟随用户走，即下载列表只展示当前用户下载的东西。这就需要新增一个 LoginUserId 用户 id 字段，把它和 EpisodeId 作为一个联合主键，这样下载的东西就可以和用户 id 绑定了。 具体代码操作：
if (![dbPointer columnExists:@"LoginUserId" inTableWithName:@"mydownload"]) {
// 如果不存在 LoginUserId 字段则将原来的表改名
[dbPointer beginTransaction];
BOOL isRollBack = NO;
@try {
if ([dbPointer executeUpdate:@"ALTER TABLE mydownload RENAME TO temp_mydownload"]) {
NSString *executeStr = @"CREATE TABLE mydownload (EpisodeId varchar(100), LoginUserId varchar(100), DownloaderFilePath varchar(100),CONSTRAINT PK_mydownload PRIMARY KEY(EpisodeId,LoginUserId) )";
if ([dbPointer executeUpdate:executeStr]) {
// 从旧数据表把旧数据插入新的数据表中
NSString *insertSql = [NSString stringWithFormat:@"INSERT INTO mydownload (EpisodeId,LoginUserId,DownloaderFilePath) select EpisodeId,'%@','1','',DownloaderFilePath from temp_mydownload", [Global sharedGlobal].loginInfo.userId];// 复制改名后的表到新建的表(注意:一列对应一列的进行复制,新增的字段可以用''来补)
if ([dbPointer executeUpdate:insertSql]) {
[dbPointer executeUpdate:@"drop table temp_mydownload"];// 删除旧表
[[NSUserDefaults standardUserDefaults] setObject:@"YES" forKey:@"hasModifyDownloadDatabase"];// 标记已经升级过下载表
};
}
}
} @catch (NSException *exception) {
isRollBack = YES;
// 事务回退
[dbPointer rollback];
} @finally {
if (!isRollBack) {
// 事务提交
[dbPointer commit];
}
}
}
复制代码
以上的总结参考了并部分摘抄了以下文章，非常感谢以下作者的分享！：
2、《sqlite并不支持建表后修改主键，或删除列，如果要修改，请参考如下做法》
原文地址：https://juejin.cn/post/6844903946016784397
参考 https://my.oschina.net/FEEDFACF/blog/901765
