d40c815760
* 媒体库增加分类，图库多选择时优化。 * 重构 JWT token 生成，使用 `New()` 函数代替直接创建实例 * 上传组件支持查看大图 (#1982) * 将参数缓存，媒体库增加分类，图库多选择时优化。 (#1978) * 媒体库增加分类，图库多选择时优化。 *修复文件上传进度显示bug&按钮样式优化 (#1986) * fix:添加内部 iframe 展示网页，优化 permission 代码 * 俩个uuid库合并一个，更新库到当前版本。 * 优化关于我们界面 * feat: 个人中心头像调整，媒体库兼容性调整。 * feat: 自动化代码前端页面美化，多余按钮收入专家模式 * feat: 增加单独生成server功能 * feat: 限制单独生成前后端的情况下的细节配置 * feat: 修复全选失败报错的问题 --------- Co-authored-by: task <121913992@qq.com> Co-authored-by: Feng.YJ <32027253+huiyifyj@users.noreply.github.com> Co-authored-by: will0523 <dygsunshine@163.com> Co-authored-by: task <ms.yangdan@gmail.com> Co-authored-by: sslee <57312216+GIS142857@users.noreply.github.com> Co-authored-by: bypanghu <bypanghu@163.com> Co-authored-by: Azir <2075125282@qq.com> Co-authored-by: piexlMax(奇淼 <qimiaojiangjizhao@gmail.com> Co-authored-by: krank <emosick@qq.com>
93 lines
2.6 KiB
Go
93 lines
2.6 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/config"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/system/request"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/utils"
|
|
"github.com/google/uuid"
|
|
"github.com/gookit/color"
|
|
"gorm.io/driver/sqlserver"
|
|
"gorm.io/gorm"
|
|
"path/filepath"
|
|
)
|
|
|
|
type MssqlInitHandler struct{}
|
|
|
|
func NewMssqlInitHandler() *MssqlInitHandler {
|
|
return &MssqlInitHandler{}
|
|
}
|
|
|
|
// WriteConfig mssql回写配置
|
|
func (h MssqlInitHandler) WriteConfig(ctx context.Context) error {
|
|
c, ok := ctx.Value("config").(config.Mssql)
|
|
if !ok {
|
|
return errors.New("mssql config invalid")
|
|
}
|
|
global.GVA_CONFIG.System.DbType = "mssql"
|
|
global.GVA_CONFIG.Mssql = c
|
|
global.GVA_CONFIG.JWT.SigningKey = uuid.New().String()
|
|
cs := utils.StructToMap(global.GVA_CONFIG)
|
|
for k, v := range cs {
|
|
global.GVA_VP.Set(k, v)
|
|
}
|
|
global.GVA_ACTIVE_DBNAME = &c.Dbname
|
|
return global.GVA_VP.WriteConfig()
|
|
}
|
|
|
|
// EnsureDB 创建数据库并初始化 mssql
|
|
func (h MssqlInitHandler) EnsureDB(ctx context.Context, conf *request.InitDB) (next context.Context, err error) {
|
|
if s, ok := ctx.Value("dbtype").(string); !ok || s != "mssql" {
|
|
return ctx, ErrDBTypeMismatch
|
|
}
|
|
|
|
c := conf.ToMssqlConfig()
|
|
next = context.WithValue(ctx, "config", c)
|
|
if c.Dbname == "" {
|
|
return ctx, nil
|
|
} // 如果没有数据库名, 则跳出初始化数据
|
|
|
|
dsn := conf.MssqlEmptyDsn()
|
|
|
|
mssqlConfig := sqlserver.Config{
|
|
DSN: dsn, // DSN data source name
|
|
DefaultStringSize: 191, // string 类型字段的默认长度
|
|
}
|
|
|
|
var db *gorm.DB
|
|
|
|
if db, err = gorm.Open(sqlserver.New(mssqlConfig), &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
global.GVA_CONFIG.AutoCode.Root, _ = filepath.Abs("..")
|
|
next = context.WithValue(next, "db", db)
|
|
return next, err
|
|
}
|
|
|
|
func (h MssqlInitHandler) InitTables(ctx context.Context, inits initSlice) error {
|
|
return createTables(ctx, inits)
|
|
}
|
|
|
|
func (h MssqlInitHandler) InitData(ctx context.Context, inits initSlice) error {
|
|
next, cancel := context.WithCancel(ctx)
|
|
defer func(c func()) { c() }(cancel)
|
|
for _, init := range inits {
|
|
if init.DataInserted(next) {
|
|
color.Info.Printf(InitDataExist, Mssql, init.InitializerName())
|
|
continue
|
|
}
|
|
if n, err := init.InitializeData(next); err != nil {
|
|
color.Info.Printf(InitDataFailed, Mssql, init.InitializerName(), err)
|
|
return err
|
|
} else {
|
|
next = n
|
|
color.Info.Printf(InitDataSuccess, Mssql, init.InitializerName())
|
|
}
|
|
}
|
|
color.Info.Printf(InitSuccess, Mssql)
|
|
return nil
|
|
}
|