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{}) db.AutoMigrate(&commonModels.Area{}, &commonModels.Camera{}, &commonModels.CameraArea{}, &commonModels.Sensor{}, &commonModels.CameraSensor{}) commonModels.SetDB(db) } // GetDB ... func GetDB() *gorm.DB { return db } func CloseDB() { db.Close() }