go项目中环境变量的配置方法

这篇文章主要介绍“Go项目中环境变量的配置方法”，在日常操作中，相信很多人在go项目中环境变量的配置方法问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家解答”go项目中环境变量的配置方法”的疑惑有所帮助！接下来，请跟着小编一起来学习吧！

目录

• 一、为什么要配置文件

• 二、在go中使用配置文件

• 三、使用结构体的方式读取

• 四、区分开发环境和生产环境

• 五、在gin中集成配置文件

一、为什么要配置文件

容易编辑，配置比较集中，方便修改，在大业务量的系统里面，通过配置会方便后人理解整个系统的架构
2、做到业务代码和环境解耦，环境配置变量不侵入到代码层面
3、不同的环境切换比较方便

二、在go中使用配置文件

viper网站地址

本地安装

go get GitHub.com/spf13/viper

简单的读取配置文件

package mainimport ( "fmt" "github.com/spf13/viper")func main() { v := viper.New() v.SetConfigFile("config.yaml") if err := v.ReadInConfig(); err != nil {  fmt.Println("读取配置文件失败", err) } fmt.Println(v.Get("name"))}

在文件的同级下创建一个config.yaml的文件

name: "test"

使用go run 文件名运行文件，这样直接运行会报错找不到文件

建议配置后可以使用右键运行项目

package mainimport ( "fmt" "github.com/spf13/viper" "os" "path")func main() { // 获取项目目录 workDir, _ := os.Getwd() v := viper.New()  // 直接拼接文件目录读取 v.SetConfigFile(path.Join(workDir, "test/config.yaml")) if err := v.ReadInConfig(); err != nil {  fmt.Println("读取配置文件失败", err) } fmt.Println(v.Get("name"))}

三、使用结构体的方式读取

简单的使用

package mainimport ( "fmt" "github.com/spf13/viper" "os" "path")type ServerConfig struct { Name string `mapstructure:"name"`}func main() { // 获取项目目录 workDir, _ := os.Getwd() v := viper.New() v.SetConfigFile(path.Join(workDir, "test/config.yaml")) if err := v.ReadInConfig(); err != nil {  fmt.Println("读取配置文件失败", err) } serverConfig := ServerConfig{} if err := v.Unmarshal(&serverConfig); err != nil {  fmt.Println("解析结构体失败", err) } fmt.Println(serverConfig.Name)}

结构体的嵌套，很常见的是mysql数据库的配置

// 定义Mysql的type mysqlConfig struct { Host string `mapstructure:"host"` Port int `mapstructure:"port"` Username string `mapstructure:"username"` PassWord string `mapstructure:"password"`}type ServerConfig struct { Name string `mapstructure:"name"` // 嵌套使用,注意这个地方要和yaml中一样 MysqlConfig MySQLConfig `mapstructure:"mysql"`}// 直接使用fmt.Println(serverConfig.MySqlConfig)

四、区分开发环境和生产环境

在Mac电脑上配置环境变量，在.bash_profile中导出一个环境变量，注意要关闭goland重新打开

# vim .bash_profileexport IS_DEV= truesource .bash_profile

定义方法直接获取环境变量

func GetEnvInfo(env string) bool { viper.AutomaticEnv() return viper.GetBool(env)}func main() { fmt.Println(GetEnvInfo("IS_DEV"))}

在目录下创建一个application.dev.yml和application.prod.yml的文件，分别存放不同环境下的配置

使用环境变量来读取配置完整代码

func GetEnvInfo1(env string) bool { viper.AutomaticEnv() return viper.GetBool(env)}// 定义mysql的type MySQLConfig struct { Host string `mapstructure:"host"` Port int `mapstructure:"port"` Username string `mapstructure:"username"` Password string `mapstructure:"password"`}type ServerConfig struct { Name string `mapstructure:"name"` // 嵌套使用,注意这个地方要和yaml中一样 MySqlConfig MySQLConfig `mapstructure:"mysql"`}func main() { // 获取项目目录 workDir, _ := os.Getwd() isDev := GetEnvInfo1("IS_DEV") configFileName := path.Join(workDir, "11.config/application.prod.yml") if isDev {  configFileName = path.Join(workDir, "11.config/application.dev.yml") } v := viper.New() v.SetConfigFile(configFileName) if err := v.ReadInConfig(); err != nil {  fmt.Println("读取配置文件失败", err) } serverConfig := ServerConfig{} if err := v.Unmarshal(&serverConfig); err != nil {  fmt.Println("解析结构体失败", err) } fmt.Println(serverConfig.MySqlConfig)}

五、在gin中集成配置文件

在和main.go同级的目录下创建两个xx.yml的文件

目录下创建一个config、global、initialize的文件夹

在config/config.go的文件中书写使用结构体读取配置的

package config// 定义mysql的type MySQLConfig struct { Host string `mapstructure:"host"` Port int `mapstructure:"port"` Username string `mapstructure:"username"` Password string `mapstructure:"password"`}type ServerConfig struct { Name string `mapstructure:"name"` // 嵌套使用,注意这个地方要和yaml中一样 MySqlConfig MySQLConfig `mapstructure:"mysql"`}

在global/global.go文件中定义全局变量(可能在全项目中某一个文件中需要获取配置)

var ( ServerConfig *config.ServerConfig = &config.ServerConfig{})

在initialize/config目录初始化配置

func GetEnvInfo(env string) bool { viper.AutomaticEnv() return viper.GetBool(env)}func InitConfig() { workDir, _ := os.Getwd() isDev := GetEnvInfo("IS_DEV") configFileName := path.Join(workDir, "11.config/application.prod.yml") if isDev {  configFileName = path.Join(workDir, "11.config/application.dev.yml") } v := viper.New() //文件的路径如何设置 v.SetConfigFile(configFileName) if err := v.ReadInConfig(); err != nil {  panic(err) } err := v.Unmarshal(&global.ServerConfig) if err != nil {  fmt.Println("读取配置失败") } fmt.Println(&global.ServerConfig)}

在mian.go中初始化配置文件并且打印出信息

func main() { // 初始化配置 initialize.InitConfig() fmt.Println(global.ServerConfig.MySqlConfig)}

到此，关于“go项目中环境变量的配置方法”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注编程网网站，小编会继续努力为大家带来更多实用的文章！