如何為你的 Golang 專案設定環境變數

💡

如果想問問題或單純回饋的話可以填寫表單唷

前言

在開發專案的時候常常需要切換不同的環境，也因此我們通常會設定不同的環境變數。不過如果直接寫死在程式當中每次修改起來實在太麻煩了，如果可以透過動態的方式傳入環境變數的話可以減少很多不必要的修改，程式也相對比較乾淨。

flag

一般在 golang 可以利用 flag 這個 package 做到類似的效果，例如：

var env string
var accessToken string
func main() {
    flag.StringVar(&env, 'ENV', 'development', 'your current env')
    flag.StringVar(&accessToken, 'ACCESS_TOKEN', 'xxx-oo-ooo', 'your API access token')
    flag.Parse()

    // start your application
}

或者也可以參考這篇文章中提到的方式，用 go build -ldflag 在 compile time 設定變數。

雖然這樣的確不用寫死值在程式裡頭，但變數仍然要事先定義好。

透過 YAML 載入

為了解決上述問題，我們可以透過 yaml（或者你喜歡的 format）來統一管理環境變數。例如：

func LoadEnv(filename string) bool {
    file, err := ioutil.ReadFile(filename)
    if err != nil {
        // 在 production，我們可能會直接用 console 設定環境變數
        return false
    }

    var config := make(map[string]string)
    yaml.Unmarshal(file, &config)
    for k, v := range config {
		os.Setenv(k, v)
	}
}

將 yaml 裡的變數載入後，在統一用 os.Setenv(k, v) 存入，這樣一來在 local 的環境下就可以很容易地調整變數，記得要把設定當給 ignore 掉，不然如果變數裡頭有敏感資料，又剛好是 public 的 repo 就會被看光光了。

後記

我寫了一個相當簡單的 repo 來做這件事，如果在開發上真的有需要的話再持續優化他吧 XD

← 透過 Vuex 與 webpack dynamic import 動態載入 module 在 Go 如何處理 connection lost？ →

如果覺得這篇文章對你有幫助的話，可以考慮下面的連結請我喝一杯 ☕ 可以讓我平凡的一天變得閃閃發光 ✨

☕Buy me a coffee

如何為你的 Golang 專案設定環境變數

目錄

前言

flag

透過 YAML 載入

後記

目錄