項目中遇到一個數據庫版本升級問題,最后還是解決了,現在來記錄下這個坑。
iOS項目中有數據庫方面的知識,之后的數據庫表比較簡單,為了實現產品經理給的需求,不得不和服務器那邊協調通過增加數據表字段的方式來達到目的。之前,沒有做數據庫版本的記錄問題,需求什么的都滾一邊,產品太不專業了。現在我的需求是在不改變原來APP上的數據的基礎上增加新的字段來實現消息推薦的已同意拒絕過期等狀態。剛開始時,覺得這個問題不是很難,所以有點大意了,中間遇到很多坑,SQLite語法很嚴謹,所以遇到各種語法問題。還好,之前到隨手記面試的時候,也遇到過這個面試題,當時沒想太多,就說每次刪掉舊的數據庫DB,現在想想當時腦袋被門擠了吧。廢話不多說,說多了也是淚啊。。
我的思路:要想確保之前的數據不丟失,應當在之前數據庫表的基礎上創建新的數據庫表,哈哈,解決方案應運而生了。
1.判斷數據庫版本號和保存數據庫版本號
NSString * const kdbManagerVersion = @"DBManagerVersion";
const static NSInteger DB_MANAGER_VER = 1;
// 升級操作
NSInteger ver = [[NSUserDefaults standardUserDefaults] integerForKey:kdbManagerVersion];
if (ver < DB_MANAGER_VER) {
[self upgrade];
}
- (void)saveDBVersion {
[[NSUserDefaults standardUserDefaults] setInteger:DB_MANAGER_VER forKey:kdbManagerVersion];
[[NSUserDefaults standardUserDefaults] synchronize];
}
2.數據庫升級
- (void)upgrade {
// 獲取舊版本號
NSInteger oldVersionNum = [[NSUserDefaults standardUserDefaults] integerForKey:kdbManagerVersion];
if (DB_MANAGER_VER <= oldVersionNum) {
return;
}
// 升級
[self upgrade:oldVersionNum];
// 保存新版本號
[self saveDBVersion];
}
- (void)upgrade:(NSInteger)oldVersion {
if (oldVersion >= DB_MANAGER_VER) {
return;
}
switch (oldVersion) {
case 0:
[self upgradeFromOldToNew];
break;
case 1:
[self upgradeFromOldToNew];
break;
case 2:
break;
default:
break;
}
oldVersion++;
// 判斷是否需要升級
[self upgrade:oldVersion];
}
- (void)upgradeFromOldToNew {
// 執行版本1到版本2的更新
[self changeDBData:_db];
}
3.數據庫升級的SQL語句操作
- (void)changeDBData:(FMDatabase *)db {
[db open];
// 變更數據庫表為一個舊數據表
NSString *sqlStr = [NSString stringWithFormat:@"ALTER TABLE %@ RENAME TO %@", self.tableName, [self.tableName stringByAppendingString:@"_Old"]];
// 執行SQL語句操作
[db executeUpdate:sqlStr];
// 創建新的數據表
NSString *executeStr = [NSString stringWithFormat:@"create table if not exists %@ (LocID integer primary key autoincrement not null,messageID text unique,Content text,TypeName text,SendTime text,CreateTime integer,Status integer,msgtype text,apply_id text,userid text,message_last_id text)",self.tableName];
BOOL bRet = [db executeUpdate:executeStr];
if (bRet) {
// 從舊數據表把舊數據插入新的數據表中
NSString *insertSql = [NSString stringWithFormat:@"INSERT INTO %@ SELECT * ,'','','' FROM %@", self.tableName, [self.tableName stringByAppendingString:@"_Old"]];
[db executeUpdate:insertSql];
}
// 刪除舊的數據表
[db executeUpdate:[NSString stringWithFormat:@"DROP TABLE %@", [self.tableName stringByAppendingString:@"_Old"]]];
[db close];
}