liuxiaolong
2019-07-04 bcfc7757e02487680d7c5132119d9f396211b1df
models/db.go
@@ -1,33 +1,31 @@
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()
//}