| | |
| | | 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" |
| | | "basic.com/valib/logger.git" |
| | | "vamicro/config" |
| | | commonModels "vamicro/camera-common/models" |
| | | ) |
| | | |
| | | var db *gorm.DB |
| | |
| | | } |
| | | db.LogMode(true) |
| | | //db.SetLogger(&DbLogger{}) |
| | | db.AutoMigrate(&commonModels.Area{}, &commonModels.Camera{},&commonModels.CameraArea{}, |
| | | db.AutoMigrate(&commonModels.Area{}, &commonModels.Camera{}, &commonModels.CameraArea{}, |
| | | &commonModels.Sensor{}, &commonModels.CameraSensor{}) |
| | | commonModels.SetDB(db) |
| | | } |
| | | |
| | | //GetDB ... |
| | | // GetDB ... |
| | | func GetDB() *gorm.DB { |
| | | return db |
| | | } |
| | | |
| | | func CloseDB() { |
| | | db.Close() |
| | | } |
| | | } |