✨ init project

service/system/sys_initdb_mysql.go

@@ -0,0 +1,97 @@
+package system
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"path/filepath"
+
+	"git.echol.cn/loser/lckt/config"
+	"github.com/gookit/color"
+
+	"git.echol.cn/loser/lckt/utils"
+
+	"git.echol.cn/loser/lckt/global"
+	"git.echol.cn/loser/lckt/model/system/request"
+	"github.com/google/uuid"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+)
+
+type MysqlInitHandler struct{}
+
+func NewMysqlInitHandler() *MysqlInitHandler {
+	return &MysqlInitHandler{}
+}
+
+// WriteConfig mysql回写配置
+func (h MysqlInitHandler) WriteConfig(ctx context.Context) error {
+	c, ok := ctx.Value("config").(config.Mysql)
+	if !ok {
+		return errors.New("mysql config invalid")
+	}
+	global.GVA_CONFIG.System.DbType = "mysql"
+	global.GVA_CONFIG.Mysql = 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 创建数据库并初始化 mysql
+func (h MysqlInitHandler) EnsureDB(ctx context.Context, conf *request.InitDB) (next context.Context, err error) {
+	if s, ok := ctx.Value("dbtype").(string); !ok || s != "mysql" {
+		return ctx, ErrDBTypeMismatch
+	}
+
+	c := conf.ToMysqlConfig()
+	next = context.WithValue(ctx, "config", c)
+	if c.Dbname == "" {
+		return ctx, nil
+	} // 如果没有数据库名, 则跳出初始化数据
+
+	dsn := conf.MysqlEmptyDsn()
+	createSql := fmt.Sprintf("CREATE DATABASE IF NOT EXISTS `%s` DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci;", c.Dbname)
+	if err = createDatabase(dsn, "mysql", createSql); err != nil {
+		return nil, err
+	} // 创建数据库
+
+	var db *gorm.DB
+	if db, err = gorm.Open(mysql.New(mysql.Config{
+		DSN:                       c.Dsn(), // DSN data source name
+		DefaultStringSize:         191,     // string 类型字段的默认长度
+		SkipInitializeWithVersion: true,    // 根据版本自动配置
+	}), &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true}); err != nil {
+		return ctx, err
+	}
+	global.GVA_CONFIG.AutoCode.Root, _ = filepath.Abs("..")
+	next = context.WithValue(next, "db", db)
+	return next, err
+}
+
+func (h MysqlInitHandler) InitTables(ctx context.Context, inits initSlice) error {
+	return createTables(ctx, inits)
+}
+
+func (h MysqlInitHandler) 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, Mysql, init.InitializerName())
+			continue
+		}
+		if n, err := init.InitializeData(next); err != nil {
+			color.Info.Printf(InitDataFailed, Mysql, init.InitializerName(), err)
+			return err
+		} else {
+			next = n
+			color.Info.Printf(InitDataSuccess, Mysql, init.InitializerName())
+		}
+	}
+	color.Info.Printf(InitSuccess, Mysql)
+	return nil
+}