go項目中環(huán)境變量的配置
更新時間：2021年07月13日 11:17:16 作者：水痕001
本文主要介紹了go項目中環(huán)境變量的配置，文中通過示例代碼介紹的非常詳細(xì)，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
一、為什么要配置文件
1、容易編輯，配置比較集中，方便修改，在大業(yè)務(wù)量的系統(tǒng)里面，通過配置會方便后人理解整個系統(tǒng)的架構(gòu)
2、做到業(yè)務(wù)代碼和環(huán)境解耦，環(huán)境配置變量不侵入到代碼層面
3、不同的環(huán)境切換比較方便
二、在go中使用配置文件
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、在文件的同級下創(chuàng)建一個config.yaml的文件
name: "test"
5、使用go run 文件名運(yùn)行文件，這樣直接運(yùn)行會報錯找不到文件
6、建議配置后可以使用右鍵運(yùn)行項目
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"))
}
三、使用結(jié)構(gòu)體的方式讀取
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("解析結(jié)構(gòu)體失敗", err)
}
fmt.Println(serverConfig.Name)
}
2、結(jié)構(gòu)體的嵌套，很常見的是mysql數(shù)據(jù)庫的配置
// 定義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)
四、區(qū)分開發(fā)環(huán)境和生產(chǎn)環(huán)境
1、在mac電腦上配置環(huán)境變量，在.bash_profile中導(dǎo)出一個環(huán)境變量，注意要關(guān)閉goland重新打開
# vim .bash_profile export IS_DEV= true source .bash_profile
2、定義方法直接獲取環(huán)境變量
func GetEnvInfo(env string) bool {
viper.AutomaticEnv()
return viper.GetBool(env)
}
func main() {
fmt.Println(GetEnvInfo("IS_DEV"))
}
3、在目錄下創(chuàng)建一個application.dev.yml和application.prod.yml的文件，分別存放不同環(huán)境下的配置
4、使用環(huán)境變量來讀取配置完整代碼
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("解析結(jié)構(gòu)體失敗", err)
}
fmt.Println(serverConfig.MySqlConfig)
}
五、在gin中集成配置文件
1、在和main.go同級的目錄下創(chuàng)建兩個xx.yml的文件
2、目錄下創(chuàng)建一個config、global、initialize的文件夾
3、在config/config.go的文件中書寫使用結(jié)構(gòu)體讀取配置的
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()
//文件的路徑如何設(shè)置
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)
}
到此這篇關(guān)于go項目中環(huán)境變量的配置的文章就介紹到這了,更多相關(guān)go 環(huán)境變量配置內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Go語言是靜態(tài)類型語言,因此變量有明確類型的,編譯器也會檢查變量類型的正確性,下面這篇文章主要給大家介紹了關(guān)于GO語言中變量的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-07-07
本文主要介紹了記一次go語言使用time.Duration類型踩過的坑，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-01-01
這篇文章主要為大家介紹了Go語言基礎(chǔ)枚舉的用法及示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2021-11-11
這篇文章主要為大家介紹了Go?Redis客戶端使用對比詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-07-07
這篇文章主要介紹了Go語言輸出函數(shù)使用詳解的相關(guān)資料,需要的朋友可以參考下2023-08-08
逃逸分析是指由編譯器決定內(nèi)存分配的位置，不需要程序員指定，這篇文章主要為大家詳細(xì)介紹了Golang中內(nèi)存逃逸分析的幾種方法，需要的可以參考一下2023-07-07
這篇文章主要介紹了golang 字符串切片去重實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12
Golang的循環(huán)語句和循環(huán)控制語句詳解
循環(huán)語句為了簡化程序中有規(guī)律的重復(fù)性操作，需要用到循環(huán)語句，和其他大多數(shù)編程語言一樣，GO的循環(huán)語句有for循環(huán)，不同的是沒有while循環(huán)，而循環(huán)控制語句可以改變循環(huán)語句的執(zhí)行過程，下面給大家介紹下go循環(huán)語句和循環(huán)控制語句的相關(guān)知識，一起看看吧2021-11-11
今天小編就為大家分享一篇關(guān)于在Go中構(gòu)建并發(fā)TCP服務(wù)器的文章，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2018-10-10
這篇文章主要介紹了Windows下Goland的環(huán)境搭建過程,本文通過圖文并茂的形式給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-10-10