| | |
| | | package models |
| | | |
| | | import ( |
| | | "github.com/golang/glog" |
| | | "github.com/jinzhu/gorm" |
| | | _ "github.com/jinzhu/gorm/dialects/sqlite" |
| | | "webserver/extend/config" |
| | | ) |
| | | |
| | | var db *gorm.DB |
| | | var err error |
| | | //import ( |
| | | // "github.com/jinzhu/gorm" |
| | | // _ "github.com/jinzhu/gorm/dialects/sqlite" |
| | | //) |
| | | // |
| | | //var db *gorm.DB |
| | | //var err error |
| | | |
| | | // Init creates a connection to mysql database and |
| | | // migrates any new models |
| | | func Init() { |
| | | // dbname, file path |
| | | db, err = gorm.Open(config.DBconf.Name, config.DBconf.FilePath) |
| | | db.LogMode(true) |
| | | db.Exec("PRAGMA foreign_keys = ON") |
| | | if err != nil { |
| | | glog.Error("db open error ", err) |
| | | } |
| | | db.AutoMigrate(&Area{}, &Camera{}, &LocalConfig{}, &VSSLocalSettingTbl{}, &Task{}, &CameraTask{}, &TaskSdk{}, &Sdk{}) |
| | | } |
| | | //func Init() { |
| | | // // dbname, file path |
| | | // db, err = gorm.Open(config.DBconf.Name, config.DBconf.FilePath) |
| | | // db.LogMode(true) |
| | | // db.Exec("PRAGMA foreign_keys = ON") |
| | | // if err != nil { |
| | | // glog.Error("db open error ", err) |
| | | // } |
| | | // //db.AutoMigrate(&Area{}, &Camera{}, &LocalConfig{}, &VSSLocalSettingTbl{}, &Task{}, &CameraTask{}, &TaskSdk{}, &Sdk{}) |
| | | //} |
| | | |
| | | //GetDB ... |
| | | func GetDB() *gorm.DB { |
| | | return db |
| | | } |
| | | |
| | | func CloseDB() { |
| | | db.Close() |
| | | } |
| | | //func GetDB() *gorm.DB { |
| | | // return db |
| | | //} |
| | | // |
| | | //func CloseDB() { |
| | | // db.Close() |
| | | //} |