|
@@ -9,10 +9,11 @@ import (
|
|
|
"gin-vue-admin/model/request"
|
|
|
"gin-vue-admin/source"
|
|
|
"gin-vue-admin/utils"
|
|
|
+ "path/filepath"
|
|
|
+
|
|
|
"github.com/spf13/viper"
|
|
|
"gorm.io/driver/mysql"
|
|
|
"gorm.io/gorm"
|
|
|
- "path/filepath"
|
|
|
)
|
|
|
|
|
|
//@author: [songzhibin97](https://github.com/songzhibin97)
|
|
@@ -71,13 +72,6 @@ func initDB(InitDBFunctions ...model.InitDBFunc) (err error) {
|
|
|
//@return: error
|
|
|
|
|
|
func InitDB(conf request.InitDB) error {
|
|
|
- BaseMysql := config.Mysql{
|
|
|
- Path: "",
|
|
|
- Dbname: "",
|
|
|
- Username: "",
|
|
|
- Password: "",
|
|
|
- Config: "charset=utf8mb4&parseTime=True&loc=Local",
|
|
|
- }
|
|
|
|
|
|
if conf.Host == "" {
|
|
|
conf.Host = "127.0.0.1"
|
|
@@ -100,15 +94,11 @@ func InitDB(conf request.InitDB) error {
|
|
|
Config: "charset=utf8mb4&parseTime=True&loc=Local",
|
|
|
}
|
|
|
|
|
|
- if err := writeConfig(global.GVA_VP, MysqlConfig); err != nil {
|
|
|
- return err
|
|
|
- }
|
|
|
- m := global.GVA_CONFIG.Mysql
|
|
|
- if m.Dbname == "" {
|
|
|
+ if MysqlConfig.Dbname == "" {
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
- linkDns := m.Username + ":" + m.Password + "@tcp(" + m.Path + ")/" + m.Dbname + "?" + m.Config
|
|
|
+ linkDns := MysqlConfig.Username + ":" + MysqlConfig.Password + "@tcp(" + MysqlConfig.Path + ")/" + MysqlConfig.Dbname + "?" + MysqlConfig.Config
|
|
|
mysqlConfig := mysql.Config{
|
|
|
DSN: linkDns, // DSN data source name
|
|
|
DefaultStringSize: 191, // string 类型字段的默认长度
|
|
@@ -118,15 +108,11 @@ func InitDB(conf request.InitDB) error {
|
|
|
SkipInitializeWithVersion: false, // 根据版本自动配置
|
|
|
}
|
|
|
if db, err := gorm.Open(mysql.New(mysqlConfig), &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true}); err != nil {
|
|
|
- //global.GVA_LOG.Error("MySQL启动异常!", zap.Any("err", err))
|
|
|
- //os.Exit(0)
|
|
|
- //return nil
|
|
|
- _ = writeConfig(global.GVA_VP, BaseMysql)
|
|
|
return nil
|
|
|
} else {
|
|
|
sqlDB, _ := db.DB()
|
|
|
- sqlDB.SetMaxIdleConns(m.MaxIdleConns)
|
|
|
- sqlDB.SetMaxOpenConns(m.MaxOpenConns)
|
|
|
+ sqlDB.SetMaxIdleConns(MysqlConfig.MaxIdleConns)
|
|
|
+ sqlDB.SetMaxOpenConns(MysqlConfig.MaxOpenConns)
|
|
|
global.GVA_DB = db
|
|
|
}
|
|
|
|
|
@@ -147,7 +133,7 @@ func InitDB(conf request.InitDB) error {
|
|
|
model.SysOperationRecord{},
|
|
|
)
|
|
|
if err != nil {
|
|
|
- _ = writeConfig(global.GVA_VP, BaseMysql)
|
|
|
+ global.GVA_DB = nil
|
|
|
return err
|
|
|
}
|
|
|
err = initDB(
|
|
@@ -163,7 +149,10 @@ func InitDB(conf request.InitDB) error {
|
|
|
source.File,
|
|
|
source.BaseMenu)
|
|
|
if err != nil {
|
|
|
- _ = writeConfig(global.GVA_VP, BaseMysql)
|
|
|
+ global.GVA_DB = nil
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ if err = writeConfig(global.GVA_VP, MysqlConfig); err != nil {
|
|
|
return err
|
|
|
}
|
|
|
global.GVA_CONFIG.AutoCode.Root, _ = filepath.Abs("..")
|