前言

今天的文章比較基礎(chǔ)，但卻是必須掌握的，而且本文有些內(nèi)容，也許你之前沒(méi)想過(guò)。希望這篇文章能夠讓你理解環(huán)境變量并掌握 Go 環(huán)境變量相關(guān)操作。

01 從安裝 Go 說(shuō)起

其實(shí)不止是安裝 Go，其他語(yǔ)言一本也會(huì)有類(lèi)似的問(wèn)題。一般來(lái)說(shuō)，安裝完 Go 后，會(huì)建議將 go 可執(zhí)行程序配置到 PATH 環(huán)境變量中。

比如我本地的 PATH 環(huán)境變量的值：

$ echo $PATH
/Users/xuxinhua/.go/bin:/usr/local/sbin:/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin:/Applications/Wireshark.app/Contents/MacOS:/Users/xuxinhua/.cargo/bin:/Users/xuxinhua/bin:/usr/local/git/bin:/Users/xuxinhua/.composer/vendor/bin:/Users/xuxinhua/go/bin

那么 PATH 環(huán)境變量的作用是什么呢？

簡(jiǎn)單一句話就是，讓你在終端執(zhí)行命令時(shí)，不需要輸入絕對(duì)路徑。比如 Go 安裝在了 ~/.go/bin 目錄下，如果我們要執(zhí)行 Go 命令，得類(lèi)似這樣：

$ ~/.go/bin/go version

是不是很麻煩？！但將 ~/.go/bin 目錄加入到 PATH 環(huán)境變量后，可以直接這樣執(zhí)行 Go 命令：

$ go version

這就是 PATH 環(huán)境變量的作用：告知去哪里查找要執(zhí)行的命令。

那么環(huán)境變量的作用是什么呢？百科上關(guān)于環(huán)境變量的解釋?zhuān)?/p>

環(huán)境變量（environment variables）一般是指在操作系統(tǒng)中用來(lái)指定操作系統(tǒng)運(yùn)行環(huán)境的一些參數(shù)，如：臨時(shí)文件夾位置和系統(tǒng)文件夾位置等。

進(jìn)程也會(huì)有自己的環(huán)境變量，一般從父進(jìn)程繼承，也可以人為指定。比如在終端運(yùn)行某個(gè)程序時(shí)，可以給它傳遞環(huán)境變量：

$ NAME=polarisxu ./xxx

進(jìn)程中就可以通過(guò) NAME 獲取到 polarisxu 這個(gè)值。

環(huán)境變量可以說(shuō)無(wú)處不在，很多時(shí)候只是我們沒(méi)有細(xì)想而已。

注：因?yàn)?PATH 環(huán)境變量的作用機(jī)制，在 Shell、Dockerfile 等中，你需要時(shí)刻意識(shí)到，PATH 環(huán)境變量的值是什么，有沒(méi)有包含你的命令路徑，對(duì)于這樣的場(chǎng)景，可能更好的辦法是寫(xiě)絕對(duì)路徑，而不是依賴(lài) PATH。

02 Go 如何使用環(huán)境變量

很多大型應(yīng)用程序，會(huì)使用環(huán)境變量進(jìn)行配置（當(dāng)然也支持其他方式配置，比如 flag）。作為配置選項(xiàng)的環(huán)境變量大大簡(jiǎn)化了應(yīng)用程序的部署。這些在云基礎(chǔ)設(shè)施中也很常見(jiàn)。

通常，基于環(huán)境變量的配置，如果環(huán)境變量沒(méi)設(shè)置，程序會(huì)有一個(gè)默認(rèn)值。

在 Go 語(yǔ)言中，和環(huán)境變量相關(guān)的 API 主要在 os 包中。下面的 API 都加上了注釋。

// Environ 以 key=value 的形式返回所有環(huán)境變量。
func Environ() []string
// ExpandEnv 根據(jù)當(dāng)前環(huán)境變量的值替換字符串中的 ${var} 或 $var。
// 對(duì)未定義變量的引用將被空字符串替換。
func ExpandEnv(s string) string
// Getenv 檢索 key 這個(gè)鍵對(duì)應(yīng)的環(huán)境變量的值。
// 如果該環(huán)境變量不存在，返回空字符串。
// 要區(qū)分空值和未設(shè)置值，請(qǐng)使用 LookupEnv。
func Getenv(key string) string
// LookupEnv 檢索 key 這個(gè)鍵對(duì)應(yīng)的環(huán)境變量的值。
// 如果該環(huán)境變量存在，則返回對(duì)應(yīng)的值(可能為空)，并且布爾值為 true。
// 否則，返回值將為空，布爾值將為 false。
func LookupEnv(key string) (string, bool)
// Setenv 設(shè)置 key 這個(gè)鍵對(duì)應(yīng)的環(huán)境變量的值。
// 如果出錯(cuò)會(huì)返回錯(cuò)誤。
func Setenv(key, value string) error
// Unsetenv 取消設(shè)置單個(gè)環(huán)境變量。
func Unsetenv(key string) error
// Clearenv 將刪除所有環(huán)境變量。
func Clearenv()

此外，os/exec 中有一個(gè) LookPath 函數(shù)，和 PATH 環(huán)境變量有關(guān)：

// 在 PATH 環(huán)境變量對(duì)應(yīng)的目錄中搜索名為 file 的可執(zhí)行文件。
// 如果文件包含 /，則不會(huì)搜索 PATH，而是正常路徑查找。
// 返回的結(jié)果可能是絕對(duì)路徑或相對(duì)于當(dāng)前目錄的相對(duì)路徑。
func LookPath(file string) (string, error)

現(xiàn)在，通過(guò)一個(gè)例子看看這些 API 如何使用。

// main.go
package main

import (
 "fmt"
  "os"
)

func main() {
  name := os.Getenv("NAME")
  fmt.Println("name is:", name)
}

然后運(yùn)行：

$ NAME=polarisxu go run main.go
name is: polarisxu

如果前面的 NAME=polarisxu 沒(méi)有，則返回的 name 是空字符串。如果希望有默認(rèn)值，該怎么做？

package main

import (
    "fmt"
    "os"
)

func main() {
    name := GetenvDefault("NAME", "xuxinhua")
    fmt.Println("name is:", name)
}

func GetenvDefault(key, defVal string) string {
    val, ok := os.LookupEnv(key)
    if ok {
        return val
    }
    return defVal
}

通過(guò) os.LookupEnv 可以得到是否設(shè)置了環(huán)境變量。這時(shí)運(yùn)行 go run main.go 的結(jié)果會(huì)是：name is: xuxinhua。

以上就是 Go 中會(huì)常用到獲取環(huán)境變量的 API。

其他 API，用到的可能性不大。有兩個(gè) API 值得提一下：Environ() 和 ExpandEnv()。

前面提到過(guò)，進(jìn)程會(huì)從父進(jìn)程繼承環(huán)境變量。這里最重要的就是 PATH 環(huán)境變量。有時(shí)候，我們通過(guò) os/exec 包執(zhí)行外部程序時(shí)，可能會(huì)提示找不到命令，這時(shí)需要確認(rèn) PATH 是否正確。可能 Shell 下 PATH 包含了命令所在目錄，但進(jìn)程可能沒(méi)包含，我們可以在程序中輸出所有環(huán)境變量：

envs := os.Environ()
for _, env := range envs {
  fmt.Println(env)
}

一行是一個(gè)完整的環(huán)境變量，比如 LANG=zh_CN.UTF-8。

再看下 ExpandEnv() 函數(shù)。有以下代碼：（省略 main 相關(guān)其他代碼）

host := os.ExpandEnv("127.0.0.1:$PORT")
fmt.Println(host)

IP:PORT 的形式是常見(jiàn)的，通常，我們會(huì)做字符串拼接：host + ":" + port，有了 os.ExpandEnv，不需要進(jìn)行拼接了，它會(huì)將 $PORT 替換為 os.Getenv("PORT") 的值。

03 小結(jié)

環(huán)境變量你會(huì)用了嗎？

本文沒(méi)有通過(guò)代碼試驗(yàn)的其他函數(shù)，建議你可以寫(xiě)代碼試試，比如看看 os.Clearenv、os.Unsetenv 能不能刪除環(huán)境變量。

到此這篇關(guān)于Go獲取與設(shè)置環(huán)境變量的文章就介紹到這了,更多相關(guān)Go設(shè)置環(huán)境變量?jī)?nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論