iOS Swift 4.0:第三方SQLite,已经创建的表格新

2019-09-26  本文已影响0人  zaq1125

项目开发过程中,可能需要给已经创建的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

上一篇 下一篇

猜你喜欢

热点阅读