Stabilize docker service startup

backend/internal/config/loadconfig.go

@@ -1,17 +1,19 @@
 package config
 
 import (
+	"errors"
 	"fmt"
 	"os"
-	"errors"
+	"strconv"
+
 	"gopkg.in/yaml.v3"
 )
 
 type Config struct {
-	Server   ServerConfig   `yaml:"server"`
-	Database DatabaseConfig `yaml:"database"`
-	Redis    RedisConfig    `yaml:"redis"`
-	RabbitMQ RabbitMQConfig `yaml:"rabbitmq"`
+	Server              ServerConfig        `yaml:"server"`
+	Database            DatabaseConfig      `yaml:"database"`
+	Redis               RedisConfig         `yaml:"redis"`
+	RabbitMQ            RabbitMQConfig      `yaml:"rabbitmq"`
 	ObservabilityConfig ObservabilityConfig `yaml:"observability"`
 }
 
@@ -45,10 +47,11 @@ type ObservabilityConfig struct {
 	Pprof PprofConfig `yaml:"pprof"`
 }
 type PprofConfig struct {
-	Enabled bool `yaml:"enabled"`
-	ApiAddr string `yaml:"api_addr"`
+	Enabled    bool   `yaml:"enabled"`
+	ApiAddr    string `yaml:"api_addr"`
 	WorkerAddr string `yaml:"worker_addr"`
 }
+
 func Load(filename string) (Config, error) {
 	data, err := os.ReadFile(filename)
 	if err != nil {
@@ -60,9 +63,71 @@ func Load(filename string) (Config, error) {
 		return Config{}, fmt.Errorf("parse config %s: %w", filename, err)
 	}
 
+	ApplyEnvOverrides(&cfg)
 	return cfg, nil
 }
 
+func ApplyEnvOverrides(cfg *Config) {
+	if cfg == nil {
+		return
+	}
+	if v := os.Getenv("SERVER_PORT"); v != "" {
+		if port, err := strconv.Atoi(v); err == nil {
+			cfg.Server.Port = port
+		}
+	}
+	if v := os.Getenv("MYSQL_HOST"); v != "" {
+		cfg.Database.Host = v
+	}
+	if v := os.Getenv("MYSQL_PORT"); v != "" {
+		if port, err := strconv.Atoi(v); err == nil {
+			cfg.Database.Port = port
+		}
+	}
+	if v := os.Getenv("MYSQL_USER"); v != "" {
+		cfg.Database.User = v
+	}
+	if v := os.Getenv("MYSQL_ROOT_PASSWORD"); v != "" {
+		cfg.Database.Password = v
+	}
+	if v := os.Getenv("MYSQL_PASSWORD"); v != "" {
+		cfg.Database.Password = v
+	}
+	if v := os.Getenv("MYSQL_DATABASE"); v != "" {
+		cfg.Database.DBName = v
+	}
+	if v := os.Getenv("REDIS_HOST"); v != "" {
+		cfg.Redis.Host = v
+	}
+	if v := os.Getenv("REDIS_PORT"); v != "" {
+		if port, err := strconv.Atoi(v); err == nil {
+			cfg.Redis.Port = port
+		}
+	}
+	if v := os.Getenv("REDIS_PASSWORD"); v != "" {
+		cfg.Redis.Password = v
+	}
+	if v := os.Getenv("REDIS_DB"); v != "" {
+		if db, err := strconv.Atoi(v); err == nil {
+			cfg.Redis.DB = db
+		}
+	}
+	if v := os.Getenv("RABBITMQ_HOST"); v != "" {
+		cfg.RabbitMQ.Host = v
+	}
+	if v := os.Getenv("RABBITMQ_PORT"); v != "" {
+		if port, err := strconv.Atoi(v); err == nil {
+			cfg.RabbitMQ.Port = port
+		}
+	}
+	if v := os.Getenv("RABBITMQ_USER"); v != "" {
+		cfg.RabbitMQ.Username = v
+	}
+	if v := os.Getenv("RABBITMQ_PASS"); v != "" {
+		cfg.RabbitMQ.Password = v
+	}
+}
+
 // bool用来表示是否使用了默认配置，true表示使用了默认配置
 func LoadLocalDev(filename string) (Config, bool, error) {
 	cfg, err := Load(filename)
@@ -76,14 +141,14 @@ func LoadLocalDev(filename string) (Config, bool, error) {
 }
 
 func DefaultLocalConfig() Config {
-	return Config{
+	cfg := Config{
 		Server: ServerConfig{
 			Port: 8080,
 		},
 		Database: DatabaseConfig{
 			Host:     "localhost",
 			Port:     3306,
-			User:	 "root",
+			User:     "root",
 			Password: "123456",
 			DBName:   "feedsystem",
 		},
@@ -101,10 +166,12 @@ func DefaultLocalConfig() Config {
 		},
 		ObservabilityConfig: ObservabilityConfig{
 			Pprof: PprofConfig{
-				Enabled: true,
-				ApiAddr: "localhost:6060",
+				Enabled:    true,
+				ApiAddr:    "localhost:6060",
 				WorkerAddr: "localhost:6061",
 			},
 		},
 	}
-}
+	ApplyEnvOverrides(&cfg)
+	return cfg
+}