42 lines
1.3 KiB
Plaintext
42 lines
1.3 KiB
Plaintext
|
package connection
|
|||
|
|
|
|||
|
|
import "gorm.io/gorm"
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
// 存放实体结构
|
|||
|
|
dbModelMap = make([]interface{}, 0)
|
|||
|
|
|
|||
|
|
//当前管理数据库
|
|||
|
|
modelDB *gorm.DB
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// RegisterDBModel 注册数据库模型到全局变量dbModelMap中。
|
|||
|
|
// 这个函数接受一个interface{}类型的参数dbModel，表示数据库模型。
|
|||
|
|
// 函数的目的是将传入的数据库模型添加到全局变量dbModelMap中，
|
|||
|
|
// 以便在其他地方可以访问和使用这些数据库模型。
|
|||
|
|
func RegisterDBModel(dbModel interface{}) {
|
|||
|
|
// 将dbModel的地址添加到dbModelMap中。
|
|||
|
|
// 这里使用地址是因为数据库模型可能比较大，通过引用存储可以提高效率。
|
|||
|
|
dbModelMap = append(dbModelMap, &dbModel)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置modelDB 类型
|
|||
|
|
func SetModelDB(db *gorm.DB) {
|
|||
|
|
modelDB = db
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BuildDB 用于遍历dbModelMap中的所有数据库模型，并检查每个模型对应的表是否存在于数据库中。
|
|||
|
|
// 如果表不存在，则自动迁移（创建）该表。这样可以确保数据库模式与程序中的模型保持同步。
|
|||
|
|
func BuildDB() {
|
|||
|
|
// 遍历dbModelMap中的每个元素
|
|||
|
|
for _, dbModel := range dbModelMap {
|
|||
|
|
|
|||
|
|
// 检查数据库中是否存在与dbModel对应的表
|
|||
|
|
tableExists := modelDB.Migrator().HasTable(dbModel)
|
|||
|
|
if !tableExists {
|
|||
|
|
// 如果表不存在，则进行自动迁移
|
|||
|
|
modelDB.AutoMigrate(dbModel)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|