You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
951 B
39 lines
951 B
package internal
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"autocode/global"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
var Gorm = new(_gorm)
|
|
|
|
type _gorm struct{}
|
|
|
|
// Config gorm 自定义配置
|
|
// Author [SliverHorn](https://github.com/SliverHorn)
|
|
func (g *_gorm) Config() *gorm.Config {
|
|
config := &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true}
|
|
_default := logger.New(NewWriter(log.New(os.Stdout, "\r\n", log.LstdFlags)), logger.Config{
|
|
SlowThreshold: 200 * time.Millisecond,
|
|
LogLevel: logger.Warn,
|
|
Colorful: true,
|
|
})
|
|
switch global.GVA_CONFIG.Mysql.LogMode {
|
|
case "silent", "Silent":
|
|
config.Logger = _default.LogMode(logger.Silent)
|
|
case "error", "Error":
|
|
config.Logger = _default.LogMode(logger.Error)
|
|
case "warn", "Warn":
|
|
config.Logger = _default.LogMode(logger.Warn)
|
|
case "info", "Info":
|
|
config.Logger = _default.LogMode(logger.Info)
|
|
default:
|
|
config.Logger = _default.LogMode(logger.Info)
|
|
}
|
|
return config
|
|
}
|