数据存储：FMDB-模型进行存取

写这个Blog的目的是为了记录的学习经历，代码写的不是很好，但愿读者理解。 平常对数据库的操做，须要设计字段，感受太繁琐了，想找个简化的方法，就想经过这样进行实现，这也就是一种思考方式。实际实现仍是看需求。具体看代码： 一、导入FMDB

import FMDB
复制代码

二、使用单利模式和必要属性

static let defaultManger = WDDataBaseManager()
typealias successBlock = () ->Void
typealias failBlock = () ->Void
private var tableName:String?
复制代码

三、懒加载建立数据库（名称我随便取得）

lazy var fmdb:FMDatabase = {
        let path = NSHomeDirectory().appending("/Documents/testDB.db")
        let db = FMDatabase(path: path)
        return db
    }()
复制代码

四、建立表

func creatTable(tableName:String) -> Void {
        fmdb.open()
        self.tableName = tableName
        let creatSql = "create table if not exists \(tableName) (id integer primary key autoincrement,model BLOB)"
        let result = fmdb.executeUpdate(creatSql, withArgumentsIn:[])
        if result{
            print("建立表成功")
        }
    }
复制代码

五、删除表

func dropTable() -> Void {
        let sql = "drop table if exists " + tableName!
        let result = fmdb.executeUpdate(sql, withArgumentsIn:[])
        if result{
            print("删除表成功")
        }
    }
复制代码

六、插入数据

func insert(model:NSObject, successBlock: successBlock, failBlock: failBlock) -> Void {
        let modelData = try! NSKeyedArchiver.archivedData(withRootObject: model, requiringSecureCoding: false)
        let insertSql = "insert into " + tableName! + " (model) values(?)"
        do {
            try fmdb.executeUpdate(insertSql, values: [modelData])
            successBlock()
        } catch {
            print(fmdb.lastError())
            failBlock()
        }
    }
复制代码

七、更新表

func update(model:NSObject,uid:Int32, successBlock: successBlock, failBlock: failBlock) -> Void {
        let modelData = try! NSKeyedArchiver.archivedData(withRootObject: model, requiringSecureCoding: false)
        let updateSql = "update " + tableName! + " set model = ? where id = ?"
        do {
            try fmdb.executeUpdate(updateSql, values: [modelData, uid])
            successBlock()
        } catch {
            print(fmdb.lastError())
            failBlock()
        }
    }
复制代码

八、查询数据（这是查询全部数据，其余按需求设计）

func selectAll() -> [NSObject] {
        var tmpArr = [NSObject]()
        let selectSql = "select * from " + tableName!
        do {
            let rs = try fmdb.executeQuery(selectSql, values:nil)
            while rs.next() {
                var model = NSObject()
                let modelData  = rs.data(forColumn:"model")
                let id = rs.int(forColumn: "id")
                model = try! NSKeyedUnarchiver.unarchiveTopLevelObjectWithData(modelData!) as! NSObject
                model.wd_fmdb_id = id
                tmpArr.append(model)
            }
        } catch {
            print(fmdb.lastError())
        }
        return tmpArr
    }
复制代码

九、删除数据（这是根据Id删除的）

func delete(uid:Int32, successBlock: successBlock, failBlock: failBlock) -> Void {
        let deleteSql = "delete from " + tableName! + " where id = ?"
        do {
            try fmdb.executeUpdate(deleteSql, values: [uid])
            successBlock()
        } catch {
            print(fmdb.lastError())
            failBlock()
        }
    }
复制代码

十、新增分类添加默认Id

private var wd_id_key: String = "key"

extension NSObject {
    open var wd_fmdb_id:Int32? {
        get {
            return (objc_getAssociatedObject(self, &wd_id_key) as? Int32)
        } set(newValue) {
            objc_setAssociatedObject(self, &wd_id_key, newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_ASSIGN)
        }
    }
}
复制代码

十一、使用方法

//1. 在Appdelegate中打开数据库
func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
        DataBaseManager.defaultManger.creatTable()
        return true
 }
//二、添加数据
WDDataBaseManager.defaultManger.insert(model: model, successBlock: {
                print("成功")
            }, failBlock: {
                 print("失败")
            })
//三、修改数据
WDDataBaseManager.defaultManger.update(model: model, uid: self.model!.wd_fmdb_id!, successBlock: {
               print("成功")
            }, failBlock: {
                print("失败")
            })
//四、删除数据
WDDataBaseManager.defaultManger.delete(uid: model.wd_fmdb_id!, successBlock: {
                print("成功")
            }, failBlock: {
                print("失败")
            })
// 五、查询数据
WDDataBaseManager.defaultManger.selectAll()
复制代码

• 遵循NSCoding 重写三个方法

import UIKit

class Model: NSObject , NSCoding{

    public var name:String?
    public var phone:String?
    public var addres:String?
    
    override init() {
        super.init()
    }
    
    func encode(with aCoder: NSCoder) {
        aCoder.encode(name, forKey: "name")
        aCoder.encode(phone, forKey: "phone")
        aCoder.encode(addres, forKey: "addres")
    }
    
    required init?(coder aDecoder: NSCoder) {
        super.init()
        self.name = aDecoder.decodeObject(forKey: "name") as? String
        self.addres = aDecoder.decodeObject(forKey: "addres") as? String
        self.phone = aDecoder.decodeObject(forKey: "phone") as? String
    }
}
复制代码

十二、效果图

1四、Demo地址:https://github.com/wudan-ios/FMDB-Model.git

---

补充Objective-C版本 一、接口文件

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

@interface NSObject (addId)

@property (nonatomic) NSInteger wd_fmdb_id;

@end

typedef void(^successBlock)(void);
typedef void(^failBlock)(void);

@interface WDDataBaseManager : NSObject

+ (instancetype)manager;
/** 建立表 */
- (void)createTableWithName:(NSString *)name;
/** 删除表 */
- (void)dropTable;
/** 添加数据 */
- (void)insertDataWithModel:(NSObject *)model successBlock:(successBlock)aSuccessBlock failBlock:(failBlock)aFailBlock;
/** 更新数据 */
- (void)updateDataWithModel:(NSObject *)model uid:(NSInteger)aUid successBlock:(successBlock)aSuccessBlock failBlock:(failBlock)aFailBlock;
/** 删除数据 */
- (void)deleteDataWithUid:(NSInteger)uid successBlock:(successBlock)aSuccessBlock failBlock:(failBlock)aFailBlock;
/** 查询所有数据 */
- (NSArray<NSObject *> *)queryAllData;

@end
NS_ASSUME_NONNULL_END
复制代码

二、实现文件

#import "WDDataBaseManager.h"
#import <FMDB.h>
#import <objc/runtime.h>


static char *wd_id_key = "key";
@implementation NSObject (addId)

- (NSInteger)wd_fmdb_id {
    NSNumber *numberValue = objc_getAssociatedObject(self, wd_id_key);
    return numberValue.integerValue;
}

- (void)setWd_fmdb_id:(NSInteger)wd_fmdb_id {
    objc_setAssociatedObject(self, wd_id_key, @(wd_fmdb_id), OBJC_ASSOCIATION_ASSIGN);
}

@end

@interface WDDataBaseManager ()

@property (nonatomic, strong) FMDatabase *fmdb;
@property (nonatomic, copy) NSString *tableName;


@end

@implementation WDDataBaseManager

+ (instancetype)manager {
    static dispatch_once_t onceToken;
    static WDDataBaseManager *manager = nil;
    dispatch_once(&onceToken, ^{
        manager = [[WDDataBaseManager alloc] init];
    });
    return manager;
}

- (void)createTableWithName:(NSString *)name {
    [self.fmdb open];
    self.tableName = name;
    NSString *sql = [NSString stringWithFormat:@"create table if not exists %@ (id integer primary key autoincrement,model BLOB)", name];
    BOOL result = [self.fmdb executeUpdate:sql];
    if (result) {
        NSLog(@"表建立成功");
    }
}

- (void)dropTable {
    NSString *sql = [NSString stringWithFormat:@"drop table if exists %@", self.tableName];
    BOOL result = [self.fmdb executeUpdate:sql];
    if (result) {
        NSLog(@"表删除成功");
    }
}

- (void)insertDataWithModel:(NSObject *)model successBlock:(successBlock)aSuccessBlock failBlock:(failBlock)aFailBlock {
    NSData *modelData = [NSKeyedArchiver archivedDataWithRootObject:model];
    NSString *sql = [NSString stringWithFormat:@"insert into %@ (model) values(?)", self.tableName];
    BOOL result = [self.fmdb executeUpdate:sql withArgumentsInArray:@[modelData]];
    if (result) {
        aSuccessBlock();
    } else {
        aFailBlock();
    }
}

- (void)updateDataWithModel:(NSObject *)model uid:(NSInteger)aUid successBlock:(successBlock)aSuccessBlock failBlock:(failBlock)aFailBlock {
    NSData *modelData = [NSKeyedArchiver archivedDataWithRootObject:model];
    NSString *sql = [NSString stringWithFormat:@"update %@ set model = ? where id = ?", self.tableName];
    BOOL result = [self.fmdb executeUpdate:sql withArgumentsInArray:@[modelData, @(aUid)]];
    if (result) {
        aSuccessBlock();
    } else {
        aFailBlock();
    }
}

- (void)deleteDataWithUid:(NSInteger)uid successBlock:(successBlock)aSuccessBlock failBlock:(failBlock)aFailBlock {
    NSString *sql = [NSString stringWithFormat:@"delete from %@ where id = ?", self.tableName];
    BOOL result = [self.fmdb executeUpdate:sql withArgumentsInArray:@[@(uid)]];
    if (result) {
        aSuccessBlock();
    } else {
        aFailBlock();
    }
}

- (NSArray<NSObject *> *)queryAllData {
    NSMutableArray *array = [NSMutableArray array];
    NSString *sql = [NSString stringWithFormat:@"select * from %@",self.tableName];
    @try {
        FMResultSet *rs = [self.fmdb executeQuery:sql];
        while (rs.next) {
            NSObject *model = [[NSObject alloc] init];
            NSData *modelData = [rs dataForColumn:@"model"];
            int uid = [rs intForColumn:@"id"];
            
            model = [NSKeyedUnarchiver unarchiveObjectWithData:modelData];
            model.wd_fmdb_id = uid;
            [array addObject:model];
        }
    } @catch (NSException *exception) {
        NSLog(@"%@", self.fmdb.lastError);
    } @finally {
        NSLog(@"查询数据");
    }
    
   return array;
}

- (FMDatabase *)fmdb {
    if (!_fmdb) {
        NSString *path = [NSHomeDirectory() stringByAppendingString:@"/Documents/testDB.db"];
        _fmdb = [[FMDatabase alloc] initWithPath:path];
    }
    return _fmdb;
}

@end
复制代码

三、使用方法

// 一、打开数据库，并建立表
- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions {
    
    [[WDDataBaseManager manager] createTableWithName:@"test"];
    
    return YES;
}
// 二、添加数据
 [[WDDataBaseManager manager] insertDataWithModel:model successBlock:^{
        // TODO:
    } failBlock:^{
        // TODO:
    }];
// 三、查询数据
[[WDDataBaseManager manager] queryAllData]
// 四、修改数据
[[WDDataBaseManager manager] updateDataWithModel:model uid:self.model.wd_fmdb_id successBlock:^{
        // TODO:
    } failBlock:^{
         // TODO:
    }];
// 五、删除数据
[[WDDataBaseManager manager] deleteDataWithUid:model.wd_fmdb_id successBlock:^{
            // TODO:
        } failBlock:^{
            // TODO:
        }];
// 六、删除表
[[WDDataBaseManager manager] dropTable];
复制代码

• 插入表中的Model须要遵循(实例代码)

- (void)encodeWithCoder:(nonnull NSCoder *)aCoder {
    [aCoder encodeObject:self.name forKey:@"name"];
    [aCoder encodeObject:self.phone forKey:@"phone"];
    [aCoder encodeObject:self.address forKey:@"address"];
}

- (nullable instancetype)initWithCoder:(nonnull NSCoder *)aDecoder {
    if (self == [super init]) {
        self.name = [aDecoder decodeObjectForKey:@"name"] ;
        self.phone = [aDecoder decodeObjectForKey:@"phone"];
        self.address = [aDecoder decodeObjectForKey:@"address"];
    }
    return self;
}
复制代码