Compare commits
2 Commits
69125cfbc2
...
69653d6bf7
| Author | SHA1 | Date | |
|---|---|---|---|
| 69653d6bf7 | |||
| e3cfcfff6f |
9
model.go
9
model.go
@ -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
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user