77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
|
|
"kra/internal/biz/system"
|
|
"kra/internal/data/model"
|
|
|
|
"github.com/pkg/errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const initOrderExcelTemplate = initOrderDictDetail + 1
|
|
|
|
type initExcelTemplate struct{}
|
|
|
|
func NewInitExcelTemplate() system.SubInitializer {
|
|
return &initExcelTemplate{}
|
|
}
|
|
|
|
func (i *initExcelTemplate) InitializerName() string {
|
|
return model.TableNameSysExportTemplate
|
|
}
|
|
|
|
func (i *initExcelTemplate) MigrateTable(ctx context.Context) (context.Context, error) {
|
|
db, ok := ctx.Value("db").(*gorm.DB)
|
|
if !ok {
|
|
return ctx, system.ErrMissingDBContext
|
|
}
|
|
return ctx, db.AutoMigrate(&model.SysExportTemplate{})
|
|
}
|
|
|
|
func (i *initExcelTemplate) TableCreated(ctx context.Context) bool {
|
|
db, ok := ctx.Value("db").(*gorm.DB)
|
|
if !ok {
|
|
return false
|
|
}
|
|
return db.Migrator().HasTable(&model.SysExportTemplate{})
|
|
}
|
|
|
|
func (i *initExcelTemplate) InitializeData(ctx context.Context) (context.Context, error) {
|
|
db, ok := ctx.Value("db").(*gorm.DB)
|
|
if !ok {
|
|
return ctx, system.ErrMissingDBContext
|
|
}
|
|
|
|
entities := []model.SysExportTemplate{
|
|
{
|
|
Name: "api",
|
|
TblName: "sys_apis",
|
|
TemplateID: "api",
|
|
TemplateInfo: `{
|
|
"path":"路径",
|
|
"method":"方法（大写）",
|
|
"description":"方法介绍",
|
|
"api_group":"方法分组"
|
|
}`,
|
|
},
|
|
}
|
|
if err := db.Create(&entities).Error; err != nil {
|
|
return ctx, errors.Wrap(err, model.TableNameSysExportTemplate+"表数据初始化失败!")
|
|
}
|
|
next := context.WithValue(ctx, i.InitializerName(), entities)
|
|
return next, nil
|
|
}
|
|
|
|
func (i *initExcelTemplate) DataInserted(ctx context.Context) bool {
|
|
db, ok := ctx.Value("db").(*gorm.DB)
|
|
if !ok {
|
|
return false
|
|
}
|
|
if errors.Is(db.First(&model.SysExportTemplate{}).Error, gorm.ErrRecordNotFound) {
|
|
return false
|
|
}
|
|
return true
|
|
}
|