Merge pull request '自定义脚本升级区分数据库' (#803) from fix-674 into V20211115

Reviewed-on: https://git.openi.org.cn/OpenI/aiforge/pulls/803
Reviewed-by: lewis <747342561@qq.com>

models/custom_migrations.go

@@ -14,6 +14,8 @@ var customMigrations = []CustomMigration{
 	{"Custom v1 Topic struct change to support chinese", syncTopicStruct},
 }
 
+var customMigrationsStatic = []CustomMigration{}
+
 func MigrateCustom(x *xorm.Engine) {
 
 	for _, m := range customMigrations {
@@ -27,6 +29,17 @@ func MigrateCustom(x *xorm.Engine) {
 
 }
 
+func MigrateCustomStatic(x *xorm.Engine) {
+	for _, m := range customMigrationsStatic {
+		log.Info("Migration: %s", m.Description)
+		if err := m.Migrate(x); err != nil {
+
+			log.Error("Migration: %v", err)
+
+		}
+	}
+}
+
 func syncTopicStruct(x *xorm.Engine) error {
 
 	query := "ALTER TABLE topic ALTER COLUMN name TYPE varchar(105);"

models/models.go

@@ -190,7 +190,7 @@ func setEngine(engine *xorm.Engine, table []interface{}, database *setting.DBInf
 	engine.SetMaxIdleConns(setting.Database.MaxIdleConns)
 	engine.SetConnMaxLifetime(setting.Database.ConnMaxLifetime)
 	engine.Sync2(table...)
-	MigrateCustom(engine)
+
 	return nil
 }
 
@@ -202,6 +202,7 @@ func SetEngine() (err error) {
 	if err = setEngine(x, tables, setting.Database); err != nil {
 		return err
 	}
+	MigrateCustom(x)
 
 	xStatistic, err = getEngine(setting.DatabaseStatistic)
 	if err != nil {
@@ -210,7 +211,7 @@ func SetEngine() (err error) {
 	if err = setEngine(xStatistic, tablesStatistic, setting.DatabaseStatistic); err != nil {
 		return err
 	}
-
+	MigrateCustomStatic(xStatistic)
 	return nil
 }