mysql.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. package initialize
  2. import (
  3. "gin-vue-admin/global"
  4. "go.uber.org/zap"
  5. "gorm.io/driver/mysql"
  6. "gorm.io/gorm"
  7. "gorm.io/gorm/logger"
  8. "os"
  9. )
  10. // 初始化数据库并产生数据库全局变量
  11. func Mysql() {
  12. admin := global.GVA_CONFIG.Mysql
  13. mysqlConfig := mysql.Config{
  14. DSN: admin.Username + ":" + admin.Password + "@(" + admin.Path + ")/" + admin.Dbname + "?" + admin.Config, // DSN data source name
  15. DefaultStringSize: 191, // string 类型字段的默认长度
  16. DisableDatetimePrecision: true, // 禁用 datetime 精度,MySQL 5.6 之前的数据库不支持
  17. DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式,MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
  18. DontSupportRenameColumn: true, // 用 `change` 重命名列,MySQL 8 之前的数据库和 MariaDB 不支持重命名列
  19. SkipInitializeWithVersion: false, // 根据版本自动配置
  20. }
  21. var gormConfig *gorm.Config
  22. if admin.LogMode { //根据配置决定是否开启日志
  23. gormConfig = &gorm.Config{
  24. Logger: logger.Default.LogMode(logger.Info),
  25. DisableForeignKeyConstraintWhenMigrating: true,
  26. }
  27. } else {
  28. gormConfig = &gorm.Config{
  29. Logger: logger.Default.LogMode(logger.Silent),
  30. DisableForeignKeyConstraintWhenMigrating: true,
  31. }
  32. }
  33. if db, err := gorm.Open(mysql.New(mysqlConfig), gormConfig); err != nil {
  34. global.GVA_LOG.Error("MySQL启动异常", zap.Any("err", err))
  35. os.Exit(0)
  36. } else {
  37. global.GVA_DB = db
  38. sqlDB, _ := db.DB()
  39. sqlDB.SetMaxIdleConns(admin.MaxIdleConns)
  40. sqlDB.SetMaxOpenConns(admin.MaxOpenConns)
  41. }
  42. }