aiforge/models/custom_migrations.go

package models

import (
	"fmt"

	"code.gitea.io/gitea/modules/log"
	"xorm.io/xorm"
)

type CustomMigration struct {
	Description string
	Migrate     func(*xorm.Engine) error
}

type CustomMigrationStatic struct {
	Description string
	Migrate     func(*xorm.Engine, *xorm.Engine) error
}

var customMigrations = []CustomMigration{
	{"Custom v1 Topic struct change to support chinese", syncTopicStruct},
}

var customMigrationsStatic = []CustomMigrationStatic{
	{"Alter user static table field type  ", alterUserStaticTable},
	{"Delete zuzhi user history data ", deleteNotDisplayUser},
}

func MigrateCustom(x *xorm.Engine) {

	for _, m := range customMigrations {
		log.Info("Migration: %s", m.Description)
		if err := m.Migrate(x); err != nil {

			log.Error("Migration: %v", err)

		}
	}

}

func MigrateCustomStatic(x *xorm.Engine, static *xorm.Engine) {
	for _, m := range customMigrationsStatic {
		log.Info("Migration: %s", m.Description)
		if err := m.Migrate(x, static); err != nil {

			log.Error("Migration: %v", err)

		}
	}
}

func syncTopicStruct(x *xorm.Engine) error {

	query := "ALTER TABLE topic ALTER COLUMN name TYPE varchar(105);"

	_, err := x.Exec(query)
	return err
}

func alterUserStaticTable(x *xorm.Engine, static *xorm.Engine) error {
	alterSql := "alter table public.user_business_analysis alter column open_i_index type double precision"

	_, err := static.Exec(alterSql)
	return err

}

func deleteNotDisplayUser(x *xorm.Engine, static *xorm.Engine) error {

	querySQL := "select id,name from public.user where type=1"
	rows, err := x.Query(querySQL)
	if err != nil {
		log.Info("select db failed,err:", err)
		return err
	}

	for i, userRow := range rows {
		log.Info("delete zuzi user, i=" + fmt.Sprint(i) + " userName=" + string(userRow["name"]))
		deleteSql := "delete from user_business_analysis where id=" + string(userRow["id"]) + " and name='" + string(userRow["name"]) + "'"
		static.Exec(deleteSql)
	}

	return nil
}