47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
|
package internal
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/flipped-aurora/gin-vue-admin/server/config"
|
||
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"gorm.io/gorm/logger"
|
||
|
|
"gorm.io/gorm/schema"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var Gorm = new(_gorm)
|
||
|
|
|
||
|
|
type _gorm struct{}
|
||
|
|
|
||
|
|
// Config gorm 自定义配置
|
||
|
|
// Author [SliverHorn](https://github.com/SliverHorn)
|
||
|
|
func (g *_gorm) Config(prefix string, singular bool) *gorm.Config {
|
||
|
|
var general config.GeneralDB
|
||
|
|
switch global.GVA_CONFIG.System.DbType {
|
||
|
|
case "mysql":
|
||
|
|
general = global.GVA_CONFIG.Mysql.GeneralDB
|
||
|
|
case "pgsql":
|
||
|
|
general = global.GVA_CONFIG.Pgsql.GeneralDB
|
||
|
|
case "oracle":
|
||
|
|
general = global.GVA_CONFIG.Oracle.GeneralDB
|
||
|
|
case "sqlite":
|
||
|
|
general = global.GVA_CONFIG.Sqlite.GeneralDB
|
||
|
|
case "mssql":
|
||
|
|
general = global.GVA_CONFIG.Mssql.GeneralDB
|
||
|
|
default:
|
||
|
|
general = global.GVA_CONFIG.Mysql.GeneralDB
|
||
|
|
}
|
||
|
|
return &gorm.Config{
|
||
|
|
Logger: logger.New(NewWriter(general), logger.Config{
|
||
|
|
SlowThreshold: 200 * time.Millisecond,
|
||
|
|
LogLevel: general.LogLevel(),
|
||
|
|
Colorful: true,
|
||
|
|
}),
|
||
|
|
NamingStrategy: schema.NamingStrategy{
|
||
|
|
TablePrefix: prefix,
|
||
|
|
SingularTable: singular,
|
||
|
|
},
|
||
|
|
DisableForeignKeyConstraintWhenMigrating: true,
|
||
|
|
}
|
||
|
|
}
|