FMDB的二次封装
再封装目的狠明确，就是用的时候更好用。 其实说白了，数据库从大了来说就是增删改查，而我们在项目中用到的基本也就是增删改查的逻辑，简单的封装了下：我的思路就是创建一个单例类，专门用来管理这些接口，具体代码看下边：
YaoFMDBManager.h
#import
typedef void(^getDataByPrimaryKey)(NSData *data);
typedef void(^getAllData)(NSArray *dataArr);
@interface YaoFMDBManager : NSObject
+ (YaoFMDBManager *)sharedInstance;
//add the data by primaryKey in table
- (BOOL)addDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey data:(NSData *)data;
//delete the data by primaryKey in table
- (BOOL)deleteDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey;
//update data by primaryKey in table
- (BOOL)updateDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey data:(NSData *)data;
//read the data by primaryKey in table
- (BOOL)readDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey data:(getDataByPrimaryKey)dataBlock;
//read all the data in table
- (BOOL)readAllDataWithTableName:(NSString *)tableName dataArr:(getAllData)allDataBlock;
//clear the dataBase
- (BOOL)clearDataBaseWithTableName:(NSString *)tableName;
@endYaoFMDBManager.m
#import "YaoFMDBManager.h"
#import "FMDatabase.h"
#define CREAT_TABLE_IFNOT_EXISTS @"create table if not exists %@ (key text primary key, data blob)"
#define DELETE_DATA_WITH_PRIMARYKEY @"delete from %@ where key = ?"
#define INSERT_TO_TABLE @"insert into %@ (key, data) values (?, ?)"
#define READ_DATA_TABLE_WITH_PRIMARYKEY @"select data from %@ where key = ?"
#define READ_ALL_DATA @"select data from %@"
#define UPDATE_DATA_WHTH_PRIMARYKEY @"update %@ set data = ? where key = ?"
#define CLEAR_ALL_DATA @"DELETE FROM %@"
@implementation YaoFMDBManager
+ (YaoFMDBManager *)sharedInstance {
static YaoFMDBManager *YaoManager = nil;
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
YaoManager = [[YaoFMDBManager alloc] init];
});
return YaoManager;
}
- (BOOL)addDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey data:(NSData *)data {
BOOL ret = false;
//get the dB path
NSString *dbPath = [self getDbPath];
//init a dataBase
FMDatabase *dataBase = [FMDatabase databaseWithPath:dbPath];
if ([dataBase open]) {
NSString *sql = [NSString stringWithFormat:CREAT_TABLE_IFNOT_EXISTS,tableName];
ret = [dataBase executeUpdate:sql];
if (ret) {
//make sure the primaryKey is uinque.
NSString *deleteSql = [NSString stringWithFormat:DELETE_DATA_WITH_PRIMARYKEY,tableName];
ret = [dataBase executeUpdate:deleteSql,primaryKey];
if (ret) {
NSString *storeURL = [NSString stringWithFormat:INSERT_TO_TABLE,tableName];
ret = [dataBase executeUpdate:storeURL,primaryKey,data];
}
}
}
[dataBase close];
return ret;
}
- (BOOL)deleteDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey {
BOOL ret = false;
NSString *dbPath = [self getDbPath];
FMDatabase *dataBase = [FMDatabase databaseWithPath:dbPath];
if ([dataBase open]) {
NSString *sql = [NSString stringWithFormat:CREAT_TABLE_IFNOT_EXISTS,tableName];
ret = [dataBase executeUpdate:sql,primaryKey];
if (ret) {
NSString *deleteSql = [NSString stringWithFormat:DELETE_DATA_WITH_PRIMARYKEY,tableName];
ret = [dataBase executeUpdate:deleteSql,primaryKey];
}
}
[dataBase close];
return ret;
}
- (BOOL)updateDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey data:(NSData *)data {
BOOL ret = false;
NSString *dbPath = [self getDbPath];
FMDatabase *dataBase = [FMDatabase databaseWithPath:dbPath];
if ([dataBase open]) {
NSString *sql = [NSString stringWithFormat:CREAT_TABLE_IFNOT_EXISTS,tableName];
ret = [dataBase executeUpdate:sql,primaryKey];
if (ret) {
NSString *updateSql = [NSString stringWithFormat:UPDATE_DATA_WHTH_PRIMARYKEY,tableName];
ret = [dataBase executeUpdate:updateSql,data,primaryKey];
}
}
[dataBase close];
return ret;
}
- (BOOL)readDataWithTableName:(NSString *)tableName primaryKey:(NSString *)primaryKey data:(getDataByPrimaryKey)dataBlock {
BOOL ret = false;
NSString *dataPath = [self getDbPath];
FMDatabase *dB = [FMDatabase databaseWithPath:dataPath];
if ([dB open]) {
NSString *creatSql = [NSString stringWithFormat:CREAT_TABLE_IFNOT_EXISTS,tableName];
ret = [dB executeUpdate:creatSql];
if (ret) {
NSString *readSql = [NSString stringWithFormat:READ_DATA_TABLE_WITH_PRIMARYKEY,tableName];
FMResultSet *resultSet = [dB executeQuery:readSql,primaryKey];
while ([resultSet next]) {
NSData *data = [resultSet dataForColumn:@"data"];
dataBlock(data);
}
}
}
[dB close];
return ret;
}
- (BOOL)readAllDataWithTableName:(NSString *)tableName dataArr:(getAllData)allDataBlock {
BOOL ret = false;
NSString *dbPath = [self getDbPath];
NSMutableArray *dataArr = [NSMutableArray array];
FMDatabase *dataBase = [FMDatabase databaseWithPath:dbPath];
if ([dataBase open]) {
NSString *sql = [NSString stringWithFormat:CREAT_TABLE_IFNOT_EXISTS,tableName];
ret = [dataBase executeUpdate:sql];
if (ret) {
NSString *readSql = [NSString stringWithFormat:READ_ALL_DATA,tableName];
FMResultSet *result = [dataBase executeQuery:readSql];
while ([result next]) {
NSData *data = [result dataForColumn:@"data"];
[dataArr addObject:data];
}
allDataBlock(dataArr);
}
}
[db close];
return ret;
}
- (BOOL)clearDataBaseWithTableName:(NSString *)tableName{
BOOL ret = false;
NSString *dbPath = [self getDbPath];
FMDatabase *dataBase = [FMDatabase databaseWithPath:dbPath];
if ([dataBase open]) {
NSString *clearSql = [NSString stringWithFormat:CLEAR_ALL_DATA,tableName];
ret = [dataBase executeUpdate:clearSql];
}
[dataBase close];
return ret;
}
#pragma mark - private function
- (NSString *)getDbPath {
NSString *doc= [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) lastObject];
return [doc stringByAppendingString:@"data.sqlite"];
}
@end
FMDBManager的使用：
NSDictionary *map = [NSDictionary dictionaryWithObjectsAndKeys:@[@111.222,@333.444],@"上海", @[@112.111,@222.222],@"重庆",@[@333.444,@555.666],@"北京",nil];
NSArray *arr = @[@"123",@"234"];
NSArray *updateArr = @[@"888888",@"9999999"];
NSData *dicData = [NSJSONSerialization dataWithJSONObject:map options:NSJSONWritingPrettyPrinted error:nil];
NSData *arrData = [NSJSONSerialization dataWithJSONObject:arr options:NSJSONWritingPrettyPrinted error:nil];
NSData *updateArrData = [NSJSONSerialization dataWithJSONObject:updateArr options:NSJSONWritingPrettyPrinted error:nil];
//增加一条
[[YaoFMDBManager sharedInstance] addDataWithTableName:@"student" primaryKey:@"yao" data:arrData];
// 再增加一条
[[YaoFMDBManager sharedInstance] addDataWithTableName:@"student" primaryKey:@"tian" data:dicData];
//删除primary key为‘yao’的数据
[[YaoFMDBManager sharedInstance] deleteDataWithTableName:@"student" primaryKey:@"yao"];
//更新primary key为‘yao’的数据
[[YaoFMDBManager sharedInstance] updateDataWithTableName:@"student" primaryKey:@"yao" data:updateArrData];
//读取primary key为'yao'的数据
__block NSArray *dataArr = [NSArray array];
[[YaoFMDBManager sharedInstance] readDataWithTableName:@"student" primaryKey:@"yao" data:^(NSData *data) {
dataArr = [NSJSONSerialization JSONObjectWithData:data options:NSJSONReadingMutableContainers error:nil];
}];
//读取primary key为'tian'的数据
__block NSDictionary *dict = [NSDictionary dictionary];
[[YaoFMDBManager sharedInstance] readDataWithTableName:@"student" primaryKey:@"tian" data:^(NSData *data) {
dict = [NSJSONSerialization JSONObjectWithData:data options:NSJSONReadingMutableLeaves error:nil];
}];
//读取整个表的数据
__block NSMutableArray *allArr = [NSMutableArray array];
[[YaoFMDBManager sharedInstance] readAllDataWithTableName:@"student" dataArr:^(NSArray *dataArr) {
for (int i = 0; i