Golang實現獲取系統環(huán)境變量的方法詳解
1、Golang 獲取系統環(huán)境變量
Golang 要獲取系統的環(huán)境變量需要導入"os"包，通過os包中的Getenv方法來獲取。
func Getenv（key string） string {
}示例代碼：
package main
import (
"fmt"
"os"
)
func main() {
fmt.Println(os.Getenv("GOPATH"))
} 2、通過godotenv庫配置環(huán)境變量
2.1 簡介
微服務應用提倡將配置存儲在環(huán)境變量中。任何從開發(fā)環(huán)境切換到生產環(huán)境時需要修改的東西都從代碼抽取到環(huán)境變量里。但是在實際開發(fā)中，如果同一臺機器運行多個項目，設置環(huán)境變量容易沖突，不實用。godotenv庫從.env文件中讀取配置， 然后存儲到程序的環(huán)境變量中。在代碼中可以使用讀取非常方便。godotenv源于一個 Ruby 的開源項目dotenv。
2.2 快速使用
第三方庫需要先安裝：
go get github.com/joho/godotenv
后使用：
package main
import (
"fmt"
"log"
"os"
"github.com/joho/godotenv"
)
func main() {
err := godotenv.Load()
if err != nil {
log.Fatal(err)
}
fmt.Println("name: ", os.Getenv("name"))
fmt.Println("age: ", os.Getenv("age"))
}然后在可執(zhí)行程序相同目錄下，添加一個.env文件：
name = zmc
age = 20
運行程序，輸出：
name = zmc
age = 20
可見，使用非常方便。默認情況下，godotenv讀取項目根目錄下的.env文件，文件中使用key = value的格式，每行一個鍵值對。調用godotenv.Load()即可加載，可直接調用os.Getenv("key")讀取。
2.3 高級特性
自動加載
如果連Load方法都不想自己調用,可以導入github.com/joho/godotenv/autoload，配置會自動讀?。?/p>
package main
import (
"fmt"
"os"
_ "github.com/joho/godotenv/autoload"
)
func main() {
fmt.Println("name: ", os.Getenv("name"))
fmt.Println("age: ", os.Getenv("age"))
}注意，由于代碼中沒有顯式用到godotenv庫，需要使用空導入，即導入時包名前添加一個_。
看autoload包的源碼，其實就是庫幫你調用了Load方法：
// src/github.com/joho/godotenv/autoload/autoload.go
package autoload
/*
You can just read the .env file on import just by doing
import _ "github.com/joho/godotenv/autoload"
And bob's your mother's brother
*/
import"github.com/joho/godotenv"
func init() {
godotenv.Load()
}加載自定義文件
默認情況下，加載的是項目根目錄下的.env文件。當然我們可以加載任意名稱的文件，文件也不必以.env為后綴：
package main
import (
"fmt"
"log"
"os"
"github.com/joho/godotenv"
)
func main() {
err := godotenv.Load("common", "dev.env", "production.env")
if err != nil {
log.Fatal(err)
}
fmt.Println("name: ", os.Getenv("name"))
fmt.Println("version: ", os.Getenv("version"))
fmt.Println("database: ", os.Getenv("database"))
}common文件內容：
name = awesome web
version = 0.0.1
dev.env文件內容：
database = sqlite
production.env文件內容：
database = mysql
運行結果：
name: awesome web
version: 0.0.1
database: sqlite
注意：Load接收多個文件名作為參數，如果不傳入文件名，默認讀取.env文件的內容。如果多個文件中存在同一個鍵，那么先出現的優(yōu)先，后出現的不生效。所以，上面輸出的database是sqlite。
注釋
.env文件中可以添加注釋，注釋以#開始，直到該行結束。
# app name name = awesome web # current version version = 0.0.1
YAML
.env文件還可以使用 YAML 格式：
name: awesome web version: 0.0.1
不存入環(huán)境變量（重要）
godotenv允許不將.env文件內容存入環(huán)境變量，使用godotenv.Read()返回一個map[string]string，可直接使用：
package main
import (
"fmt"
"log"
"github.com/joho/godotenv"
)
func main() {
var myEnv map[string]string
myEnv, err := godotenv.Read()
if err != nil {
log.Fatal(err)
}
fmt.Println("name: ", myEnv["name"])
fmt.Println("version: ", myEnv["version"])
}輸出：
name: awesome web
version: 0.0.1
直接操作map，簡單直接！
數據源
除了讀取文件，還可以從string、中讀取配置：io.Reader
package main
import (
"fmt"
"log"
"github.com/joho/godotenv"
)
func main() {
content := `
name: awesome web
version: 0.0.1
`
myEnv, err := godotenv.Unmarshal(content)
if err != nil {
log.Fatal(err)
}
fmt.Println("name: ", myEnv["name"])
fmt.Println("version: ", myEnv["version"])
}輸出：
name: awesome web
version: 0.0.1
只要實現了io.Reader接口，就能作為數據源?？梢詮奈募?code>os.File），網絡（net.Conn），bytes.Buffer等多種來源讀?。?/p>
package main
import (
"bytes"
"fmt"
"log"
"os"
"github.com/joho/godotenv"
)
func main() {
file, _ := os.OpenFile(".env", os.O_RDONLY, 0666)
myEnv, err := godotenv.Parse(file)
if err != nil {
log.Fatal(err)
}
fmt.Println("name: ", myEnv["name"])
fmt.Println("version: ", myEnv["version"])
buf := &bytes.Buffer{}
buf.WriteString("name: awesome web @buffer")
buf.Write([]byte{'\n'})
buf.WriteString("version: 0.0.1")
myEnv, err = godotenv.Parse(buf)
if err != nil {
log.Fatal(err)
}
fmt.Println("name: ", myEnv["name"])
fmt.Println("version: ", myEnv["version"])
}輸出：
version: 0.0.1
name: awesome web @buffer
version: 0.0.1
注意，從字符串讀取和從io.Reader讀取使用的方法是不同的。前者為Unmarshal，后者是Parse。
生成.env文件
可以通過程序生成一個.env文件的內容，可以直接寫入到文件中：
package main
import (
"bytes"
"log"
"github.com/joho/godotenv"
)
func main() {
buf := &bytes.Buffer{}
buf.WriteString("name = awesome web")
buf.WriteByte('\n')
buf.WriteString("version = 0.0.1")
env, err := godotenv.Parse(buf)
if err != nil {
log.Fatal(err)
}
err = godotenv.Write(env, "./.env")
if err != nil {
log.Fatal(err)
}
}查看生成的.env文件：
name="awesome web"
version="0.0.1"
還可以返回一個字符串：
package main
import (
"bytes"
"fmt"
"log"
"github.com/joho/godotenv"
)
func main() {
buf := &bytes.Buffer{}
buf.WriteString("name = awesome web")
buf.WriteByte('\n')
buf.WriteString("version = 0.0.1")
env, err := godotenv.Parse(buf)
if err != nil {
log.Fatal(err)
}
content, err := godotenv.Marshal(env)
if err != nil {
log.Fatal(err)
}
fmt.Prin輸出：
ame="awesome web"
version="0.0.1"
命令行模式
godotenv還提供了一個命令行的模式：
godotenv -f ./.env command args
前面在go get安裝godotenv時，godotenv就已經安裝在$GOPATH/bin目錄下了，我習慣把$GOPATH/bin加入系統PATH，所以godotenv命令可以直接使用。
命令行模式就是讀取指定文件（如果不通過-f指定，則使用.env文件），設置環(huán)境變量，然后運行后面的程序。
我們簡單寫一個程序驗證一下：
package main
import (
"fmt"
"os"
)
func main() {
fmt.Println(os.Getenv("name"))
fmt.Println(os.Getenv("version"))
}使用godotenv運行一下：
godotenv -f ./.env go run main.go
輸出：
awesome web
0.0.1
多個環(huán)境
實踐中，一般會根據APP_ENV環(huán)境變量的值加載不同的文件：
package main
import (
"fmt"
"log"
"os"
"github.com/joho/godotenv"
)
func main() {
env := os.Getenv("GODAILYLIB_ENV")
if env == "" {
env = "development"
}
err := godotenv.Load(".env." + env)
if err != nil {
log.Fatal(err)
}
err = godotenv.Load()
if err != nil {
log.Fatal(err)
}
fmt.Println("name: ", os.Getenv("name"))
fmt.Println("version: ", os.Getenv("version"))
fmt.Println("database: ", os.Getenv("database"))
}我們先讀取環(huán)境變量GODAILYLIB_ENV，然后讀取對應的.env. + env，最后讀取默認的.env文件。
前面也提到過，先讀取到的優(yōu)先。我們可以在默認的.env文件中配置基礎信息和一些默認的值， 如果在開發(fā)/測試/生產環(huán)境需要修改，那么在對應的.env.development/.env.test/.env.production文件中再配置一次即可。
.env文件內容：
name = awesome web version = 0.0.1 database = file
.env.development文件內容：
database = sqlite3
.env.production文件內容：
database = mysql
運行程序：
# 默認是開發(fā)環(huán)境 $ go run main.go name: awesome web version: 0.0.1 database: sqlite3 # 設置為生產環(huán)境 $ GODAILYLIB_ENV=production go run main.go name: awesome web version: 0.0.1 database: mysql
2.4 一點源碼
godotenv讀取文件內容，為什么可以使用os.Getenv訪問：
// src/github.com/joho/godotenv/godotenv.go
func loadFile(filename string, overload bool) error {
envMap, err := readFile(filename)
if err != nil {
return err
}
currentEnv := map[string]bool{}
rawEnv := os.Environ()
for _, rawEnvLine := range rawEnv {
key := strings.Split(rawEnvLine, "=")[0]
currentEnv[key] = true
}
for key, value := range envMap {
if !currentEnv[key] || overload {
os.Setenv(key, value)
}
}
returnnil
}因為godotenv調用os.Setenv將鍵值對設置到環(huán)境變量中了。
到此這篇關于Golang實現獲取系統環(huán)境變量的方法詳解的文章就介紹到這了,更多相關Golang獲取系統環(huán)境變量內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
當存儲或傳輸 大整數（int64） 時,往往會出現精度丟失的問題,本文通過一個示例來詳細分析原因,并給出解決方案,文中的示例代碼講解詳細,感興趣的小伙伴可以了解下2026-01-01
Golang的Reflect包（reflect）是一個強大的內置包，它提供了在運行時進行程序反射的功能，幫助我們編寫更加靈活、通用且動態(tài)的代碼，為Golang開發(fā)者帶來了更多的可能性，感興趣的同學可以參考一下2023-06-06
某些情況下，對一個已存在的切片或數組進行切分操作可能會導致內存泄漏，本文主要介紹了Go 切片導致內存泄露的幾種原因，感興趣的可以了解一下2023-05-05
這篇文章主要介紹了golang 中獲取字符串個數 ，本文給大家介紹的非常詳細,具有一定的參考借鑒價值，需要的朋友可以參考下2019-08-08
反射是通過實體對象獲取反射對象(Value、Type)，然后可以操作相應的方法。本文將利用Go語言中的反射reflect實現獲取接口變量信息，需要的可以參考一下2022-05-05
這篇文章主要介紹了Golang反射修改變量值,也就是Golang反射三大定律中的前兩個，即從interface{}到反射對象和從反射對象到interface{}，需要的朋友可以參考下2022-12-12
這篇文章主要介紹了Go之interface的具體使用,小編覺得挺不錯的，現在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-03-03
這篇文章主要介紹了go mod tidy拉取依賴包bug問題及解決方案,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-09-09
這篇文章主要介紹了golang切片擴容規(guī)則實現，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-03-03
Fork/Join本質上是一種任務分解，將一個很大的任務分解成若干個小任務，然后再對小任務進一步分解，直到最小顆粒度，然后并發(fā)執(zhí)行，對Golang的Fork/Join實現代碼感興趣的朋友跟隨小編一起看看吧2023-01-01