qixiaoning
2025-10-29 9b17a8dcbc1f94eb117a37b3b24ca9dae0d2d588
camera-service/models/db.go
@@ -1,11 +1,12 @@
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
@@ -15,21 +16,27 @@
// 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{})
   // 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 ...
// GetDB ...
func GetDB() *gorm.DB {
   return db
}
func CloseDB() {
   db.Close()
}
}