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

update connect timeout etc.

Browse Source
This commit is contained in:
bryan 2026-01-31 08:37:47 +08:00
parent e3cfcfff6f
commit 69653d6bf7
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model.go
View File

@ -117,7 +117,7 @@ func DbOf(name string) *gorm.DB {
if !ok { if !ok {
panic(errNoModelFound(name)) panic(errNoModelFound(name))
} }
log.Debugf("start get db from model(%s)", name) log.Debugf("start get db from model(%s) ...", name)
var ctx, cancelFunc = context.WithTimeout(m.db.Statement.Context, 3*time.Second) var ctx, cancelFunc = context.WithTimeout(m.db.Statement.Context, 3*time.Second)
var db = gorm.NewSessionWithContext(m.db, ctx) var db = gorm.NewSessionWithContext(m.db, ctx)
var err = db.Error var err = db.Error
@ -140,6 +140,7 @@ func RedisOf(name string) redis.Conn {
if !ok { if !ok {
panic(errNoModelFound(name)) panic(errNoModelFound(name))
} }
log.Debugf("start get redis from model(%s) ...", name)
var ctx, cancelFunc = context.WithTimeout(context.Background(), 3*time.Second) var ctx, cancelFunc = context.WithTimeout(context.Background(), 3*time.Second)
var conn, err = m.redisPool.GetContext(ctx) var conn, err = m.redisPool.GetContext(ctx)
if err != nil { if err != nil {
@ -147,6 +148,7 @@ func RedisOf(name string) redis.Conn {
log.Warningf("get redis connect from pool failed, err:[%s], pool stats:[%+v]", err, m.redisPool.Stats) log.Warningf("get redis connect from pool failed, err:[%s], pool stats:[%+v]", err, m.redisPool.Stats)
panic(err) panic(err)
} }
log.Debugf("success get redis from model(%s)", name)
defer cancelFunc() defer cancelFunc()
return conn return conn
} }