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 { 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
if err != nil { //if err != nil {
cancelFunc() // cancelFunc()
log.Warningf("new gorm session failed, err:[%s]", err) // log.Warningf("new gorm session failed, err:[%s]", err)
panic(err) // panic(err)
} //}
defer cancelFunc() //defer cancelFunc()
var db = gorm.NewSession(m.db)
log.Debugf("success get db from model(%s)", name) log.Debugf("success get db from model(%s)", name)
return db return db
} }