diff --git a/agent/init/migration/migrate.go b/agent/init/migration/migrate.go index af5194a7e..45344140b 100644 --- a/agent/init/migration/migrate.go +++ b/agent/init/migration/migrate.go @@ -10,6 +10,7 @@ import ( func Init() { InitAgentDB() InitTaskDB() + InitAlertDB() global.LOG.Info("Migration run successfully") } @@ -36,8 +37,6 @@ func InitAgentDB() { migrations.UpdateMcpServer, migrations.InitCronjobGroup, migrations.AddColumnToAlert, - migrations.MigrateAlertMethodConfigIDs, - migrations.AddAlertAuditUser, migrations.UpdateWebsiteSSL, migrations.AddQuickJump, migrations.UpdateMcpServerAddType, @@ -103,3 +102,14 @@ func InitTaskDB() { panic(err) } } + +func InitAlertDB() { + m := gormigrate.New(global.AlertDB, gormigrate.DefaultOptions, []*gormigrate.Migration{ + migrations.MigrateAlertMethodConfigIDs, + migrations.AddAlertAuditUser, + }) + if err := m.Migrate(); err != nil { + global.LOG.Error(err) + panic(err) + } +}