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

fix cancel bug

Browse Source
This commit is contained in:
bryan 2026-01-31 08:51:42 +08:00
parent 69653d6bf7
commit afc1ac82d5
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
model.go
View File

@ -117,16 +117,17 @@ 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(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)
panic(err)
}
defer cancelFunc()
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)
// panic(err)
//}
//defer cancelFunc()
var db = gorm.NewSession(m.db)
log.Debugf("success get db from model(%s)", name)
return db
}