Swift

[Swift]在swift中使用FMDB

2016-11-24  本文已影响548人  流火绯瞳

FMDBOC中使用较多, 来到swift的世界,依然不想放弃他,下面就来看一下如何在swift中使用;

说明: 因为文章中的所有代码块都是完整的功能,可直接使用,所以导致文章篇幅过长, 其实内容不是很多,可直接下载demo:LQQZYY/FMDB-Swift, 线程安全的操作数据库:FMDB-Swift-Queue,需要修改数据模型及表的字段为自己的需求才能使用, 格式可保持不变.

1. 准备

首先,从github-FMDB下载其源文件,解压后将src文件夹下的fmdb添加到项目中,然后,将extra文件夹下的Swift extensions也添加进来;
然后,在项目Build PhasesLink Binary With Libraries添加系统依赖库 libsqlite3.tbd
最后,添加桥接文件,如果你不会添加,请参考[Swift]创建Objective-C混编桥接文件
在桥接文件中导入头文件:

#import "FMDB.h"

编译一下,这时你会发现Swift extensions文件夹下的文件会报错:


应该是还没有升级到swift3.0,根据提示,自动矫正; 系统提示的矫正完后,还会有两个下面这样的错误:

我的做法是将函数的返回值类型NSData! 改为 Data!,即:


另一个错误做一样的更改.
再编译,应该就没有错误了,好了,准备完毕!!!

2. 使用

使用和OC很相似,只是写法上的不同, 需要注意的是,怎么创建单例的FMDatabase实例对象, 我这里使用的是懒加载的模式:

private static var database: FMDatabase = {
        
        if LZSqlite.path == nil || (LZSqlite.path?.characters.count)! <= 0 {
            
            LZSqlite.createSqliteWithName("myDB.db")
        }
        
        let db = FMDatabase.init(path: LZSqlite.path!)
        return db!
    }()

这里的path是数据库的保存路径

private static var path: String!

我没有把这个属性直接让外部访问,是因为,我单独写了一个方法,来创建数据库文件:

static func createSqliteWithName(_ name: String) {
        
        var fileName = ""
        let strArr = name.components(separatedBy: ".")
        
        if strArr.last == "sqlite" || strArr.last == "db" {
            
            fileName = name
        } else {
            
            fileName = name + ".db"
        }
        
        let path = NSHomeDirectory() + "/Documents/" + fileName
        
        LZSqlite.path = path
        
        let fm = FileManager.default
        
        if fm.fileExists(atPath: path) == false {
            
            fm.createFile(atPath: path, contents: nil, attributes: nil)
        }
        
        print("dataBase path: \(LZSqlite.path)")
    }

数据库文件创建好了,下面就是建表了,因为每个项目中表的格式都不一样,需求多样,这里只是示例创建方法,真正使用的时候根据自己的额需求来设计即可:

static func createTable(_ name: String) {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            
            return
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        
        if LZSqlite.database.tableExists(name) == false {
            
            let create = "CREATE TABLE IF NOT EXISTS '\(name)' (ID TEXT UNIQUE NOT NULL, groupName TEXT, groupID TEXT NOT NULL, nickName TEXT, userName TEXT, psw TEXT, urlString TEXT, email TEXT, dsc TEXT)"
            
            do {
                try LZSqlite.database.executeUpdate(sql: create)
            } catch {
                
                print("Error: table \(name) create fail")
            }
        }
        
        LZSqlite.database.close()
    }

这里写SQL语句的时候需要注意, 占位符 ?在这里好像不能使用,至少我使用的时候是不行的, 因为swift中插入字符串真的是十分的方便, 我就直接使用这种简单而又清楚的方式来拼接SQL语句;
表的删除,可使用这个方法:

static func deleteTable(_ name: String) {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(name) {
            
            let drop = "DROP TABLE '\(name)'"

            do {
                try LZSqlite.database.executeUpdate(sql: drop)
            } catch  {
                
                print("Error: table \(name) drop failed")
            }
        }

下面这个方法,是为已有的表添加一列元素:

static func alterElement(_ element: String, toTable table: String) {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            do {
                try LZSqlite.database.executeUpdate(sql: "ALTER TABLE '\(table)' ADD '\(element)' TEXT")
            } catch {
                
                print("Error: alter new element: \"\(element)\" to table: \"\(table)\" failed")
            }
        }
        
        LZSqlite.database.close()
    }

表建好了,接下来就是 / / /** 查** 等操作了:

2.1 增

直接上代码了:

static func insert(model: LZDataModel, toTable table: String) {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            let insert = "INSERT INTO '\(table)' (ID, groupName, nickName, userName, psw, urlString, email, dsc, groupID) VALUES ('\(model.identifier)', '\(model.groupName)', '\(model.nickName)', '\(model.userName)', '\(model.password)', '\(model.urlString)', '\(model.email)', '\(model.dsc)', '\(model.groupID)')"
            
            do {
                try LZSqlite.database.executeUpdate(sql: insert)
            } catch {
                
                print("Error: insert new model into table:\(table) failed")
            }
        }
        
        LZSqlite.database.close()
    }

这是往表内添加一条数据,model是自定义的;

**需要注意的是: **如果model的属性是可选型, 那么存入数据库的也是可选类型, 这样在执行查询/修改等操作的时候, 稍不注意就会出现问题, 例如: 如果我们使用ID来更新某条数据,你传进来的ID参数多数情况下不是可选型,而是真真正正的字符串, 而你存进数据库的是可选型这样是找不到这条数据的, 所以在存入数据库前, 要对属性的可选型进行解包;


如上图所示:
下面十条数据,是我没有对可选型属性进行解包,直接存储的;
上面的数据,是我解包后存储的;
其中的区别很明显, 为了使用方便, 个人建议解包后再存储, 虽然解包的过程会有些繁琐.
下面这个方法, 是我加入了对可选型进行解包的操作:

 static func insertOptional(model: LZDataModel, toTable table: String) {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            
            var identifier: String = ""
            var groupName: String = ""
            var nickName: String = ""
            var userName: String = ""
            var password: String = ""
            var urlString: String = ""
            var email: String = ""
            var dsc: String = ""
            var groupID: String = ""
            
            if let tmp = model.identifier {
                
                identifier = tmp
            }
            
            if let tmp = model.groupName {
                groupName = tmp
            }
            
            if let tmp = model.nickName {
                nickName = tmp
            }
            
            if let tmp = model.userName {
                userName = tmp
            }
            
            if let tmp = model.password {
                password = tmp
            }
            
            if let tmp = model.urlString {
                urlString = tmp
            }
            
            if let tmp = model.email {
                email = tmp
            }
            
            if let tmp = model.dsc {
                dsc = tmp
            }
            
            if let tmp = model.groupID {
                groupID = tmp
            }
            
            let insert = "INSERT INTO '\(table)' (ID, groupName, nickName, userName, psw, urlString, email, dsc, groupID) VALUES ('\(identifier)', '\(groupName)', '\(nickName)', '\(userName)', '\(password)', '\(urlString)', '\(email)', '\(dsc)', '\(groupID)')"
            
            do {
                try LZSqlite.database.executeUpdate(sql: insert)
            } catch {
                
                print("Error: insert new model into table:\(table) failed")
            }
        }
        
        LZSqlite.database.close()
    }

因为数据模型一般都是我们定义好的,不会有变化,所以我们知道哪些属性是可选型,哪些不是,所以我们只需要判断其是否有值即可, 直接这样写是没有问题的;

2.2 删

删除的操作就比较简单了:

static func delete(model: LZDataModel, fromTable table: String) {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            do {
                try LZSqlite.database.executeUpdate(sql: "DELETE FROM '\(table)' WHERE ID = '\(model.identifier)'")
            } catch  {
                
                print("Error: delete model from table: \"\(table)\" failed")
            }
        }
        
        LZSqlite.database.close()
    }
2.3 改

修改的操作,情况相对较多,如果你能确定修改的内容, 可以只修改需要修改的,如果不能确定, 就将整条数据更新一下, 这里,我是更新的整条数据, 当然,如果你能确定某个属性一旦存储就不会改变,例如ID, 可以不用更改:

static func update(model: LZDataModel, inTable table: String) {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            let update = "UPDATE '\(table)' SET userName = '\(model.userName)', psw = '\(model.password)', nickName = '\(model.userName)', groupName = '\(model.groupName)', dsc = '\(model.dsc)', urlString = '\(model.urlString)', groupID = '\(model.groupID)', email = '\(model.email)' WHERE ID = '\(model.identifier)'"
            
            do {
                try LZSqlite.database.executeUpdate(sql: update)
            } catch  {
                
                print("Error: update table: \"\(table)\" failed")
            }
        }
        
        LZSqlite.database.close()
    }
2.4 查

查询的情况也挺多: 查询全部数据, 查询某条数据, 查询某个区间的数据, 查询数据库中数据的数目等;
这里直接给出代码:
查询所有数据:

static func selectAllFromTable(_ table: String) -> [LZDataModel]? {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return nil
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            let select = "SELECT * FROM '\(table)'"
            do {
                let fs = try LZSqlite.database.executeQuery(sql: select)
                
                var tempModels = [LZDataModel]()
                
                while fs.next() {
                    
                    let model = LZDataModel()
                    model.identifier = fs.string(forColumn: "ID")
                    model.nickName = fs.string(forColumn: "nickName")
                    model.userName = fs.string(forColumn: "userName")
                    model.password = fs.string(forColumn: "psw")
                    model.urlString = fs.string(forColumn: "urlString")
                    model.dsc = fs.string(forColumn: "dsc")
                    model.groupName = fs.string(forColumn: "groupName")
                    model.groupID = fs.string(forColumn: "groupID")
                    model.email = fs.string(forColumn: "email")
                    
                    tempModels.append(model)
                }
                
                fs.close()
                LZSqlite.database.close()
                return tempModels
            } catch  {
                
                print("Error: select models from table: \"\(table)\" failed")
            }
            
        }
        
        LZSqlite.database.close()
        return nil
    }

查询某条数据:

static func selectModelWithID(_ identifier: String, fromTable table: String) -> LZDataModel? {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return nil
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            do {
                let fs = try LZSqlite.database.executeQuery(sql: "SELECT * FROM '\(table)' WHERE ID = '\(identifier)'")
                if fs.next() {
                    
                    let model = LZDataModel()
                    model.identifier = fs.string(forColumn: "ID")
                    model.nickName = fs.string(forColumn: "nickName")
                    model.userName = fs.string(forColumn: "userName")
                    model.password = fs.string(forColumn: "psw")
                    model.urlString = fs.string(forColumn: "urlString")
                    model.dsc = fs.string(forColumn: "dsc")
                    model.groupName = fs.string(forColumn: "groupName")
                    model.groupID = fs.string(forColumn: "groupID")
                    model.email = fs.string(forColumn: "email")
                    
                    fs.close()
                    LZSqlite.database.close()
                    return model
                }
            } catch  {
                
                print("Error: select model from table: \"\(table)\" with id: \"\(identifier)\" failed")
            }
        }
        
        LZSqlite.database.close()
        return nil
    }

查询某个区间的数据:

static func selectPart(_ range: CountableClosedRange<Int>, fromTable table: String) -> [LZDataModel]? {
    
    
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return nil
        }
        
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            let select = "SELECT * FROM '\(table)' LIMIT \(range.lowerBound), \(range.upperBound - range.lowerBound)"
            
            
            do {
                let fs = try LZSqlite.database.executeQuery(sql: select)
                
                var tempModels = [LZDataModel]()
                
                while fs.next() {
                    
                    let model = LZDataModel()
                    model.identifier = fs.string(forColumn: "ID")
                    model.nickName = fs.string(forColumn: "nickName")
                    model.userName = fs.string(forColumn: "userName")
                    model.password = fs.string(forColumn: "psw")
                    model.urlString = fs.string(forColumn: "urlString")
                    model.dsc = fs.string(forColumn: "dsc")
                    model.groupName = fs.string(forColumn: "groupName")
                    model.groupID = fs.string(forColumn: "groupID")
                    model.email = fs.string(forColumn: "email")
                    
                    tempModels.append(model)
                }
                
                fs.close()
                LZSqlite.database.close()
                return tempModels
            } catch {
                
                print("Error: select models from table: \"\(table)\" failed")
            }
            
        }
        
        LZSqlite.database.close()
        
        return nil
    }
这里函数参数的区间类型可查看我的另一篇文章:[Swift]使用区间作为函数的参数

查询已存储数据的数目:

static func countOf(table: String) -> Int {
        
        if LZSqlite.database.open() == false {
            
            LZSqlite.database.close()
            return 0
        }
        LZSqlite.database.setShouldCacheStatements(true)
        if LZSqlite.database.tableExists(table) {
            
            
            
            do {
                let fs = try LZSqlite.database.executeQuery(sql: "SELECT count(*) FROM '\(table)'")
                
                var count = 0
                if fs.next() {
                    
                     count = Int(fs.int(forColumn: "count(*)"))
                }
                
                fs.close()
                LZSqlite.database.close()
                return count
            } catch  {
                
                print("Error: select the element count of \(table) failed")
            }
        }
        
        LZSqlite.database.close()
        return 0
    }

3. 测试

以上方法都是完整的,可以直接使用, 这里给出部分使用方法:

3.1 创建sqlite文件
LZSqlite.createSqliteWithName("sqliteName")
3.2 删除sqlite文件
LZSqlite.clearSqlite()
3.3 创建表
LZSqlite.createTable("table")
3.4 删除表
LZSqlite.deleteTable("table")
3.5 向表内添加一列
LZSqlite.alterElement("ddddd", toTable: "table")
3.6 插入数据
for _ in 0...10 {
            
            let model = LZDataModel()
            model.identifier = "id\(arc4random()%10000 + 100)"
            model.userName = "name\(arc4random()%100 + 10)"
            model.groupID = "groupid\(arc4random()%1000 + 100)"
            model.groupName = "group\(arc4random()%100 + 10)"
            model.password = "password\(arc4random()%1000 + 100)"
            
          // 对可选项进行解包
//            LZSqlite.insertOptional(model: model, toTable: "table")

        // 不对可选项进行解包
            LZSqlite.insert(model: model, toTable: "table")
        }
3.7 删除数据
LZSqlite.delete(model: model, fromTable: "table")
3.8 更新数据
LZSqlite.update(model: model, inTable: "table")
3.9 查询数据库中数据数目
LZSqlite.countOf(table: "table")
3.10 查询所有数据
LZSqlite.selectAllFromTable("table")

3.11 查询某条数据

LZSqlite.selectModelWithID("id124" , fromTable: "table")
3.12 查询某个区间的数据
LZSqlite.selectPart(10...20, fromTable: "table")

是不是很方便?
本工具的github地址:** LQQZYY/FMDB-Swift**
线程安全的操作数据库:FMDB-Swift-Queue

如果对你有帮助还请star或fork来鼓励一下,谢谢!

(完)

Github LQQZYY
CSDN博客 流火绯瞳
新浪微博 杯水_沧海
QQ 302934443
上一篇下一篇

猜你喜欢

热点阅读