Merge pull request #191 from sunmi-OS/feature/imp

update orm
sunmi-OS · Feb 6, 2025 · 32af8af · 32af8af
2 parents a80e62e + 4721308
commit 32af8af
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/db/orm/gorm.go b/db/orm/gorm.go
@@ -164,6 +164,11 @@ func openORM(dbname string) (*gorm.DB, error) {
 		db.SetMaxIdleConns(viper.C.GetInt(dbname + ".MaxIdleConns"))
 		// 最大打开连接数
 		db.SetMaxOpenConns(viper.C.GetInt(dbname + ".MaxOpenConns"))
+		// Set connMaxIdleTime
+		connMaxIdleTime := viper.GetEnvConfig(dbname + ".ConnMaxIdleTime").Int64()
+		if connMaxIdleTime > 0 {
+			db.SetConnMaxIdleTime(time.Duration(connMaxIdleTime) * time.Second)
+		}
 		return orm, nil
 	default:
 		return nil, fmt.Errorf("not support sql driver [%s]", dbType)