80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
|
package orm
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"SciencesServer/config"
|
|||
|
|
"SciencesServer/serve/orm/logic"
|
|||
|
|
"fmt"
|
|||
|
|
"log"
|
|||
|
|
"os"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"gorm.io/gorm/logger"
|
|||
|
|
"gorm.io/gorm/schema"
|
|||
|
|
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
orm *gorm.DB
|
|||
|
|
|
|||
|
|
engines = map[string]func() logic.IEngine{
|
|||
|
|
"mysql": mysql, "sqlite": sqlite,
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func mysql() logic.IEngine {
|
|||
|
|
return &logic.Mysql{
|
|||
|
|
User: config.SettingInfo.Engine.Mysql.User, Password: config.SettingInfo.Engine.Mysql.Password,
|
|||
|
|
Host: config.SettingInfo.Engine.Mysql.Host, Port: config.SettingInfo.Engine.Mysql.Port,
|
|||
|
|
DBName: config.SettingInfo.Engine.Mysql.DBName, Parameters: config.SettingInfo.Engine.Mysql.Parameters,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func sqlite() logic.IEngine {
|
|||
|
|
return &logic.Sqlite{Path: config.SettingInfo.Engine.Sqlite.Path, Name: config.SettingInfo.Engine.Sqlite.Name}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Init() {
|
|||
|
|
handle, has := engines[config.SettingInfo.Engine.DBMode]
|
|||
|
|
|
|||
|
|
if !has {
|
|||
|
|
panic(fmt.Sprintf("Unknown Engine Mode：%d", config.SettingInfo.Engine.DBMode))
|
|||
|
|
}
|
|||
|
|
option := &gorm.Config{
|
|||
|
|
DisableForeignKeyConstraintWhenMigrating: true,
|
|||
|
|
NamingStrategy: schema.NamingStrategy{
|
|||
|
|
TablePrefix: config.SettingInfo.Engine.TablePrefix,
|
|||
|
|
SingularTable: !config.SettingInfo.Engine.Complex,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
if config.SettingInfo.Engine.Debug {
|
|||
|
|
option.Logger = logger.New(
|
|||
|
|
log.New(os.Stdout, "\r\n", log.LstdFlags),
|
|||
|
|
logger.Config{
|
|||
|
|
SlowThreshold: time.Second,
|
|||
|
|
LogLevel: logger.Info,
|
|||
|
|
Colorful: false,
|
|||
|
|
IgnoreRecordNotFoundError: true,
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
}
|
|||
|
|
db, err := gorm.Open(handle().DSN(), option)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
panic("Orm Open Error：" + err.Error())
|
|||
|
|
}
|
|||
|
|
_db, _ := db.DB()
|
|||
|
|
_db.SetMaxIdleConns(config.SettingInfo.Engine.MaxIdleConns)
|
|||
|
|
_db.SetMaxOpenConns(config.SettingInfo.Engine.MaxOpenConns)
|
|||
|
|
_db.SetConnMaxLifetime(time.Duration(config.SettingInfo.Engine.MaxLifetime) * time.Second)
|
|||
|
|
|
|||
|
|
orm = db
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func GetDB() *gorm.DB {
|
|||
|
|
if _, err := orm.DB(); err != nil {
|
|||
|
|
Init()
|
|||
|
|
}
|
|||
|
|
return orm
|
|||
|
|
}
|