|
@@ -5,6 +5,8 @@ import (
|
|
|
"gorm.io/driver/mysql"
|
|
|
"gorm.io/gorm"
|
|
|
"gorm.io/gorm/logger"
|
|
|
+ "log"
|
|
|
+ "runtime"
|
|
|
)
|
|
|
|
|
|
var (
|
|
@@ -20,15 +22,21 @@ func SetupDBLink() error {
|
|
|
DatabaseSetting.Db,
|
|
|
DatabaseSetting.OtherParams,
|
|
|
)
|
|
|
- DBLink, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
|
|
|
- DisableForeignKeyConstraintWhenMigrating: true,
|
|
|
- Logger: logger.Default.LogMode(logger.Info),
|
|
|
- })
|
|
|
+ //判断当前系统是windows还是linux
|
|
|
+ //runtime获取的是当前Go语言框架的参数,不是实际运行的操作系统的参数
|
|
|
+ sysType := runtime.GOOS
|
|
|
+ if sysType == "windows" {
|
|
|
+ DBLink, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
|
|
|
+ DisableForeignKeyConstraintWhenMigrating: true,
|
|
|
+ Logger: logger.Default.LogMode(logger.Info),
|
|
|
+ })
|
|
|
+ log.Print("当前系统为windows")
|
|
|
+ } else if sysType == "linux" {
|
|
|
+ DBLink, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
|
|
+ }
|
|
|
+
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
- //DBLink.SingularTable(true)
|
|
|
- //DBLink.DB().SetMaxIdleConns(DatabaseSetting.MaxIdleConn)
|
|
|
- //DBLink.DB().SetMaxOpenConns(DatabaseSetting.MaxOpenConn)
|
|
|
return nil
|
|
|
}
|