package models
|
|
import (
|
commonModels "vamicro/camera-common/models"
|
"vamicro/config"
|
|
"basic.com/valib/logger.git"
|
"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() {
|
db, err = gorm.Open(config.DBconf.Name, "../config/camera-service.db")
|
|
if err != nil {
|
logger.Debug("db open error ", err)
|
}
|
db.LogMode(true)
|
//db.SetLogger(&DbLogger{})
|
// err1 := db.Debug().AutoMigrate(&commonModels.Area{}, &commonModels.Camera{}, &commonModels.CameraArea{},
|
// &commonModels.Sensor{}, &commonModels.CameraSensor{}, &commonModels.TrainTags{})
|
err1 := db.Debug().AutoMigrate(&commonModels.TrainTags{})
|
if err1 != nil {
|
logger.Debug("自动迁移失败 ", err1)
|
}
|
|
commonModels.SetDB(db)
|
}
|
|
// GetDB ...
|
func GetDB() *gorm.DB {
|
return db
|
}
|
|
func CloseDB() {
|
db.Close()
|
}
|
