Golang如何实现基础的配置管理_Golang 配置管理开发实践

server:
  host: 0.0.0.0
  port: 8080
database:
  dsn: "user:pass@tcp(localhost:3306)/mydb"
log_level: info

type Config struct {
    Server struct {
        Host string `yaml:"host"`
        Port int    `yaml:"port"`
    } `yaml:"server"`
    Database struct {
        DSN string `yaml:"dsn"`
    } `yaml:"database"`
    LogLevel string `yaml:"log_level"`
}

func LoadConfig(path string) (*Config, error) {
    data, err := os.ReadFile(path)
    if err != nil {
        return nil, err
    }
    var cfg Config
    err = yaml.Unmarshal(data, &cfg)
    if err != nil {
        return nil, err
    }
    return &cfg, nil
}

func (c *Config) ApplyEnvOverrides() {
    if host := os.Getenv("SERVER_HOST"); host != "" {
        c.Server.Host = host
    }
    if port := os.Getenv("SERVER_PORT"); port != "" {
        if p, err := strconv.Atoi(port); err == nil {
            c.Server.Port = p
        }
    }
}

import "github.com/spf13/viper"

viper.SetConfigName("config")
viper.SetConfigType("yaml")
viper.AddConfigPath(".")
viper.ReadInConfig()

var cfg Config
viper.Unmarshal(&cfg)

// 支持动态获取
port := viper.GetInt("server.port")

type App struct {
    cfg *Config
}

func NewApp(cfg *Config) *App {
    return &App{cfg: cfg}
}

func (a *App) Run() {
    // 使用 a.cfg 启动服务
}