iOS中SQLite3数据库修改字段名

1,715 阅读3分钟

关于数据库的增、删、改、查等基本操作本文不去介绍,网上大把的详细教程,这里只介绍实际使用中的一种情况:数据库中修改字段名称。

在产品更迭的过程中我们有时候需要修改数据库表中已存在的字段名或字段类型,这个时候我们去baidu.comjianshu.com的话找到的答案应该是以下两种:

  • alter table user CHANGE new1 new4 int;
  • alter table user rename column a to b;

然而这两种方法都并不能在SQLite3中起作用,因为在 SQLite 中,除了重命名表和在已有的表中添加列,ALTER TABLE 命令不支持其他操作。也就是说,SQLite无法使用rename/change语句修改字段名了。但如果我们产品更迭过程中不巧有这种奇葩需求了,怎么办呢?这时候只能采取迂回政策了:

如何把tableA中的字段oldColumn修改为newColumn

解决方法:

  1. 在tableA中添加字段newColumn
ALTER TABLE tableA ADD COLUMN newColumn INTEGER
  1. 将旧字段oldColumn的值赋值给新字段newColumn
UPDATE tableA SET newColumn = oldColumn
  1. 删除旧字段oldColumn ALTER TABLE tableA DROP COLUMN oldColumn 然后到第三步你会发现又被SQLite坑了:SQLite无法删除字段!!

第3步解决办法:(依旧是迂回政策)
3.1. 先创建备份表tableA_backup将tableA数据备份

CREATE TEMPORARY TABLE tableA_backup(id INTEGER PRIMARY KEY AUTOINCREMENT, newColumn INTEGER)

3.2 将原表tableA中的数据对应的复制到tableA_backup表中

INSERT INTO tableA_backup SELECT id, newColumn FROM tableA

不要将旧字段oldColumn复制过去,备份表tableA_backup中没有oldColumn字段
3.3 此时实际已经得到了改过字段名的表:tableA_backup,然后将原表tableA删掉

DROP TABLE tableA

3.4 然后创建表tableA

CREATE TABLE tableA(id INTEGER PRIMARY KEY AUTOINCREMENT, newColumn INTEGER)

3.5 再把备份表tableA_backup中的数据复制到tableA中

INSERT INTO tableA SELECT id, newColumn FROM tableA_backup

3.6 把备份表tableA_backup删除,Have done!

DROP TABLE tableA_backup



修改字段名及类型的思路就是上面这些步骤,下面给贴上代码方便大家阅读使用:
# 方法调用:
            [self alterTable:@"currentList"
             renameOldColumn:@"isvipa"
                 toNewColumn:@"vipType"
            andNewColumnType:@"INTEGER"
                withDatabase:db
   allRenamedColumnsAndTypes:@"id INTEGER PRIMARY KEY AUTOINCREMENT, isinsider INTEGER, vipType INTEGER, groupcode TEXT, groupname TEXT, grouppicpath TEXT, uuid TEXT, AESkey TEXT"
       allRenamedColumnNames:@"id, isinsider, vipType, groupcode, groupname, grouppicpath, uuid, AESkey"];

/**
 修改表中一个字段名 (sqlite3无法使用rename/change语句直接修改字段名)

 @param tableName 表名
 @param oldColumnName 要被修改的字段名
 @param newColumnName 字段被修改后的名称
 @param newColumnType 字段被修改后的类型
 @param db 数据库对象
 @param renamedColumnsAndTypes 修改后所有的字段及对应的类型  eg: id INTEGER PRIMARY KEY AUTOINCREMENT, isinsider INTEGER, vipType INTEGER, groupcode TEXT, groupname TEXT, grouppicpath TEXT, uuid TEXT, AESkey TEXT
 @param renamedColumnNames 修改后所有的字段名  eg: id, isinsider, vipType, groupcode, groupname, grouppicpath, uuid, AESkey FROM currentList
 */
-(void)alterTable:(NSString *)tableName
  renameOldColumn:(NSString *)oldColumnName toNewColumn:(NSString *)newColumnName andNewColumnType:(NSString *)newColumnType
     withDatabase:(FMDatabase *)db
allRenamedColumnsAndTypes:(NSString *)renamedColumnsAndTypes allRenamedColumnNames:(NSString *)renamedColumnNames
{
    if (![db columnExists:oldColumnName inTableWithName:tableName])
        return;
    
    DLog(@"%@表中存在%@字段,修改%@字段名称为%@", tableName, oldColumnName, oldColumnName, newColumnName);
    // 1.添加新字段
    if (![db executeUpdate:[NSString stringWithFormat:@"ALTER TABLE %@ ADD COLUMN %@ %@", tableName, newColumnName, newColumnType]]) {
        DLog(@"%@修改字段:添加新字段%@失败", tableName, newColumnName);
    }

    // 2.将旧字段赋值给新字段
    if (![db executeUpdate:[NSString stringWithFormat:@"UPDATE %@ SET %@ = %@", tableName, newColumnName, oldColumnName]]) {
        DLog(@"%@修改字段:%@赋值%@字段失败", tableName, oldColumnName, newColumnName);
        return;
    }
    
    // 3.删除旧字段(sqlite3无法使用DROP删除字段,先复制备份,再创建新的表)
    if (![db executeUpdate:@"BEGIN TRANSACTION"]) {
        DLog(@"BEGIN TRANSACTION失败");
    }
    NSString *table_backup = [NSString stringWithFormat:@"%@_backup", tableName];
    NSString *sql = [NSString stringWithFormat:@"CREATE TEMPORARY TABLE %@(%@)", table_backup, renamedColumnsAndTypes];
    if (![db executeUpdate:sql]) {
        DLog(@"CREATE TEMPORARY TABLE %@ 失败", table_backup);
    }
    if (![db executeUpdate:[NSString stringWithFormat:@"INSERT INTO %@ SELECT %@ FROM %@", table_backup, renamedColumnNames, tableName]]) {
        DLog(@"INSERT INTO %@ 失败", table_backup);
    }
    if (![db executeUpdate:[NSString stringWithFormat:@"DROP TABLE %@", tableName]]) {
        DLog(@"DROP TABLE %@ 失败", tableName);
    }
    if (![db executeUpdate:[NSString stringWithFormat:@"CREATE TABLE %@(%@)", tableName, renamedColumnsAndTypes]]) {
        DLog(@"CREATE TABLE %@ 失败", tableName);
    }
    if (![db executeUpdate:[NSString stringWithFormat:@"INSERT INTO %@ SELECT %@ FROM %@", tableName, renamedColumnNames, table_backup]]) {
        DLog(@"INSERT INTO %@ SELECT FROM %@ 失败", tableName, table_backup);
    }
    if (![db executeUpdate:[NSString stringWithFormat:@"DROP TABLE %@", table_backup]]) {
        DLog(@"DROP TABLE %@ 失败", table_backup);
    }
    if (![db executeUpdate:@"COMMIT"]) {
        DLog(@"COMMIT失败");
    }
}