Compare commits

..

2 Commits

Author SHA1 Message Date
69653d6bf7 update connect timeout etc. 2026-01-31 08:37:47 +08:00
e3cfcfff6f update connect timeout etc. 2026-01-31 08:36:56 +08:00

View File

@ -117,9 +117,10 @@ func DbOf(name string) *gorm.DB {
if !ok {
panic(errNoModelFound(name))
}
log.Debugf("start get db from model(%s)", name)
var ctx, cancelFunc = context.WithTimeout(context.Background(), 3*time.Second)
var db, err = gorm.NewSessionWithContext(m.db, ctx)
log.Debugf("start get db from model(%s) ...", name)
var ctx, cancelFunc = context.WithTimeout(m.db.Statement.Context, 3*time.Second)
var db = gorm.NewSessionWithContext(m.db, ctx)
var err = db.Error
if err != nil {
cancelFunc()
log.Warningf("new gorm session failed, err:[%s]", err)
@ -139,6 +140,7 @@ func RedisOf(name string) redis.Conn {
if !ok {
panic(errNoModelFound(name))
}
log.Debugf("start get redis from model(%s) ...", name)
var ctx, cancelFunc = context.WithTimeout(context.Background(), 3*time.Second)
var conn, err = m.redisPool.GetContext(ctx)
if err != nil {
@ -146,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)
panic(err)
}
log.Debugf("success get redis from model(%s)", name)
defer cancelFunc()
return conn
}