SQLite 数据库升级——新增字段处理总结

一、场景

在工作中，由于新需求总是不断，我们经常会遇到项目中的一些表的结构要改变，比如最常见的就是 新增字段 了。这里我总结一下我遇到这种情况时的处理方法，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];
            }
        }

    }
复制代码

以上的总结参考了并部分摘抄了以下文章，非常感谢以下作者的分享！：

1、《sqlite alter table添加多列》

2、《sqlite并不支持建表后修改主键，或删除列，如果要修改，请参考如下做法》

3、《FMDB数据库升级增加表字段》

原文地址：https://juejin.cn/post/6844903946016784397
参考 https://my.oschina.net/FEEDFACF/blog/901765