|
|
@@ -1,317 +0,0 @@
|
|
|
-package source
|
|
|
-
|
|
|
-import (
|
|
|
- "fmt"
|
|
|
- information "gin-vue-admin/cmd/information/extra"
|
|
|
- data "gin-vue-admin/cmd/information/system"
|
|
|
- "gin-vue-admin/global"
|
|
|
- "gin-vue-admin/model"
|
|
|
- "os"
|
|
|
- "strings"
|
|
|
-
|
|
|
- gormadapter "github.com/casbin/gorm-adapter/v3"
|
|
|
- "github.com/gookit/color"
|
|
|
- "gorm.io/driver/mysql"
|
|
|
- "gorm.io/gorm"
|
|
|
- "gorm.io/gorm/logger"
|
|
|
-)
|
|
|
-
|
|
|
-type DatabaseInfo struct {
|
|
|
- Value string `gorm:"column:Value"`
|
|
|
- VariableName string `gorm:"column:Variable_name"`
|
|
|
-}
|
|
|
-
|
|
|
-var Mysql = &_mysql{_config: &gorm.Config{}}
|
|
|
-
|
|
|
-type _mysql struct {
|
|
|
- db *gorm.DB
|
|
|
- err error
|
|
|
- _config *gorm.Config
|
|
|
-
|
|
|
- old string
|
|
|
- input string
|
|
|
- version string
|
|
|
- character string
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) Init() {
|
|
|
- if global.GVA_CONFIG.Mysql.LogMode {
|
|
|
- m._config.Logger = logger.Default.LogMode(logger.Info)
|
|
|
- } else {
|
|
|
- m._config.Logger = logger.Default.LogMode(logger.Silent)
|
|
|
- }
|
|
|
- m._config.DisableForeignKeyConstraintWhenMigrating = true
|
|
|
- m.db, m.err = gorm.Open(mysql.New(mysql.Config{
|
|
|
- DSN: global.GVA_CONFIG.Mysql.Dsn(),
|
|
|
- DefaultStringSize: 191,
|
|
|
- DisableDatetimePrecision: true,
|
|
|
- DontSupportRenameIndex: true,
|
|
|
- DontSupportRenameColumn: true,
|
|
|
- SkipInitializeWithVersion: false,
|
|
|
- }), m._config)
|
|
|
- global.GVA_DB = m.db
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) AutoMigrateTables() {
|
|
|
- if !global.GVA_DB.Migrator().HasTable("casbin_rule") {
|
|
|
- m.err = global.GVA_DB.Migrator().CreateTable(&gormadapter.CasbinRule{})
|
|
|
- }
|
|
|
- m.err = m.db.AutoMigrate(
|
|
|
- new(model.SysApi),
|
|
|
- new(model.SysUser),
|
|
|
- new(model.SysBaseMenu),
|
|
|
- new(model.SysAuthority),
|
|
|
- new(model.SysDictionary),
|
|
|
- new(model.JwtBlacklist),
|
|
|
- new(model.SysOperationRecord),
|
|
|
- new(model.SysDictionaryDetail),
|
|
|
- new(model.SysBaseMenuParameter),
|
|
|
-
|
|
|
- new(model.WorkflowNode),
|
|
|
- new(model.WorkflowEdge),
|
|
|
- new(model.WorkflowProcess),
|
|
|
- new(model.WorkflowEndPoint),
|
|
|
- new(model.WorkflowStartPoint),
|
|
|
-
|
|
|
- new(model.ExaFileUploadAndDownload),
|
|
|
- new(model.ExaFile),
|
|
|
- new(model.ExaFileChunk),
|
|
|
- )
|
|
|
- if m.err != nil {
|
|
|
- color.Warn.Printf("[Mysql] --> 初始化数据表失败, err: %v\n", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- color.Info.Println("[Mysql] --> 初始化数据表成功! ")
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) InitData() {
|
|
|
- if m.err = data.Api.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> sys_apis 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.Admin.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> sys_users 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.Casbin.Init(); m.err != nil {
|
|
|
- color.Error.Println("\n[Mysql] --> casbin_rule 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.BaseMenu.Init(); m.err != nil {
|
|
|
- color.Error.Println("\n[Mysql] --> sys_base_menus 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.Authority.Init(); m.err != nil {
|
|
|
- color.Error.Println("\n[Mysql] --> sys_authorities 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.AuthorityMenu.Init(); m.err != nil {
|
|
|
- color.Error.Println("\n[Mysql] --> authority_menu 视图创建失败!, err:%v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.Dictionary.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> dictionaries 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.AuthoritiesMenus.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> sys_authority_menus 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.DataAuthorities.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> sys_data_authority_id 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.DictionaryDetail.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> sys_dictionary_details 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = data.Workflow.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> 工作流相关 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- if m.err = information.File.Init(); m.err != nil {
|
|
|
- color.Warn.Println("\n[Mysql] --> exa_file_upload_and_downloads 表初始数据失败, err: %v", m.err)
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- color.Info.Println("\n[Mysql] --> 初始化数据成功!\n")
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) CheckDatabase() {
|
|
|
- var unknownDatabase = fmt.Sprintf("Unknown database '%v'", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- m.Init()
|
|
|
- if m.err != nil {
|
|
|
- if strings.Split(m.err.Error(), ": ")[1] == unknownDatabase {
|
|
|
- color.Debug.Print("\n[Mysql] -->配置文件的数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Println("不存在!\n")
|
|
|
- color.Debug.Println("您的配置文件所配置的数据库不存在,请选择:")
|
|
|
- color.Debug.Print("0:请自行创建配置文件所配置的数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} \n", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print("1:尝试使用sql为您创建配置文件所配置的数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} \n", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Println("2:忽略错误! 注意: 如果不修复, 将会退出初始化数据的进程!")
|
|
|
- color.Warn.Println("\n注意!!!!!!!")
|
|
|
- color.Warn.Println("输入1之后,如果配置文件的mysql用户名为root才会有百分百的权限去创建数据库,不是root的话就会跳过创建数据库步骤!\n")
|
|
|
- color.Debug.Println("请输入指令:")
|
|
|
- if n, _ := fmt.Scanln(&m.input); n != 0 {
|
|
|
- if m.input == "1" {
|
|
|
- if global.GVA_CONFIG.Mysql.Username == "root" {
|
|
|
- m.database()
|
|
|
- } else {
|
|
|
- color.Debug.Print("\n很抱歉,您的配置文件的mysql用户名配置不是root,不确定你有无权限创建数据库,为您跳过创建数据库操作,请自行创建配置文件所配置的数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} \n", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- }
|
|
|
- } else if m.input == "2" {
|
|
|
- os.Exit(0)
|
|
|
- } else {
|
|
|
- color.Warn.Println("[Mysql] --> 请自行创建数据库!")
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) CheckUtf8mb4() {
|
|
|
- m.Character()
|
|
|
- if m.character != "utf8mb4" {
|
|
|
- color.Info.Println("您当前的数据库编码不是utf8mb4,请选择:")
|
|
|
- color.Info.Println("0:请自行修改数据的编码为utf8mb4!")
|
|
|
- color.Info.Println("1:尝试使用sql为您修改编码为utf8mb4!")
|
|
|
- color.Info.Println("2:忽略错误! 注意如果不修复,生成初始数据的时候也许或许可能有几率报错的喔!")
|
|
|
- color.Info.Println("请输入指令:")
|
|
|
- if n, _ := fmt.Scanln(&m.input); n != 0 {
|
|
|
- if m.input == "1" {
|
|
|
- m.utf8mb4()
|
|
|
- } else if m.input == "2" {
|
|
|
- return
|
|
|
- } else {
|
|
|
- color.Warn.Println("[Mysql] --> 请自行修改数据的编码为utf8mb4!")
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) Info() {
|
|
|
- m.Version()
|
|
|
- color.Debug.Print("\n您当前的数据库版本: ")
|
|
|
- color.LightGreen.Printf(" {%v} ", m.version)
|
|
|
- color.Debug.Print(", 使用的数据库是: ")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print(", 数据库编码是: ")
|
|
|
- color.LightGreen.Printf(" {%v} \n\n", m.character)
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) Version() {
|
|
|
- color.Debug.Println("[Mysql] -->获取数据库版本中.......")
|
|
|
- if err := global.GVA_DB.Raw("SELECT VERSION() AS version;").Scan(&m.version).Error; err != nil {
|
|
|
- color.Info.Printf("[Mysql] -->获取数据库版本失败! err: %v", err)
|
|
|
- m.version = "未知版本~~~"
|
|
|
- }
|
|
|
- color.Debug.Printf("\n[Mysql] -->获取数据库版本成功!\n")
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) Character() {
|
|
|
- var info DatabaseInfo
|
|
|
- color.Debug.Println("\n[Mysql] -->获取数据库编码中.......")
|
|
|
- if err := global.GVA_DB.Raw("show variables like 'character_set_database' ").Scan(&info).Error; err != nil {
|
|
|
- color.Error.Printf("[Mysql] -->获取数据库编码失败! err:%v\n", err)
|
|
|
- m.character = "未知编码~~~"
|
|
|
- }
|
|
|
- color.Debug.Println("\n[Mysql] -->获取数据库编码成功!\n")
|
|
|
- m.character = info.Value
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) utf8mb4() {
|
|
|
- color.Debug.Print("\n[Mysql] --> 设置数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print("数据库的编码为utf8mb4中.......\n")
|
|
|
- if err := global.GVA_DB.Debug().Exec("ALTER DATABASE " + global.GVA_CONFIG.Mysql.Dbname + " CHARACTER SET `utf8mb4` COLLATE `utf8mb4_general_ci`").Error; err != nil {
|
|
|
- color.Debug.Print("\n[Mysql] --> 设置数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print("数据库的编码为utf8mb4失败!请手动修改数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Println("的编码为utf8mb4\n")
|
|
|
- return
|
|
|
- }
|
|
|
- color.Info.Print("\n[Mysql] --> 设置数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print("的编码为utf8mb4成功!\n")
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) database() {
|
|
|
- m.old = global.GVA_CONFIG.Mysql.Dbname
|
|
|
- global.GVA_CONFIG.Mysql.Dbname = "mysql"
|
|
|
- color.Debug.Printf("\n[Mysql] --> 正在连接 mysql 数据库中.......\n")
|
|
|
- m.Init()
|
|
|
- if m.err != nil {
|
|
|
- color.Error.Printf("\n[Mysql] --> 链接 mysql 数据库失败!, err: %v\n", m.err)
|
|
|
- color.Error.Printf("[Mysql] --> 请自行创建配置文件所需的数据库!\n")
|
|
|
- os.Exit(0)
|
|
|
- }
|
|
|
- color.Debug.Printf("\n[Mysql] --> 连接 mysql 数据库成功\n")
|
|
|
- global.GVA_CONFIG.Mysql.Dbname = m.old
|
|
|
- color.Debug.Print("\n[Mysql] --> 正在为您创建数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print("中.......\n")
|
|
|
- if m.err = global.GVA_DB.Exec("CREATE DATABASE IF NOT EXISTS " + global.GVA_CONFIG.Mysql.Dbname + " DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci;").Error; m.err != nil {
|
|
|
- color.Debug.Print("\n[Mysql] --> 创建数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print("失败!请手动修改数据库名为")
|
|
|
- color.LightGreen.Printf(" {%v} \n", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- os.Exit(0)
|
|
|
- return
|
|
|
- }
|
|
|
- color.Debug.Print("\n[Mysql] --> 正在为您创建数据库名为:")
|
|
|
- color.LightGreen.Printf(" {%v} ", global.GVA_CONFIG.Mysql.Dbname)
|
|
|
- color.Debug.Print("成功!\n")
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (m *_mysql) zero() {
|
|
|
- var info DatabaseInfo
|
|
|
- color.Info.Println("\n[Mysql]--> 获取数据库数据中.......")
|
|
|
- if err := global.GVA_DB.Raw("show variables like 'sql_mode';").Scan(&info).Error; err != nil {
|
|
|
- color.Error.Printf("\n[Mysql]-->获取数据库数据失败! err:%v\n", err)
|
|
|
- }
|
|
|
- color.Info.Println("\n[Mysql]--> 处理数据库返回数据.......")
|
|
|
- var values = strings.Split(info.Value, ",")
|
|
|
- info.Value = ""
|
|
|
- for i, value := range values {
|
|
|
- if value == "NO_ZERO_IN_DATE" || value == "NO_ZERO_DATE" {
|
|
|
- } else {
|
|
|
- if i == len(values)-1 {
|
|
|
- info.Value += value
|
|
|
- } else {
|
|
|
- info.Value += value + ","
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- if err := global.GVA_DB.Exec("set global sql_mode='" + info.Value + "';").Error; err != nil {
|
|
|
- color.Error.Printf("\n[Mysql]--> 设置数据库的零值失效失败! err:%v\n", err)
|
|
|
- return
|
|
|
- }
|
|
|
- color.Info.Println("\n[Mysql]--> 设置数据库零值失效成功")
|
|
|
-}
|
pixel