使用Viper读取环境变量失败求助（Go语言）

看起来你遇到的问题是Viper没有把app.yaml里的环境变量名称转换成实际的环境变量值对吧？我来帮你梳理下解决方案～

首先，你的app.yaml里的配置项值写的是环境变量的名称（比如dsn: RESTFUL_APP_DSN），但Viper默认不会自动识别这些字符串为环境变量并去读取它们的实际值。我们可以通过两种方式来解决这个问题：

方案一：修改配置文件使用Viper的环境变量占位符 #

Viper支持在配置文件中直接用${环境变量名}的语法引用环境变量，这是更标准的做法。你可以先把app.yaml改成这样：

# app.yaml
dsn: ${RESTFUL_APP_DSN}
jwt_verification_key: ${RESTFUL_APP_JWT_VERIFICATION_KEY}
jwt_signing_key: ${RESTFUL_APP_JWT_SIGNING_KEY}
jwt_signing_method: "HS256"

然后在config.go里调整Viper的初始化逻辑，确保它能解析这些占位符：

package config

import (
	"fmt"
	"strings"
	"github.com/go-ozzo/ozzo-validation"
	"github.com/spf13/viper"
)

// Config stores the application-wide configurations
var Config appConfig

type appConfig struct {
	DSN                  string `mapstructure:"dsn"`
	JWTVerificationKey   string `mapstructure:"jwt_verification_key"`
	JWTSigningKey        string `mapstructure:"jwt_signing_key"`
	JWTSigningMethod     string `mapstructure:"jwt_signing_method"`
}

func init() {
	// 配置Viper读取yaml文件（路径根据你的实际情况调整）
	viper.SetConfigName("app")
	viper.SetConfigType("yaml")
	viper.AddConfigPath(".")

	// 读取配置文件
	if err := viper.ReadInConfig(); err != nil {
		fmt.Printf("Error reading config file: %v\n", err)
		panic(err)
	}

	// 开启自动读取环境变量，让Viper能解析配置里的${}占位符
	viper.AutomaticEnv()

	// 可选：如果配置键和环境变量命名规则有差异，设置键替换器（这里你的命名已经一致，可省略）
	viper.SetEnvKeyReplacer(strings.NewReplacer("_", "_"))

	// 将配置解析到全局Config变量
	if err := viper.Unmarshal(&Config); err != nil {
		fmt.Printf("Error unmarshaling config: %v\n", err)
		panic(err)
	}

	// 验证配置合法性
	if err := validation.ValidateStruct(&Config); err != nil {
		fmt.Printf("Config validation failed: %v\n", err)
		panic(err)
	}
}

方案二：保持原app.yaml，手动映射环境变量 #

如果你不想修改配置文件的写法，也可以在读取配置后，手动把每个配置项的值作为环境变量名去获取实际值：

func init() {
	// 读取配置文件的步骤和方案一一致
	viper.SetConfigName("app")
	viper.SetConfigType("yaml")
	viper.AddConfigPath(".")

	if err := viper.ReadInConfig(); err != nil {
		fmt.Printf("Error reading config file: %v\n", err)
		panic(err)
	}

	// 手动将配置项的值作为环境变量键，读取实际值
	Config.DSN = viper.GetString(viper.GetString("dsn"))
	Config.JWTVerificationKey = viper.GetString(viper.GetString("jwt_verification_key"))
	Config.JWTSigningKey = viper.GetString(viper.GetString("jwt_signing_key"))
	Config.JWTSigningMethod = viper.GetString("jwt_signing_method")

	// 验证配置
	if err := validation.ValidateStruct(&Config); err != nil {
		fmt.Printf("Config validation failed: %v\n", err)
		panic(err)
	}
}

额外注意事项 #

不管用哪种方案，都要确保你的环境变量已经正确设置：

• Linux/macOS终端：

export RESTFUL_APP_DSN="your_actual_dsn"
export RESTFUL_APP_JWT_VERIFICATION_KEY="your_verification_key"
export RESTFUL_APP_JWT_SIGNING_KEY="your_signing_key"

• Windows命令行：

set RESTFUL_APP_DSN=your_actual_dsn
set RESTFUL_APP_JWT_VERIFICATION_KEY=your_verification_key
set RESTFUL_APP_JWT_SIGNING_KEY=your_signing_key

这样调整后，Viper应该就能正确读取到环境变量的实际值了～

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