package initialize import ( "os" "github.com/flipped-aurora/gin-vue-admin/server/global" "github.com/flipped-aurora/gin-vue-admin/server/initialize/internal" "github.com/flipped-aurora/gin-vue-admin/server/model/autocode" "github.com/flipped-aurora/gin-vue-admin/server/model/example" "github.com/flipped-aurora/gin-vue-admin/server/model/system" "go.uber.org/zap" "gorm.io/driver/mysql" "gorm.io/gorm" "gorm.io/gorm/logger" ) //@author: SliverHorn //@function: Gorm //@description: 初始化数据库并产生数据库全局变量 //@return: *gorm.DB func Gorm() *gorm.DB { switch global.GVA_CONFIG.System.DbType { case "mysql": return GormMysql() default: return GormMysql() } } // MysqlTables //@author: SliverHorn //@function: MysqlTables //@description: 注册数据库表专用 //@param: db *gorm.DB func MysqlTables(db *gorm.DB) { err := db.AutoMigrate( system.SysUser{}, system.SysAuthority{}, system.SysApi{}, system.SysBaseMenu{}, system.SysBaseMenuParameter{}, system.JwtBlacklist{}, system.SysDictionary{}, system.SysDictionaryDetail{}, example.ExaFileUploadAndDownload{}, example.ExaFile{}, example.ExaFileChunk{}, example.ExaCustomer{}, system.SysOperationRecord{}, system.SysAutoCodeHistory{}, // Code generated by github.com/flipped-aurora/gin-vue-admin/server Begin; DO NOT EDIT. autocode.AutoCodeExample{}, autocode.Unit{}, autocode.ProblemType{}, autocode.UnitHistory{}, autocode.ProblemReply{}, autocode.Organization{}, // Code generated by github.com/flipped-aurora/gin-vue-admin/server End; DO NOT EDIT. ) if err != nil { global.GVA_LOG.Error("register table failed", zap.Any("err", err)) os.Exit(0) } global.GVA_LOG.Info("register table success") } //@author: SliverHorn //@function: GormMysql //@description: 初始化Mysql数据库 //@return: *gorm.DB func GormMysql() *gorm.DB { m := global.GVA_CONFIG.Mysql if m.Dbname == "" { return nil } dsn := m.Username + ":" + m.Password + "@tcp(" + m.Path + ")/" + m.Dbname + "?" + m.Config mysqlConfig := mysql.Config{ DSN: dsn, // DSN data source name DefaultStringSize: 191, // string 类型字段的默认长度 DisableDatetimePrecision: true, // 禁用 datetime 精度,MySQL 5.6 之前的数据库不支持 DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式,MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引 DontSupportRenameColumn: true, // 用 `change` 重命名列,MySQL 8 之前的数据库和 MariaDB 不支持重命名列 SkipInitializeWithVersion: false, // 根据版本自动配置 } if db, err := gorm.Open(mysql.New(mysqlConfig), gormConfig()); err != nil { //global.GVA_LOG.Error("MySQL启动异常", zap.Any("err", err)) //os.Exit(0) //return nil return nil } else { sqlDB, _ := db.DB() sqlDB.SetMaxIdleConns(m.MaxIdleConns) sqlDB.SetMaxOpenConns(m.MaxOpenConns) return db } } //@author: SliverHorn //@function: gormConfig //@description: 根据配置决定是否开启日志 //@param: mod bool //@return: *gorm.Config func gormConfig() *gorm.Config { config := &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true} switch global.GVA_CONFIG.Mysql.LogMode { case "silent", "Silent": config.Logger = internal.Default.LogMode(logger.Silent) case "error", "Error": config.Logger = internal.Default.LogMode(logger.Error) case "warn", "Warn": config.Logger = internal.Default.LogMode(logger.Warn) case "info", "Info": config.Logger = internal.Default.LogMode(logger.Info) default: config.Logger = internal.Default.LogMode(logger.Info) } return config }