feat: update migrate rule

This commit is contained in:
engigu
2024-01-13 13:43:30 +08:00
parent 547aa933c5
commit bff6abd57e
5 changed files with 38 additions and 4 deletions
+18 -3
View File
@@ -40,7 +40,6 @@ func InitAuthTableData() {
}
func Setup() {
setting.Setup()
db := models.Setup()
defer func(db *gorm.DB) {
err := db.Close()
@@ -49,15 +48,31 @@ func Setup() {
}
}(db)
db.AutoMigrate(
if setting.AppSetting.InitData != "enable" {
return
}
entry := logrus.WithFields(logrus.Fields{
"prefix": "[Init Data]",
})
tables := []interface{}{
&models.Auth{},
&models.SendTasks{},
&models.SendWays{},
&models.SendTasksLogs{},
&models.SendTasksIns{},
&models.Settings{},
)
}
for _, table := range tables {
tableName := db.NewScope(table).TableName()
entry.Infof("Migrate table: %s", tableName)
db.AutoMigrate(table)
}
entry.Infof("Init Account data...")
InitAuthTableData()
entry.Infof("All table data init done.")
}