go項目中環境變量的配置方法

這篇文章主要介紹“go項目中環境變量的配置方法”，在日常操作中，相信很多人在go項目中環境變量的配置方法問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”go項目中環境變量的配置方法”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

目錄

• 一、為什么要配置文件

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

• 三、使用結構體的方式讀取

• 四、區分開發環境和生產環境

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

一、為什么要配置文件

1、容易編輯，配置比較集中，方便修改，在大業務量的系統里面，通過配置會方便后人理解整個系統的架構
2、做到業務代碼和環境解耦，環境配置變量不侵入到代碼層面
3、不同的環境切換比較方便

二、在go中使用配置文件

1、viper網站地址

2、本地安裝

go get github.com/spf13/viper

3、簡單的讀取配置文件

package main

import (
 "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"))
}

4、在文件的同級下創建一個config.yaml的文件

name: "test"

5、使用go run 文件名運行文件，這樣直接運行會報錯找不到文件

6、建議配置后可以使用右鍵運行項目

package main

import (
 "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"))
}

三、使用結構體的方式讀取

1、簡單的使用

package main

import (
 "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)
}

2、結構體的嵌套，很常見的是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)

四、區分開發環境和生產環境

1、在mac電腦上配置環境變量，在.bash_profile中導出一個環境變量，注意要關閉goland重新打開

# vim .bash_profile
export IS_DEV= true
source .bash_profile

2、定義方法直接獲取環境變量

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

3、在目錄下創建一個application.dev.yml和application.prod.yml的文件，分別存放不同環境下的配置

4、使用環境變量來讀取配置完整代碼

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中集成配置文件

1、在和main.go同級的目錄下創建兩個xx.yml的文件

2、目錄下創建一個config、global、initialize的文件夾

3、在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"`
}

4、在global/global.go文件中定義全局變量(可能在全項目中某一個文件中需要獲取配置)

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

5、在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)
}

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

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

到此，關于“go項目中環境變量的配置方法”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！