项目开发过程中,可能需要给已经创建的table再新增一个字段。
我的项目当初是用swift写的,数据库用的是SQLite.swift。
//工具类
class YKSQLiteManager {
// MARK:- 设计单例对象
// 1.创建类的实例变量
// let是线程安全
static let instance = YKSQLiteManager()
// 2.对外提供获取单例的接口
// 定义类方法class func 方法名称() -> 返回值
class func shareInstance() -> YKSQLiteManager {
return instance
}
// 数据库链接
var database: Connection!
init() {
do {
let path = NSHomeDirectory() + "/Documents/yourTable.sqlite3"
//print("path\(path)")
database = try Connection(path)
} catch { print(error) }
}
}
//我创建的表
struct GatewayTable {
/// 数据库管理者
private let sqlManager = YKSQLiteManager.shareInstance()
// 表
private let tGateway = Table("tGateway_table")
// 表字段
let gateway_id = Expression<String>("gateway_id")
let loc_id = Expression<String>("loc_id")
init() {
// 建表
try! sqlManager.database.run(tGateway.create(ifNotExists: true, block: { t in
t.column(gateway_id, primaryKey: true)
t.column(loc_id)
}))
}
}
后续开发过程中,需要给tGateway再新增一个字段wifi_Info,
SQLite中表有一个addColumn方法,于是
let wifi_Info = Expression<String>("wifi_Info")
tGateway.addColumn(wifi_Info, defaultValue: "unknown")
然后直接报错,闪退。果然事情没这么简单。
于是上网找,看到https://blog.csdn.net/heyuzu1/article/details/60779522这篇文章
于是
struct GatewayTable {
/// 数据库管理者
private let sqlManager = YKSQLiteManager.shareInstance()
/// 表
private let tGateway = Table("tGateway_table")
/// 表字段
let gateway_id = Expression<String>("gateway_id")
let loc_id = Expression<String>("loc_id")
let wifi_Info = Expression<String>("wifi_Info")
init() {
// 建表
try! sqlManager.database.run(tGateway.create(ifNotExists: true, block: { t in
t.column(gateway_id, primaryKey: true)
t.column(loc_id)
t.column(wifi_Info)
}))
isWifi_InfoExist()
}
func isWifi_InfoExist() {
do {
var isExist = false
let expression = tGateway.expression
let colunmnNames = try sqlManager.database.prepare(expression.template,expression.bindings).columnNames
for colName in colunmnNames {
if colName == "wifi_Info" {
isExist = true
break
}
}
if !isExist {
tGateway.addColumn(wifi_Info, defaultValue: "unknown")
}
} catch {
}
}
然后跑一下,没有闪退了。但是发现tGateway.addColumn(wifi_Info, defaultValue: "unknown")执行结果是失败,tGateway表没有新增这一列。
原因在于没有用数据库管理去执行这条指令啊,相当于把代码写了,但是没有编译执行。
于是把tGateway.addColumn(wifi_Info, defaultValue: "unknown")改成
sqlManager.database.run(tGateway.addColumn(wifi_Info, defaultValue: "unknown"))
func isWifi_InfoExist() {
do {
var isExist = false
let expression = tGateway.expression
let colunmnNames = try sqlManager.database.prepare(expression.template,expression.bindings).columnNames
for colName in colunmnNames {
if colName == "wifi_Info" {
isExist = true
break
}
}
if !isExist {
do {
try sqlManager.database.run(tGateway.addColumn(wifi_Info, defaultValue: "unknown"))
} catch {
print("addError")
}
}
} catch {
}
}
这样就可以得到想要的结果了。
//新增各种类型的一列,泛型版
class func addColumn<V:Value>(table:Table,coluName:String,column:Expression<V>,defaultValue:V) {
do {
var isExist = false
let expression = table.expression
let colunmnNames = try YKSQLiteManager.shareInstance().database.prepare(expression.template,expression.bindings).columnNames
for colName in colunmnNames {
if colName == coluName {
isExist = true
break
}
}
if !isExist {
do {
try YKSQLiteManager.shareInstance().database.run(table.addColumn(column, defaultValue:defaultValue))
} catch {
print("addError")
}
}
} catch {
}
}
PS. swift和SQLite使用还不是很熟练,如有误,欢迎指出。
参考资料:
SQLite.swift的基本使用:https://www.jianshu.com/p/30e31282c4b9
新增一列:https://blog.csdn.net/heyuzu1/article/details/60779522