193 lines
7.2 KiB
Go
193 lines
7.2 KiB
Go
package gorm_v2
|
||
|
||
import (
|
||
"catface/app/global/my_errors"
|
||
"catface/app/global/variable"
|
||
"errors"
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"go.uber.org/zap"
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/driver/postgres"
|
||
"gorm.io/driver/sqlserver"
|
||
"gorm.io/gorm"
|
||
gormLog "gorm.io/gorm/logger"
|
||
"gorm.io/plugin/dbresolver"
|
||
)
|
||
|
||
// 获取一个 mysql 客户端
|
||
func GetOneMysqlClient() (*gorm.DB, error) {
|
||
sqlType := "Mysql"
|
||
readDbIsOpen := variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + ".IsOpenReadDb")
|
||
return GetSqlDriver(sqlType, readDbIsOpen)
|
||
}
|
||
|
||
// 获取一个 sqlserver 客户端
|
||
func GetOneSqlserverClient() (*gorm.DB, error) {
|
||
sqlType := "SqlServer"
|
||
readDbIsOpen := variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + ".IsOpenReadDb")
|
||
return GetSqlDriver(sqlType, readDbIsOpen)
|
||
}
|
||
|
||
// 获取一个 postgresql 客户端
|
||
func GetOnePostgreSqlClient() (*gorm.DB, error) {
|
||
sqlType := "Postgresql"
|
||
readDbIsOpen := variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + ".IsOpenReadDb")
|
||
return GetSqlDriver(sqlType, readDbIsOpen)
|
||
}
|
||
|
||
// 获取数据库驱动, 可以通过options 动态参数连接任意多个数据库
|
||
func GetSqlDriver(sqlType string, readDbIsOpen int, dbConf ...ConfigParams) (*gorm.DB, error) {
|
||
|
||
var dbDialector gorm.Dialector
|
||
if val, err := getDbDialector(sqlType, "Write", dbConf...); err != nil {
|
||
variable.ZapLog.Error(my_errors.ErrorsDialectorDbInitFail+sqlType, zap.Error(err))
|
||
} else {
|
||
dbDialector = val
|
||
}
|
||
gormDb, err := gorm.Open(dbDialector, &gorm.Config{
|
||
SkipDefaultTransaction: true,
|
||
PrepareStmt: true,
|
||
Logger: redefineLog(sqlType), //拦截、接管 gorm v2 自带日志
|
||
})
|
||
if err != nil {
|
||
//gorm 数据库驱动初始化失败
|
||
return nil, err
|
||
}
|
||
|
||
// 如果开启了读写分离，配置读数据库（resource、read、replicas）
|
||
// 读写分离配置只
|
||
if readDbIsOpen == 1 {
|
||
if val, err := getDbDialector(sqlType, "Read", dbConf...); err != nil {
|
||
variable.ZapLog.Error(my_errors.ErrorsDialectorDbInitFail+sqlType, zap.Error(err))
|
||
} else {
|
||
dbDialector = val
|
||
}
|
||
resolverConf := dbresolver.Config{
|
||
Replicas: []gorm.Dialector{dbDialector}, // 读 操作库，查询类
|
||
Policy: dbresolver.RandomPolicy{}, // sources/replicas 负载均衡策略适用于
|
||
}
|
||
err = gormDb.Use(dbresolver.Register(resolverConf).SetConnMaxIdleTime(time.Second * 30).
|
||
SetConnMaxLifetime(variable.ConfigGormv2Yml.GetDuration("Gormv2."+sqlType+".Read.SetConnMaxLifetime") * time.Second).
|
||
SetMaxIdleConns(variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + ".Read.SetMaxIdleConns")).
|
||
SetMaxOpenConns(variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + ".Read.SetMaxOpenConns")))
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 查询没有数据，屏蔽 gorm v2 包中会爆出的错误
|
||
// https://github.com/go-gorm/gorm/issues/3789 此 issue 所反映的问题就是我们本次解决掉的
|
||
_ = gormDb.Callback().Query().Before("gorm:query").Register("disable_raise_record_not_found", MaskNotDataError)
|
||
|
||
// https://github.com/go-gorm/gorm/issues/4838
|
||
_ = gormDb.Callback().Create().Before("gorm:before_create").Register("CreateBeforeHook", CreateBeforeHook)
|
||
// 为了完美支持gorm的一系列回调函数
|
||
_ = gormDb.Callback().Update().Before("gorm:before_update").Register("UpdateBeforeHook", UpdateBeforeHook)
|
||
|
||
// 为主连接设置连接池(43行返回的数据库驱动指针)
|
||
if rawDb, err := gormDb.DB(); err != nil {
|
||
return nil, err
|
||
} else {
|
||
rawDb.SetConnMaxIdleTime(time.Second * 30)
|
||
rawDb.SetConnMaxLifetime(variable.ConfigGormv2Yml.GetDuration("Gormv2."+sqlType+".Write.SetConnMaxLifetime") * time.Second)
|
||
rawDb.SetMaxIdleConns(variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + ".Write.SetMaxIdleConns"))
|
||
rawDb.SetMaxOpenConns(variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + ".Write.SetMaxOpenConns"))
|
||
// 全局sql的debug配置
|
||
if variable.ConfigGormv2Yml.GetBool("Gormv2.SqlDebug") {
|
||
return gormDb.Debug(), nil
|
||
} else {
|
||
return gormDb, nil
|
||
}
|
||
}
|
||
}
|
||
|
||
// 获取一个数据库方言(Dialector),通俗的说就是根据不同的连接参数，获取具体的一类数据库的连接指针
|
||
func getDbDialector(sqlType, readWrite string, dbConf ...ConfigParams) (gorm.Dialector, error) {
|
||
var dbDialector gorm.Dialector
|
||
dsn := getDsn(sqlType, readWrite, dbConf...)
|
||
switch strings.ToLower(sqlType) {
|
||
case "mysql":
|
||
dbDialector = mysql.Open(dsn)
|
||
case "sqlserver", "mssql":
|
||
dbDialector = sqlserver.Open(dsn)
|
||
case "postgres", "postgresql", "postgre":
|
||
dbDialector = postgres.Open(dsn)
|
||
default:
|
||
return nil, errors.New(my_errors.ErrorsDbDriverNotExists + sqlType)
|
||
}
|
||
return dbDialector, nil
|
||
}
|
||
|
||
// 根据配置参数生成数据库驱动 dsn
|
||
func getDsn(sqlType, readWrite string, dbConf ...ConfigParams) string {
|
||
Host := variable.ConfigGormv2Yml.GetString("Gormv2." + sqlType + "." + readWrite + ".Host")
|
||
DataBase := variable.ConfigGormv2Yml.GetString("Gormv2." + sqlType + "." + readWrite + ".DataBase")
|
||
Port := variable.ConfigGormv2Yml.GetInt("Gormv2." + sqlType + "." + readWrite + ".Port")
|
||
User := variable.ConfigGormv2Yml.GetString("Gormv2." + sqlType + "." + readWrite + ".User")
|
||
Pass := variable.ConfigGormv2Yml.GetString("Gormv2." + sqlType + "." + readWrite + ".Pass")
|
||
Charset := variable.ConfigGormv2Yml.GetString("Gormv2." + sqlType + "." + readWrite + ".Charset")
|
||
|
||
if len(dbConf) > 0 {
|
||
if strings.ToLower(readWrite) == "write" {
|
||
if len(dbConf[0].Write.Host) > 0 {
|
||
Host = dbConf[0].Write.Host
|
||
}
|
||
if len(dbConf[0].Write.DataBase) > 0 {
|
||
DataBase = dbConf[0].Write.DataBase
|
||
}
|
||
if dbConf[0].Write.Port > 0 {
|
||
Port = dbConf[0].Write.Port
|
||
}
|
||
if len(dbConf[0].Write.User) > 0 {
|
||
User = dbConf[0].Write.User
|
||
}
|
||
if len(dbConf[0].Write.Pass) > 0 {
|
||
Pass = dbConf[0].Write.Pass
|
||
}
|
||
if len(dbConf[0].Write.Charset) > 0 {
|
||
Charset = dbConf[0].Write.Charset
|
||
}
|
||
} else {
|
||
if len(dbConf[0].Read.Host) > 0 {
|
||
Host = dbConf[0].Read.Host
|
||
}
|
||
if len(dbConf[0].Read.DataBase) > 0 {
|
||
DataBase = dbConf[0].Read.DataBase
|
||
}
|
||
if dbConf[0].Read.Port > 0 {
|
||
Port = dbConf[0].Read.Port
|
||
}
|
||
if len(dbConf[0].Read.User) > 0 {
|
||
User = dbConf[0].Read.User
|
||
}
|
||
if len(dbConf[0].Read.Pass) > 0 {
|
||
Pass = dbConf[0].Read.Pass
|
||
}
|
||
if len(dbConf[0].Read.Charset) > 0 {
|
||
Charset = dbConf[0].Read.Charset
|
||
}
|
||
}
|
||
}
|
||
|
||
switch strings.ToLower(sqlType) {
|
||
case "mysql":
|
||
// INFO 这里改动为 True； 有效
|
||
return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=true&loc=Local", User, Pass, Host, Port, DataBase, Charset)
|
||
case "sqlserver", "mssql":
|
||
return fmt.Sprintf("server=%s;port=%d;database=%s;user id=%s;password=%s;encrypt=disable", Host, Port, DataBase, User, Pass)
|
||
case "postgresql", "postgre", "postgres":
|
||
return fmt.Sprintf("host=%s port=%d dbname=%s user=%s password=%s sslmode=disable TimeZone=Asia/Shanghai", Host, Port, DataBase, User, Pass)
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 创建自定义日志模块，对 gorm 日志进行拦截、
|
||
func redefineLog(sqlType string) gormLog.Interface {
|
||
return createCustomGormLog(sqlType,
|
||
SetInfoStrFormat("[info] %s\n"), SetWarnStrFormat("[warn] %s\n"), SetErrStrFormat("[error] %s\n"),
|
||
SetTraceStrFormat("[traceStr] %s [%.3fms] [rows:%v] %s\n"), SetTracWarnStrFormat("[traceWarn] %s %s [%.3fms] [rows:%v] %s\n"), SetTracErrStrFormat("[traceErr] %s %s [%.3fms] [rows:%v] %s\n"))
|
||
}
|