SQLite.swift对SQLite进行了全面的封装。作者采用了链式编程的写法，让数据库的管理变得优雅，可读性也很强。下面就说说如何使用
创建数据库
private init() {
// 获取保存数据库的路径
let path = getDocumentsPath()
let dataPath = "\(path)/dataDB.sqlite3"
do {
// 初始化方法，第一个参数Location指的是数据库的位置，有下面3种入参
// database = try Connection(.inMemory)//数据库存在内存里
// database = try Connection(.temporary)//临时数据库，使用完会被释放掉
// 存在硬盘中
database = try Connection(dataPath)
database?.userVersion = 1 //数据库版本
database?.busyTimeout = 5 //尝试重新连接语句的时间
// 也是尝试重新连接语句安全设置，跟busyTimeout2选1即可
// database?.busyHandler({ tries in
// if tries >= 5 {
// return false
// }
// return true
// })
} catch _ {
database = nil
}
}
创建表
static func createTable() throws {
guard let DB = BaseDBModel.sharedInstance.database else {
throw DataAccessError.datastoreConnectionError
}
do {
// ifNotExists：是否不存在的情况才会创建，记得设置为true
// withoutRowid： 是否自动创建自增的rowid
_ = try DB.run(table.create(ifNotExists: true, withoutRowid: false, block: { t in
t.column(name)
t.column(age)
}))
print("测试表 创建成功")
} catch _ {
print("测试表 创建失败")
throw DataAccessError.datastoreConnectionError
}
}
插入数据
static func insertWithData (array: Array<[String: Any]>) throws -> Void {
guard let DB = BaseDBModel.sharedInstance.database else {
throw DataAccessError.datastoreConnectionError
}
for dic in array {
let insert = table.insert(name <- isNonEmpStrAny(dic["name"]), age <- isNonEmpStrAny(dic["age"]))
do {
let rowId = try DB.run(insert)
guard rowId >= 0 else {
print("插入失败")
throw DataAccessError.insertError
}
print("插入成功：\(rowId)")
} catch {
print(error)
print("插入失败！")
throw DataAccessError.insertError
}
}
}
查询所有数据
static func seleAllData () throws -> Array<[String : Any]> {
guard let DB = BaseDBModel.sharedInstance.database else {
throw DataAccessError.datastoreConnectionError
}
var array: [[String : Any]] = []
let items = try DB.prepare(table)
for item in items {
let dic = ["name" : item[name], "age" : item[age]]
array.append(dic)
}
return array
}
修改数据
/// 根据age修改name
static func changeDataWith(modName: String,queryAge: String) throws -> Bool {
guard let DB = BaseDBModel.sharedInstance.database else {
throw DataAccessError.datastoreConnectionError
}
let query = table.filter(age == queryAge)
if try DB.run(query.update(name <- modName)) > 0 {
return true
} else {
throw DataAccessError.upDataError
}
}
删除所有数据
static func deleteAllData () throws -> Void {
guard let DB = BaseDBModel.sharedInstance.database else {
throw DataAccessError.datastoreConnectionError
}
if let count = try? DB.run(table.delete()) {
print("删除的条数为：\(count)")
} else {
print("删除失败")
throw DataAccessError.deleteError
}
}
实际使用
// 插入数据
try? TestDBModel.insertWithData(array: array)
let resultArray1: [[String : Any]] = try! TestDBModel.seleAllData()
printLog("第一次查询结果" + String(resultArray1.count))
for dic in resultArray1 {
printLog("name : " + (dic["name"] as! String) + ", age : " + (dic["age"] as! String))
}
// 把age为20的数据name改成王五
let _ = try? TestDBModel.changeDataWith(modName: "王五", queryAge: "20")
let resultArray2: [[String : Any]] = try! TestDBModel.seleAllData()
printLog("第二次查询结果" + String(resultArray2.count))
for dic in resultArray2 {
printLog("name : " + (dic["name"] as! String) + ", age : " + (dic["age"] as! String))
}
// 删除数据
try? TestDBModel.deleteAllData()
let resultArray3: [[String : Any]] = try! TestDBModel.seleAllData()
printLog("第三次查询结果" + String(resultArray3.count))
输出结果：
2023-04-27 14:03:03.603: 第一次查询结果2
2023-04-27 14:03:03.603: name : 张三, age : 20
2023-04-27 14:03:03.604: name : 李四, age : 21
2023-04-27 14:03:03.606: 第二次查询结果2
2023-04-27 14:03:03.606: name : 王五, age : 20
2023-04-27 14:03:03.607: name : 李四, age : 21
2023-04-27 14:03:03.608: 第三次查询结果0
可以发现增、删、改、查都达到了预期结果
主键
创建表的时候设置
_ = try DB.run(table.create(ifNotExists: true, withoutRowid: false, block: { t in
t.column(name, primaryKey: true)
t.column(age)
}))
let array = [["name" : "张三", "age" : "20"], ["name" : "李四", "age" : "21"], ["name" : "李四", "age" : "22"]]
// 插入数据
try? TestDBModel.insertWithData(array: array)
let resultArray1: [[String : Any]] = try! TestDBModel.seleAllData()
printLog("第一次查询结果" + String(resultArray1.count))
for dic in resultArray1 {
printLog("name : " + (dic["name"] as! String) + ", age : " + (dic["age"] as! String))
}
输出结果：
2023-04-27 14:15:30.710: 第一次查询结果2
2023-04-27 14:15:30.710: name : 张三, age : 20
2023-04-27 14:15:30.710: name : 李四, age : 21
我们可以发现["name" : "李四", "age" : "22"]这条数据没有存到库中，也就是主键起了作用
联合主键
_ = try DB.run(table.create(ifNotExists: true, withoutRowid: false, block: { t in
t.column(name)
t.column(age)
t.primaryKey(name, age)
}))
let array = [["name" : "张三", "age" : "20"], ["name" : "李四", "age" : "21"], ["name" : "李四", "age" : "22"], ["name" : "李四", "age" : "22"], ["name" : "张三", "age" : "21"]]
// 插入数据
try? TestDBModel.insertWithData(array: array)
let resultArray1: [[String : Any]] = try! TestDBModel.seleAllData()
printLog("第一次查询结果" + String(resultArray1.count))
for dic in resultArray1 {
printLog("name : " + (dic["name"] as! String) + ", age : " + (dic["age"] as! String))
}
输出结果：
2023-04-27 14:23:20.862: 第一次查询结果4
2023-04-27 14:23:20.862: name : 张三, age : 20
2023-04-27 14:23:20.863: name : 李四, age : 21
2023-04-27 14:23:20.863: name : 李四, age : 22
2023-04-27 14:23:20.863: name : 张三, age : 21
可以发现["name" : "李四", "age" : "22"]这条数据没有存到库中，也就是联合主键起了作用
另：暂时还没有发现如何修改主键，也就是表已经创建好了，name是主键，过了几天发现要用age当主键。是不支持还是没找到方法，有大神知道可以评论。
新增字段
开发中，有时候我们需要在表中新增列，可以使用如下方法，新增需要的列
var isExist = false
// 获取该表所有的字段名称
let expression = table.expression
let colunmnNames = try DB.prepare(expression.template,expression.bindings).columnNames
for colName in colunmnNames {
print(colName)
// 判断新增的字段名称是否存在
if colName == "achievement" {
isExist = true
break
}
}
// 不存在则添加
if !isExist {
try DB.run(table.addColumn(achievement, defaultValue: ""))
}
这样在这张表里面就新增achievement列成功了。
查询
多条件查询
let query = table.filter(age == "20").filter(achievement == "66")
let items = try DB.prepare(query)
// select(name) //只查询name字段
// limit(2, offset: 1) // 分页
// group(achievement) //分组
// order(achievement .asc) //排序
prepare、run、scalar使用条件
run 适用于 delete、insert、update 等
scalar 适用于 exists(是否存在)、count（总数量）等ScalarQuery类型返回值的
prepare 适用于其他QueryType类型返回值的
SQL语句查询，直接写好语句用run或者prepare。用scalar查询也可以，但是只会返回第一条数据的第一个字段，也就是count（*）等类型的查询可以用，返回直接是结果
let query = "select * from t_TestDB"
let items1 = try DB.run(query)
let items2 = try DB.prepare(query)
for item in items2 {
printLog(item) //一条数据
printLog(item[0]) //name
}
问题是查询出来的数据不是字典类型，是一个数组类型，需要自己通过下标去取值
