gomod/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

optimize config

Browse Source
This commit is contained in:
bryan 2025-04-08 08:49:31 +08:00
parent 32d791051b
commit 2f869861ef
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
gorm.go
View File

@ -38,8 +38,8 @@ type Config struct {
ExtraParameters string `toml:"extra_parameters"` //数据库连接扩展参数
}
func NewDB(dbcfg *Config) (*gorm.DB, error) {
return open(dbcfg)
func NewDB(dbcfg Config) (*gorm.DB, error) {
return open(&dbcfg)
}
func NewSession(db *gorm.DB) *gorm.DB {
@ -67,7 +67,7 @@ func (dbcfg *Config) Id() string {
return dbcfg.id
}
func GetDsn(dbcfg *Config) (dsn string) {
func (dbcfg *Config) GetDsn() (dsn string) {
switch dbcfg.Type {
case "mysql":
//dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
@ -97,11 +97,21 @@ func GetDsn(dbcfg *Config) (dsn string) {
return dsn
}
func (dbcfg Config) GetSecDsn() string {
dsn := dbcfg.GetDsn()
if len(dbcfg.Password) > 5 {
p, l := dbcfg.Password, len(dbcfg.Password)
return strings.Replace(dsn, p[2:l-3], "*****", 1)
} else {
return dsn
}
}
func open(dbcfg *Config) (*gorm.DB, error) {
var (
db *gorm.DB
err error
dsn = GetDsn(dbcfg)
dsn = dbcfg.GetDsn()
cacheValue, inCacheYes = _dbCache.Load(dbcfg.Id())
cachedDb, inCacheYes2 = cacheValue.(*gorm.DB)
)