fix: cap mysql connection pool

2026-09-27 18:36:00 +08:00
parent 4f69dc302b
commit 5bf68727fa
4 changed files with 40 additions and 9 deletions
@@ -24,7 +24,9 @@ type Config struct {
 }

 type MySQLConfig struct {
-	DSN string `json:"dsn"`
+	DSN          string `json:"dsn"`
+	MaxOpenConns int    `json:"max_open_conns"`
+	MaxIdleConns int    `json:"max_idle_conns"`
 }

 type RedisConfig struct {
@@ -90,7 +92,9 @@ func defaultConfig() *Config {
 	return &Config{
 		Listen: ":6061",
 		MySQL: MySQLConfig{
-			DSN: "root:root@tcp(127.0.0.1:3306)/meeting_im?parseTime=true&charset=utf8mb4,utf8&loc=Local",
+			DSN:          "root:root@tcp(127.0.0.1:3306)/meeting_im?parseTime=true&charset=utf8mb4,utf8&loc=Local",
+			MaxOpenConns: 15,
+			MaxIdleConns: 5,
 		},
 		Redis: RedisConfig{
 			Addr:      "127.0.0.1:6379",
@@ -120,6 +124,8 @@ func defaultConfig() *Config {
 func applyEnv(cfg *Config) {
 	setString(&cfg.Listen, "IM_WORKBENCH_LISTEN")
 	setString(&cfg.MySQL.DSN, "IM_WORKBENCH_MYSQL_DSN")
+	setInt(&cfg.MySQL.MaxOpenConns, "IM_WORKBENCH_MYSQL_MAX_OPEN_CONNS")
+	setInt(&cfg.MySQL.MaxIdleConns, "IM_WORKBENCH_MYSQL_MAX_IDLE_CONNS")
 	setString(&cfg.Redis.Addr, "IM_WORKBENCH_REDIS_ADDR")
 	setString(&cfg.Redis.Username, "IM_WORKBENCH_REDIS_USERNAME")
 	setString(&cfg.Redis.Password, "IM_WORKBENCH_REDIS_PASSWORD")
@@ -214,6 +220,14 @@ func setString(target *string, key string) {
 	}
 }

+func setInt(target *int, key string) {
+	if value := strings.TrimSpace(os.Getenv(key)); value != "" {
+		if parsed, err := strconv.Atoi(value); err == nil {
+			*target = parsed
+		}
+	}
+}
+
 func setBool(target *bool, key string) {
 	switch strings.ToLower(strings.TrimSpace(os.Getenv(key))) {
 	case "1", "true", "yes", "on":