feat: 优化数据库迁移与添加老表迁移

This commit is contained in:
王一之
2023-07-22 00:48:41 +08:00
parent f5bb0cad3e
commit b1ab66ecf9
9 changed files with 130 additions and 32 deletions

View File

@@ -3,6 +3,13 @@ package migrations
import (
"github.com/go-gormigrate/gormigrate/v2"
"gorm.io/gorm"
entity2 "mayfly-go/internal/db/domain/entity"
"mayfly-go/internal/machine/domain/entity"
entity3 "mayfly-go/internal/mongo/domain/entity"
entity6 "mayfly-go/internal/msg/domain/entity"
entity4 "mayfly-go/internal/redis/domain/entity"
entity5 "mayfly-go/internal/sys/domain/entity"
entity7 "mayfly-go/internal/tag/domain/entity"
)
// T2022 TODO 在此之前的数据库表结构初始化, 目前先使用mayfly-go.sql文件初始化数据库结构
@@ -10,6 +17,87 @@ func T2022() *gormigrate.Migration {
return &gormigrate.Migration{
ID: "2022",
Migrate: func(tx *gorm.DB) error {
if err := tx.AutoMigrate(&entity.AuthCert{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity.Machine{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity.MachineFile{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity.MachineMonitor{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity.MachineScript{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity.MachineCronJob{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity.MachineCronJobExec{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity.MachineCronJobRelate{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity2.Db{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity2.DbSql{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity2.DbSqlExec{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity3.Mongo{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity4.Redis{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity5.Account{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity5.AccountRole{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity5.Config{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity5.SysLog{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity5.Resource{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity5.Role{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity5.RoleResource{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity6.Msg{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity7.TagTree{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity7.TagTreeTeam{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity7.Team{}); err != nil {
return err
}
if err := tx.AutoMigrate(&entity7.TeamMember{}); err != nil {
return err
}
return nil
},
Rollback: func(tx *gorm.DB) error {