Golang使用Viper加载YAML与环境变量失败，求解决方案

问题原因 #

你的代码存在两个核心问题：

1. .env文件未被加载：Viper的AutomaticEnv()仅读取系统环境变量，但不会自动加载.env文件中的变量，导致配置文件里的${VAR}占位符找不到对应值。
2. 未开启配置文件占位符替换：Viper默认不会解析YAML中的${VAR}格式占位符，需要显式配置环境变量替换逻辑。

解决方案 #

步骤1：加载.env文件

使用github.com/joho/godotenv包将.env变量加载到系统环境中（也可使用Viper自身方法，但godotenv更直观）。先安装依赖：

go get github.com/joho/godotenv

步骤2：修正Viper配置代码

确保结构体添加mapstructure标签（Viper默认用该标签映射YAML字段），并开启环境变量替换：

package main

import (
	"log"
	"strings"

	"github.com/joho/godotenv"
	"github.com/spf13/viper"
)

type Config struct {
	Server struct {
		Name string `mapstructure:"name"`
		Host string `mapstructure:"host"`
		Port int    `mapstructure:"port"`
	} `mapstructure:"server"`
	Database struct {
		Drivers struct {
			Mysql struct {
				Driver   string `mapstructure:"driver"`
				Host     string `mapstructure:"host"`
				Username string `mapstructure:"username"`
				Password string `mapstructure:"password"`
				Dbname   string `mapstructure:"dbname"`
				Port     int    `mapstructure:"port"`
			} `mapstructure:"mysql"`
			Postgres struct {
				Driver   string `mapstructure:"driver"`
				Host     string `mapstructure:"host"`
				Username string `mapstructure:"username"`
				Password string `mapstructure:"password"`
				Dbname   string `mapstructure:"dbname"`
				Port     int    `mapstructure:"port"`
			} `mapstructure:"postgres"`
		} `mapstructure:"drivers"`
	} `mapstructure:"database"`
}

func main() {
	// 加载.env文件到系统环境变量
	if err := godotenv.Load(); err != nil {
		log.Printf("警告：加载.env文件失败: %v", err)
	}

	// 配置Viper
	viper.AddConfigPath(".")
	viper.SetConfigName("config") // 自动识别.yml后缀
	viper.SetConfigType("yaml")

	// 开启环境变量读取，并支持替换配置文件中的${VAR}占位符
	viper.AutomaticEnv()
	// 可选：支持嵌套配置键与环境变量的映射（如database.drivers.mysql.host对应DATABASE_DRIVERS_MYSQL_HOST）
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))

	// 读取配置文件
	if err := viper.ReadInConfig(); err != nil {
		log.Fatalf("读取配置文件失败: %v", err)
	}

	// 解析到结构体
	var cfg Config
	if err := viper.Unmarshal(&cfg); err != nil {
		log.Fatalf("解析配置到结构体失败: %v", err)
	}

	// 验证结果
	log.Printf("服务器名称: %s", cfg.Server.Name)
	log.Printf("MySQL主机: %s", cfg.Database.Drivers.Mysql.Host)
	log.Printf("MySQL用户名: %s", cfg.Database.Drivers.Mysql.Username)
}

关键说明 #

• godotenv.Load()：将.env中的变量注入到系统环境，确保Viper能读取到这些变量。
• viper.AutomaticEnv()：让Viper优先使用环境变量覆盖配置文件中的对应值，同时解析${VAR}占位符。
• mapstructure标签：必须与YAML中的字段名完全对应，否则Viper无法正确映射结构体字段。

内容的提问来源于stack exchange，提问作者Dean