From ba429ec261e64c9a08130434d2062f0cd0125df5 Mon Sep 17 00:00:00 2001 From: sujit Date: Tue, 1 Aug 2023 17:06:38 +0545 Subject: [PATCH] feat: remove resolver --- database/gorm.go | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/database/gorm.go b/database/gorm.go index 3d49d58..4134d74 100644 --- a/database/gorm.go +++ b/database/gorm.go @@ -13,7 +13,6 @@ import ( "gorm.io/driver/sqlserver" "gorm.io/gorm" glog "gorm.io/gorm/logger" - "gorm.io/plugin/dbresolver" "github.com/oarkflow/framework/contracts/database/orm" "github.com/oarkflow/framework/database/support" @@ -49,16 +48,17 @@ func NewGormDB(ctx context.Context, connection string, config *gorm.Config, disa if maxIdleConnections == 0 { maxIdleConnections = int(float64(maxOpenConnections) - (20 / float64(maxOpenConnections) * 100)) } - resolver := dbresolver.Register(dbresolver.Config{}) + sqlDB, err := db.DB() + if err != nil { + return nil, err + } + sqlDB.SetMaxIdleConns(maxIdleConnections) + sqlDB.SetMaxOpenConns(maxOpenConnections) if maxIdleTime > 0 { - resolver = resolver.SetConnMaxIdleTime(time.Duration(maxIdleTime) * time.Minute) + sqlDB.SetConnMaxIdleTime(time.Duration(maxIdleTime) * time.Minute) } if maxLifeTime > 0 { - resolver = resolver.SetConnMaxLifetime(time.Duration(maxLifeTime) * time.Hour) - } - err = db.Use(resolver.SetMaxIdleConns(maxIdleConnections).SetMaxOpenConns(maxOpenConnections)) - if err != nil { - return nil, err + sqlDB.SetConnMaxLifetime(time.Duration(maxLifeTime) * time.Hour) } } return &GormDB{