From e3cfcfff6f4cf6fd94594f9920363619d496116e Mon Sep 17 00:00:00 2001 From: bryan Date: Sat, 31 Jan 2026 08:36:56 +0800 Subject: [PATCH] update connect timeout etc. --- model.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/model.go b/model.go index d16dbac..dcc01d1 100644 --- a/model.go +++ b/model.go @@ -118,8 +118,9 @@ func DbOf(name string) *gorm.DB { 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) + 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)