/*
數(shù)據(jù)庫(kù)升級(jí)流程
1.每一次運(yùn)行程序，判斷數(shù)據(jù)庫(kù)是否存在。如果不存在，創(chuàng)建數(shù)據(jù)庫(kù)。如果存在，進(jìn)行其他后續(xù)處理。
2.第一次安裝app，創(chuàng)建數(shù)據(jù)庫(kù)，并將版本號(hào)存入數(shù)據(jù)庫(kù)中。同時(shí)保存一個(gè)當(dāng)前版本號(hào)加1的字段到數(shù)據(jù)庫(kù)中。
3.更新app，從數(shù)據(jù)庫(kù)中讀取到上一次保存的版本號(hào)字段。比如要升級(jí)為2，會(huì)直接從case2開始執(zhí)行。修改完數(shù)據(jù)結(jié)構(gòu)后，再一次將版本號(hào)字段存入數(shù)據(jù)庫(kù)。
4.每一次數(shù)據(jù)庫(kù)結(jié)構(gòu)有更新，直接在后面加case語(yǔ)句即可。
*/
YSqliteManager.h
//
// YSqliteManager.h
// DatabaseUpdateDemo
//
// Created by *** on 2017/2/15.
// Copyright ? 2017年 yuanzhiying. All rights reserved.
//
#import <Foundation/Foundation.h>
#import <sqlite3.h>
#import "YUserData.h"
@interface YSqliteManager : NSObject {
sqlite3 *database;//數(shù)據(jù)庫(kù)
NSString *databasePath;//數(shù)據(jù)庫(kù)文件路徑
BOOL isOpend;//是否已經(jīng)打開數(shù)據(jù)庫(kù)
}
#pragma mark - 初始化
//初始化數(shù)據(jù)庫(kù)
+ (YSqliteManager *)shareManager;
//打開數(shù)據(jù)庫(kù)
- (BOOL)openDatabase;
//關(guān)閉數(shù)據(jù)庫(kù)
- (void)closeDatabase;
#pragma mark - 所有表的操作
#pragma mark 用戶表
//是否存在
- (BOOL)isExistUserWithUserID:(NSInteger)_userid;
//插入
- (void)insertUserInfoWithUserInfo:(YUserData *)_userinfo;
//更新
- (void)updateUserInfoWithUserInfo:(YUserData *)_userinfo UserID:(NSInteger)_userid;
//獲取用戶數(shù)據(jù)
- (YUserData *)getUserInfoWithUserID:(NSInteger)_userid;
//刪除用戶數(shù)據(jù)
- (void)deleteUserInfoDataWithUserID:(NSInteger)_userid;
@end
YSqliteManager.m
//
// YSqliteManager.m
// DatabaseUpdateDemo
//
// Created by *** on 2017/2/15.
// Copyright ? 2017年 yuanzhiying. All rights reserved.
//
#import "YSqliteManager.h"
static YSqliteManager *sharedManager = nil;
static dispatch_once_t predicate;
static NSString *APP_DB_IDENTIFIER = @"iOS_DatabaseUpdateDemo";
//數(shù)據(jù)庫(kù)保存目錄
static NSString *databaseFilePath() {
NSArray *paths = NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES);
NSString *directory = [paths objectAtIndex:0];
NSString *dbPath = [directory stringByAppendingPathComponent:[[[NSBundle mainBundle] bundleIdentifier] stringByAppendingString:@".db"]];
NSLog(@"dbPath:%@",dbPath);
[[NSFileManager defaultManager] createDirectoryAtPath:dbPath withIntermediateDirectories:YES attributes:nil error:nil];
return [[dbPath stringByAppendingPathComponent:APP_DB_IDENTIFIER] stringByAppendingString:@".sqlite"];
}
@implementation YSqliteManager
#pragma mark - 初始化
//單例
+ (YSqliteManager *)shareManager {
dispatch_once(&predicate, ^{
sharedManager = [[self alloc] init];
});
return sharedManager;
}
//構(gòu)造方法
- (instancetype)init {
//初始化文件路徑
databasePath = databaseFilePath();
//設(shè)置數(shù)據(jù)庫(kù)版本為1
NSInteger dbVersion = 1;
if (self = [super init]) {
//判斷本地有沒(méi)有數(shù)據(jù)庫(kù)文件
//如果數(shù)據(jù)庫(kù)文件不存在
if (![[NSFileManager defaultManager] fileExistsAtPath:databasePath]) {
//不存在，初始化數(shù)據(jù)庫(kù)
BOOL isCreateSuccess = [self initDatabase];
if (isCreateSuccess) {
//初始化數(shù)據(jù)庫(kù)成功
isOpend = [self openDatabase];
}
} else {
//存在數(shù)據(jù)庫(kù)，獲取版本信息
isOpend = [self openDatabase];
dbVersion = [self getDatabaseVersion];
}
// NSLog(@"log dbVersion:%ld",(long)dbVersion);
/*
根據(jù)數(shù)據(jù)庫(kù)版本號(hào)，逐級(jí)升級(jí)數(shù)據(jù)庫(kù)
之后更新，只需在后面添加case即可（不可包含break，需要逐個(gè)case執(zhí)行升級(jí)!!!）
*/
switch (dbVersion) {
case 1://用戶第一次安裝
{
//將新版本號(hào)存入數(shù)據(jù)庫(kù)
if (![self isExistDatabaseVersion]) {
[self insertDatabaseVersionWithVersion:(dbVersion + 1)];
} else {
[self updateDatabaseVersionWithVersion:(dbVersion + 1)];
}
dbVersion = [self getDatabaseVersion];
}
case 2://修改數(shù)據(jù)庫(kù)結(jié)構(gòu)
{
/*
更新內(nèi)容：1->2
1.user_info表增加age字段
*/
//執(zhí)行對(duì)應(yīng)更新sql語(yǔ)句組
[self updateDatabaseWithNewVersion:dbVersion];
//將新版本號(hào)存入數(shù)據(jù)庫(kù)
if (![self isExistDatabaseVersion]) {
[self insertDatabaseVersionWithVersion:(dbVersion + 1)];
} else {
[self updateDatabaseVersionWithVersion:(dbVersion + 1)];
}
dbVersion = [self getDatabaseVersion];
}
case 3://修改數(shù)據(jù)庫(kù)結(jié)構(gòu)
{
/*
更新內(nèi)容：2->3
1.user_info表增加sex字段
2.增加app_data表
*/
//執(zhí)行對(duì)應(yīng)更新sql語(yǔ)句組
[self updateDatabaseWithNewVersion:dbVersion];
//將新版本號(hào)存入數(shù)據(jù)庫(kù)
if (![self isExistDatabaseVersion]) {
[self insertDatabaseVersionWithVersion:(dbVersion + 1)];
} else {
[self updateDatabaseVersionWithVersion:(dbVersion + 1)];
}
dbVersion = [self getDatabaseVersion];
}
default:
break;
}
}
return self;
}
#pragma mark - 數(shù)據(jù)庫(kù)相關(guān)操作
//打開數(shù)據(jù)庫(kù)
- (BOOL)openDatabase {
if (!isOpend) {
if (sqlite3_open([databasePath UTF8String], &database) != SQLITE_OK) {
sqlite3_close(database);
} else {
isOpend = YES;
}
}
return isOpend;
}
//關(guān)閉數(shù)據(jù)庫(kù)
- (void)closeDatabase {
if (isOpend) {
sqlite3_close(database);
isOpend = NO;
}
}
//初始化數(shù)據(jù)庫(kù)
- (BOOL)initDatabase {
NSString *path = [[NSBundle mainBundle] pathForResource:[NSString stringWithFormat:@"sql_default"] ofType:@"plist"];
NSArray *array = [[NSArray alloc] initWithContentsOfFile:path];
if (array != nil) {
if (sqlite3_open([databasePath UTF8String], &database) != SQLITE_OK) {
sqlite3_close(database);
return NO;
} else {
char *errorMsg = nil;
for (int i = 0; i < array.count; i ++) {
if (sqlite3_exec(database, [[array objectAtIndex:i] UTF8String], nil, nil, &errorMsg) != SQLITE_OK) {
sqlite3_free(errorMsg);
return NO;
}
sqlite3_free(errorMsg);
}
}
} else {
return NO;
}
return YES;
}
//執(zhí)行更新sql語(yǔ)句組
- (void)updateDatabaseWithNewVersion:(NSInteger)_newversion {
NSString *path = [[NSBundle mainBundle] pathForResource:[NSString stringWithFormat:@"sql_update_%ld",(long)_newversion] ofType:@"plist"];
if ([[NSFileManager defaultManager] fileExistsAtPath:path]) {
NSArray *array = [[NSArray alloc] initWithContentsOfFile:path];
if (array != nil) {
if (sqlite3_open([databasePath UTF8String], &database) != SQLITE_OK) {
sqlite3_close(database);
return;
} else {
char *errorMsg = nil;
for (int i = 0; i < array.count; i ++) {
if (sqlite3_exec(database, [[array objectAtIndex:i] UTF8String], nil, nil, &errorMsg) != SQLITE_OK) {
sqlite3_free(errorMsg);
return;
}
sqlite3_free(errorMsg);
}
}
}
}
}
#pragma mark - 數(shù)據(jù)庫(kù)版本號(hào) 操作
//是否存在版本號(hào)數(shù)據(jù)
- (BOOL)isExistDatabaseVersion {
@synchronized(sharedManager) {
BOOL state = NO;
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "SELECT COUNT(db_version_num) FROM db_version";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
if (sqlite3_step(statement) == SQLITE_ROW) {
if (sqlite3_column_int(statement, 0) == 1) {
state = YES;
}
}
}
sqlite3_finalize(statement);
}
return state;
}
}
//插入數(shù)據(jù)庫(kù)版本號(hào)
- (void)insertDatabaseVersionWithVersion:(NSInteger)_dbVersion {
@synchronized(sharedManager) {
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "INSERT INTO db_version(db_version_num) VALUES(?)";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
sqlite3_bind_int64(statement, 1, _dbVersion);
if (sqlite3_step(statement) == SQLITE_DONE) {
// NSLog(@"插入成功");
}else{
// NSLog(@"插入失敗");
}
}
sqlite3_finalize(statement);
}
}
}
//更新數(shù)據(jù)庫(kù)新版本號(hào)
- (void)updateDatabaseVersionWithVersion:(NSInteger)_dbVersion {
@synchronized(sharedManager) {
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "UPDATE db_version SET db_version_num=?";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
sqlite3_bind_int64(statement, 1, _dbVersion);
if (sqlite3_step(statement) == SQLITE_DONE) {
//更新成功
}
}
sqlite3_finalize(statement);
}
}
}
//獲取數(shù)據(jù)庫(kù)版本號(hào)
- (NSInteger)getDatabaseVersion {
@synchronized(sharedManager) {
NSInteger db_version_num = 0;
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "SELECT db_version_num FROM db_version";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
if (sqlite3_step(statement) == SQLITE_ROW) {
db_version_num = sqlite3_column_int(statement, 0);
}
}
sqlite3_finalize(statement);
}
return db_version_num;
}
}
#pragma mark - 所有表的操作
#pragma mark 用戶表
//是否存在用戶
- (BOOL)isExistUserWithUserID:(NSInteger)_userid {
@synchronized(sharedManager) {
BOOL state = NO;
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "SELECT COUNT(user_id) FROM user_info WHERE user_id=?";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
sqlite3_bind_int64(statement, 1, _userid);
if (sqlite3_step(statement) == SQLITE_ROW) {
if (sqlite3_column_int(statement, 0) == 1) {
state = YES;
}
}
}
sqlite3_finalize(statement);
}
return state;
}
}
//插入
- (void)insertUserInfoWithUserInfo:(YUserData *)_userinfo {
@synchronized(sharedManager) {
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "INSERT INTO user_info (user_id,user_name,age,sex) VALUES(?,?,?,?)";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
sqlite3_bind_int64(statement, 1, _userinfo.userID);
sqlite3_bind_text(statement, 2, [_userinfo.userName UTF8String], -1, SQLITE_TRANSIENT);
sqlite3_bind_int64(statement, 3, _userinfo.age);
sqlite3_bind_text(statement, 4, [_userinfo.sex UTF8String], -1, SQLITE_TRANSIENT);
if (sqlite3_step(statement) == SQLITE_DONE) {
}
}
sqlite3_finalize(statement);
}
}
}
//更新
- (void)updateUserInfoWithUserInfo:(YUserData *)_userinfo UserID:(NSInteger)_userid {
@synchronized(sharedManager) {
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "UPDATE user_info set user_id=?,user_name=?,age=?,sex=? WHERE user_id=?";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
sqlite3_bind_int64(statement, 1, _userinfo.userID);
sqlite3_bind_text(statement, 2, [_userinfo.userName UTF8String], -1, SQLITE_TRANSIENT);
sqlite3_bind_int64(statement, 3, _userinfo.age);
sqlite3_bind_text(statement, 4, [_userinfo.sex UTF8String], -1, SQLITE_TRANSIENT);
sqlite3_bind_int64(statement, 5, _userinfo.userID);
if (sqlite3_step(statement) == SQLITE_DONE) {
}
}
sqlite3_finalize(statement);
}
}
}
//獲取用戶數(shù)據(jù)
- (YUserData *)getUserInfoWithUserID:(NSInteger)_userid {
@synchronized(sharedManager) {
YUserData *tempUserInfo = nil;
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "SELECT user_id,user_name,age,sex FROM user_info WHERE user_id=?";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
sqlite3_bind_int64(statement, 1, _userid);
while (sqlite3_step(statement) == SQLITE_ROW) {
tempUserInfo = [[YUserData alloc] init];
tempUserInfo.userID = sqlite3_column_int(statement, 0);
char *username = (char*)sqlite3_column_text(statement, 1);
if (username != nil) {
tempUserInfo.userName = [NSString stringWithUTF8String:username];
username = nil;
}
tempUserInfo.age = sqlite3_column_int(statement, 2);
char *sex = (char*)sqlite3_column_text(statement, 3);
if (sex != nil) {
tempUserInfo.sex = [NSString stringWithUTF8String:sex];
sex = nil;
}
}
}
sqlite3_finalize(statement);
}
return tempUserInfo;
}
}
//刪除用戶數(shù)據(jù)
- (void)deleteUserInfoDataWithUserID:(NSInteger)_userid {
@synchronized(sharedManager) {
if (isOpend) {
sqlite3_stmt *statement;
const char *sqlQuery = "DELETE FROM user_info WHERE user_id=?";
if (sqlite3_prepare_v2(database, sqlQuery, -1, &statement, nil) == SQLITE_OK) {
sqlite3_bind_int64(statement, 1, _userid);
if (sqlite3_step(statement) == SQLITE_DONE) {
}
}
sqlite3_finalize(statement);
}
}
}
@end
QQ20170215-174043@2x.png
